Searched refs:CreateBatchNormTraining (Results 1 – 4 of 4) sorted by relevance
71 builder.AddInstruction(HloInstruction::CreateBatchNormTraining( in TEST_F()
637 static std::unique_ptr<HloInstruction> CreateBatchNormTraining(
147 CreateBatchNormTraining(shape, operands(0), operands(1), operands(2), in CreateFromProto()1062 HloInstruction::CreateBatchNormTraining(const Shape& shape, in CreateBatchNormTraining() function in xla::HloInstruction
1325 builder->AddInstruction(HloInstruction::CreateBatchNormTraining( in ParseInstructionRhs()