Searched refs:resharded_rhs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 549 auto resharded_rhs = rhs.Reshard(*lhs_sharding_transposed_to_match_rhs); in PartitionBaseCase() local
552 create_sharded_dot(lhs.hlo(), resharded_rhs.hlo(), b, conv_window)); in PartitionBaseCase()
1553 auto resharded_rhs = in PartitionBaseCase() local
1556 auto dot, create_sharded_dot(resharded_lhs.hlo(), resharded_rhs.hlo(), in PartitionBaseCase()
1573 auto resharded_rhs = in PartitionBaseCase() local
1577 resharded_rhs.hlo(), b, conv_window)); in PartitionBaseCase()
|
| D | spmd_partitioner_test.cc | 620 auto resharded_rhs = in TEST_F() local
639 resharded_rhs), in TEST_F()
5164 auto resharded_rhs = in TEST_F() local
5171 op::AllReduce(op::Dot(lhs, resharded_rhs))), in TEST_F()
5933 auto resharded_rhs = AllOf( in TEST_F() local
5936 EXPECT_THAT(root, AllOf(op::Convolution(lhs, resharded_rhs), in TEST_F()
6115 auto resharded_rhs = AllOf( in TEST_F() local
6118 EXPECT_THAT(root, AllOf(op::Convolution(lhs, resharded_rhs), in TEST_F()
6656 auto resharded_rhs = in TEST_F() local
6659 EXPECT_THAT(root, AllOf(op::Convolution(resharded_lhs, resharded_rhs), in TEST_F()
|