Searched defs:batch_norm_grad (Results 1 – 2 of 2) sorted by relevance
271 Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override { in HandleBatchNormGrad()
1059 Status ShapeVerifier::HandleBatchNormGrad(HloInstruction* batch_norm_grad) { in HandleBatchNormGrad()