Searched refs:default_sharding (Results 1 – 2 of 2) sorted by relevance
189 HloSharding default_sharding = HloSharding::Replicate(); in TEST_F() local192 default_sharding); in TEST_F()200 default_sharding); in TEST_F()209 default_sharding); in TEST_F()213 default_sharding); in TEST_F()223 default_sharding); in TEST_F()227 default_sharding); in TEST_F()
309 HloSharding default_sharding = in HandleBatchNormTraining() local317 inst->set_sharding(default_sharding); in HandleBatchNormTraining()413 HloSharding default_sharding = in HandleBatchNormInference() local421 inst->set_sharding(default_sharding); in HandleBatchNormInference()591 HloSharding default_sharding = in HandleBatchNormGrad() local599 inst->set_sharding(default_sharding); in HandleBatchNormGrad()