Searched defs:batch_norm_inference (Results 1 – 2 of 2) sorted by relevance
251 HloInstruction* batch_norm_inference) override { in HandleBatchNormInference()
732 HloInstruction* batch_norm_inference) { in HandleBatchNormInference()