Searched refs:batch_norm_inference (Results 1 – 4 of 4) sorted by relevance
732 HloInstruction* batch_norm_inference) { in HandleBatchNormInference() argument733 return CheckShape(batch_norm_inference, in HandleBatchNormInference()735 batch_norm_inference->operand(0)->shape(), in HandleBatchNormInference()736 batch_norm_inference->operand(1)->shape(), in HandleBatchNormInference()737 batch_norm_inference->operand(2)->shape(), in HandleBatchNormInference()738 batch_norm_inference->operand(3)->shape(), in HandleBatchNormInference()739 batch_norm_inference->operand(4)->shape(), in HandleBatchNormInference()740 batch_norm_inference->feature_index())); in HandleBatchNormInference()
90 const HloInstruction* batch_norm_inference) override;
251 HloInstruction* batch_norm_inference) override { in HandleBatchNormInference() argument
95 HloInstruction* batch_norm_inference) override;