Searched refs:loop_parallelism_ (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | transpose.cc | 723 absl::c_accumulate(loop_parallelism_, int{1}, std::multiplies<int>()); in BuildPlanNodes()
776 agendum.num_tasks_at_loop / loop_parallelism_[agendum.loop_id]; in BuildPlanNodes()
799 num_iterations, loop_parallelism_[agendum.loop_id]); in BuildPlanNodes()
825 task_id == loop_parallelism_[agendum.loop_id] - 1) { in BuildPlanNodes()
857 num_iterations, loop_parallelism_[agendum.loop_id]); in BuildPlanNodes()
1191 loop_parallelism_ = ChooseParallelizationStrategy(inverse_permutation); in Initialize()
1193 absl::c_accumulate(loop_parallelism_, int{1}, std::multiplies<int>()); in Initialize()
1323 absl::StrJoin(loop_parallelism_, ","), outer_block_elems_a_, in ToString()
|
| H A D | transpose.h | 212 std::vector<int> loop_parallelism_; variable
|