Searched defs:default_sharding (Results 1 – 2 of 2) sorted by relevance
309 HloSharding default_sharding = in HandleBatchNormTraining() local413 HloSharding default_sharding = in HandleBatchNormInference() local591 HloSharding default_sharding = in HandleBatchNormGrad() local
189 HloSharding default_sharding = HloSharding::Replicate(); in TEST_F() local