Home
last modified time | relevance | path

Searched refs:matching_existing_index (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2526 int64 matching_existing_index = 0; in HandleInfeed() local
2527 for (; matching_existing_index < per_branch_partitioned_shapes.size(); in HandleInfeed()
2528 ++matching_existing_index) { in HandleInfeed()
2531 per_branch_partitioned_shapes[matching_existing_index])) { in HandleInfeed()
2535 if (matching_existing_index < per_branch_partitioned_shapes.size()) { in HandleInfeed()
2536 conditional_branch_indices[i] = matching_existing_index; in HandleInfeed()
2928 int64 matching_existing_index = 0; in HandleOutfeed() local
2929 for (; matching_existing_index < per_branch_partitioned_shapes.size(); in HandleOutfeed()
2930 ++matching_existing_index) { in HandleOutfeed()
2933 per_branch_partitioned_shapes[matching_existing_index])) { in HandleOutfeed()
[all …]