Home
last modified time | relevance | path

Searched refs:batch_norm_grad (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc743 Status ShapeVerifier::HandleBatchNormGrad(HloInstruction* batch_norm_grad) { in HandleBatchNormGrad() argument
744 return CheckShape(batch_norm_grad, ShapeInference::InferBatchNormGradShape( in HandleBatchNormGrad()
745 batch_norm_grad->operand(0)->shape(), in HandleBatchNormGrad()
746 batch_norm_grad->operand(1)->shape(), in HandleBatchNormGrad()
747 batch_norm_grad->operand(2)->shape(), in HandleBatchNormGrad()
748 batch_norm_grad->operand(3)->shape(), in HandleBatchNormGrad()
749 batch_norm_grad->operand(4)->shape(), in HandleBatchNormGrad()
750 batch_norm_grad->feature_index())); in HandleBatchNormGrad()
Dhlo_cost_analysis.h91 Status HandleBatchNormGrad(const HloInstruction* batch_norm_grad) override;
Dhlo_evaluator.h247 Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override { in HandleBatchNormGrad() argument
Dhlo_verifier.h96 Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override;