Searched refs:resharded_lhs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 5193 auto resharded_lhs = in TEST_F() local
5201 op::Dot(resharded_lhs, rhs_slice))); in TEST_F()
5969 auto resharded_lhs = AllOf( in TEST_F() local
5972 EXPECT_THAT(root, AllOf(op::Convolution(resharded_lhs, rhs), in TEST_F()
6042 auto resharded_lhs = AllOf( in TEST_F() local
6046 AllOf(op::Convolution(resharded_lhs, rhs), op::Shape("f32[5,1,1,512]")); in TEST_F()
6152 auto resharded_lhs = AllOf( in TEST_F() local
6155 EXPECT_THAT(root, AllOf(op::Convolution(resharded_lhs, rhs), in TEST_F()
6301 auto resharded_lhs = in TEST_F() local
6308 op::CollectivePermute(op::Slice(resharded_lhs))); in TEST_F()
[all …]
|
| D | dot_handler.cc | 566 auto resharded_lhs = lhs.Reshard(*rhs_sharding_transposed_to_match_lhs); in PartitionBaseCase() local
569 create_sharded_dot(resharded_lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionBaseCase()
1551 auto resharded_lhs = in PartitionBaseCase() local
1556 auto dot, create_sharded_dot(resharded_lhs.hlo(), resharded_rhs.hlo(), in PartitionBaseCase()
1562 auto resharded_lhs = in PartitionBaseCase() local
1566 auto dot, create_sharded_dot(resharded_lhs.hlo(), in PartitionBaseCase()
|