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