Searched refs:CreateBatchNormGrad (Results 1 – 4 of 4) sorted by relevance
113 builder.AddInstruction(HloInstruction::CreateBatchNormGrad( in TEST_F()
972 static std::unique_ptr<HloInstruction> CreateBatchNormGrad(
179 instruction = CreateBatchNormGrad(shape, operands(0), operands(1), in CreateFromProto()1641 HloInstruction::CreateBatchNormGrad(const Shape& shape, HloInstruction* operand, in CreateBatchNormGrad() function in xla::HloInstruction
2399 return builder->AddInstruction(HloInstruction::CreateBatchNormGrad( in CreateInstruction()