Searched refs:conditional_branch_indices (Results 1 – 1 of 1) sorted by relevance
2522 std::vector<int32> conditional_branch_indices(num_partitions_); in HandleInfeed() local2536 conditional_branch_indices[i] = matching_existing_index; in HandleInfeed()2538 conditional_branch_indices[i] = per_branch_partitioned_shapes.size(); in HandleInfeed()2557 LiteralUtil::CreateR1<int32>(conditional_branch_indices))); in HandleInfeed()2924 std::vector<int32> conditional_branch_indices(num_partitions_); in HandleOutfeed() local2938 conditional_branch_indices[i] = matching_existing_index; in HandleOutfeed()2940 conditional_branch_indices[i] = per_branch_partitioned_shapes.size(); in HandleOutfeed()2960 LiteralUtil::CreateR1<int32>(conditional_branch_indices))); in HandleOutfeed()