Searched refs:sharding1 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_test.cc | 166 HloSharding sharding1 = HloSharding::Replicate(); in TEST_F() local
168 EXPECT_TRUE(hash_compare_equal(sharding1, sharding2)); in TEST_F()
172 HloSharding sharding1 = HloSharding::AssignDevice(1); in TEST_F() local
174 EXPECT_TRUE(hash_compare_equal(sharding1, sharding2)); in TEST_F()
178 HloSharding sharding1 = HloSharding::AssignDevice(1); in TEST_F() local
180 EXPECT_FALSE(hash_compare_equal(sharding1, sharding2)); in TEST_F()
184 HloSharding sharding1 = HloSharding::Tile(MakeArray({2, 2}, {0, 3, 2, 1})); in TEST_F() local
186 EXPECT_TRUE(hash_compare_equal(sharding1, sharding2)); in TEST_F()
193 HloSharding sharding1 = HloSharding::Replicate(); in TEST_F() local
195 EXPECT_FALSE(hash_compare_equal(sharding1, sharding2)); in TEST_F()
[all …]
|
| D | hlo_sharding_metadata.cc | 58 bool ShardingMatches(const HloSharding& sharding1, in ShardingMatches() argument
60 auto single_sharding1 = sharding1.ExtractSingleSharding(); in ShardingMatches()
69 return sharding1 == sharding2; in ShardingMatches()
|