Searched refs:HloBatchNormInferenceInstruction (Results 1 – 3 of 3) sorted by relevance
126 HloBatchNormInferenceInstruction::HloBatchNormInferenceInstruction( in HloBatchNormInferenceInstruction() function in xla::HloBatchNormInferenceInstruction138 HloBatchNormInferenceInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()142 return absl::make_unique<HloBatchNormInferenceInstruction>( in CloneWithNewOperandsImpl()
76 class HloBatchNormInferenceInstruction : public HloBatchNormInstruction {78 explicit HloBatchNormInferenceInstruction(
1337 return absl::make_unique<HloBatchNormInferenceInstruction>( in CreateBatchNormInference()