Searched refs:sd_pairs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 1269 std::vector<std::pair<int64, int64>> sd_pairs(num_partitions); in PartitionBaseCase() local
1272 sd_pairs[source] = {source, in PartitionBaseCase()
1283 &body_b, cp_input, sd_pairs, in PartitionBaseCase()
1305 &body_b, cp_input, sd_pairs, in PartitionBaseCase()
1357 std::vector<std::pair<int64, int64>> sd_pairs(num_partitions); in PartitionBaseCase() local
1360 sd_pairs[source] = {source, in PartitionBaseCase()
1365 &cp_b, p, sd_pairs, (*lhs.state().next_channel_id)++); in PartitionBaseCase()
|