Searched refs:CreateBatchNormGrad (Results 1 – 4 of 4) sorted by relevance
114 builder.AddInstruction(HloInstruction::CreateBatchNormGrad( in TEST_F()
648 static std::unique_ptr<HloInstruction> CreateBatchNormGrad(
156 instruction = CreateBatchNormGrad(shape, operands(0), operands(1), in CreateFromProto()1081 HloInstruction::CreateBatchNormGrad(const Shape& shape, HloInstruction* operand, in CreateBatchNormGrad() function in xla::HloInstruction
1357 instruction = builder->AddInstruction(HloInstruction::CreateBatchNormGrad( in ParseInstructionRhs()