Searched refs:HloBatchNormInferenceInstruction (Results 1 – 3 of 3) sorted by relevance
161 HloBatchNormInferenceInstruction::HloBatchNormInferenceInstruction( in HloBatchNormInferenceInstruction() function in xla::HloBatchNormInferenceInstruction173 HloBatchNormInferenceInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()177 return std::make_unique<HloBatchNormInferenceInstruction>( in CloneWithNewOperandsImpl()
139 class HloBatchNormInferenceInstruction : public HloBatchNormInstruction {141 explicit HloBatchNormInferenceInstruction(
1636 return std::make_unique<HloBatchNormInferenceInstruction>( in CreateBatchNormInference()