Home
last modified time | relevance | path

Searched refs:HandleBatchNormTraining (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc42 Status HandleBatchNormTraining(HloInstruction* batch_norm) override;
121 Status Visitor::HandleBatchNormTraining(HloInstruction* batch_norm) { in HandleBatchNormTraining() function in xla::gpu::__anon8b68da910111::Visitor
Dir_emitter.h98 Status HandleBatchNormTraining(HloInstruction* batch_norm) override;
Dir_emitter.cc620 Status IrEmitter::HandleBatchNormTraining(HloInstruction*) { in HandleBatchNormTraining() function in xla::gpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc51 Status HandleBatchNormTraining(HloInstruction* batch_norm) override;
152 Status BatchNormExpanderVisitor::HandleBatchNormTraining( in HandleBatchNormTraining() function in xla::BatchNormExpanderVisitor
Dhlo_evaluator.h279 Status HandleBatchNormTraining(HloInstruction* batch_norm_training) override { in HandleBatchNormTraining() function
Dhlo_cost_analysis.h96 Status HandleBatchNormTraining(
Ddfs_hlo_visitor_with_default.h62 Status HandleBatchNormTraining(HloInstructionPtr hlo) override { in HandleBatchNormTraining() function
Dhlo_verifier.h104 Status HandleBatchNormTraining(HloInstruction* batch_norm_training) override;
Ddfs_hlo_visitor.h277 virtual Status HandleBatchNormTraining(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc520 Status HloCostAnalysis::HandleBatchNormTraining(const HloInstruction*) { in HandleBatchNormTraining() function in xla::HloCostAnalysis
Dhlo_verifier.cc1037 Status ShapeVerifier::HandleBatchNormTraining( in HandleBatchNormTraining() function in xla::ShapeVerifier
Dhlo_instruction.cc3065 return visitor->HandleBatchNormTraining(this); in Visit()