Searched refs:right_halo (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 628 auto right_halo = AllOf(op::CollectivePermute(op::Slice(lhs)), in TEST_F() local
635 op::Concatenate(left_halo, lhs, right_halo), in TEST_F()
864 auto right_halo = AllOf(op::Shape("f32[2,2]{1,0}"), in TEST_F() local
869 op::Pad(op::Concatenate(sharded_input, right_halo), op::Constant())), in TEST_F()
943 auto right_halo = AllOf(op::Shape("f32[2,2]{1,0}"), in TEST_F() local
948 op::Pad(op::Concatenate(sharded_input, right_halo), op::Constant())), in TEST_F()
987 auto right_halo = AllOf(op::Shape("f32[1,2]{1,0}"), in TEST_F() local
993 op::Pad(op::Concatenate(left_halo, sharded_input, right_halo), in TEST_F()
1105 auto right_halo = AllOf(op::CollectivePermute(op::Slice(lhs)), in TEST_F() local
1110 op::Concatenate(left_halo, lhs, right_halo), in TEST_F()
[all …]
|
| D | spmd_partitioner_util.cc | 877 auto right_halo = in ExchangeHalo() local
880 concat_pieces.push_back(right_halo); in ExchangeHalo()
|