Searched refs:activation_sharding (Results 1 – 1 of 1) sorted by relevance
588 HloSharding activation_sharding = in HandleBatchNormGrad() local597 inst->set_sharding(activation_sharding); in HandleBatchNormGrad()