Searched refs:HloBatchNormTrainingInstruction (Results 1 – 3 of 3) sorted by relevance
105 HloBatchNormTrainingInstruction::HloBatchNormTrainingInstruction( in HloBatchNormTrainingInstruction() function in xla::HloBatchNormTrainingInstruction114 HloBatchNormTrainingInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()118 return absl::make_unique<HloBatchNormTrainingInstruction>( in CloneWithNewOperandsImpl()
59 class HloBatchNormTrainingInstruction : public HloBatchNormInstruction {61 explicit HloBatchNormTrainingInstruction(const Shape& shape,
1067 return absl::make_unique<HloBatchNormTrainingInstruction>( in CreateBatchNormTraining()