Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc722 HloInstruction* batch_norm_training) { in HandleBatchNormTraining() argument
723 return CheckShape(batch_norm_training, in HandleBatchNormTraining()
725 batch_norm_training->operand(0)->shape(), in HandleBatchNormTraining()
726 batch_norm_training->operand(1)->shape(), in HandleBatchNormTraining()
727 batch_norm_training->operand(2)->shape(), in HandleBatchNormTraining()
728 batch_norm_training->feature_index())); in HandleBatchNormTraining()
Dhlo_cost_analysis.h88 const HloInstruction* batch_norm_training) override;
Dhlo_evaluator.h255 Status HandleBatchNormTraining(HloInstruction* batch_norm_training) override { in HandleBatchNormTraining() argument
Dhlo_verifier.h93 Status HandleBatchNormTraining(HloInstruction* batch_norm_training) override;