Home
last modified time | relevance | path

Searched defs:HandleBatchNormTraining (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc88 Status Visitor::HandleBatchNormTraining(HloInstruction* batch_norm) { in HandleBatchNormTraining() function in xla::gpu::__anona02ebda70111::Visitor
Dir_emitter.cc749 Status IrEmitter::HandleBatchNormTraining(HloInstruction*) { in HandleBatchNormTraining() function in xla::gpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h55 Status HandleBatchNormTraining(HloInstructionPtr hlo) override { in HandleBatchNormTraining() function
Dbatchnorm_expander.cc147 Status BatchNormExpanderVisitor::HandleBatchNormTraining( in HandleBatchNormTraining() function in xla::BatchNormExpanderVisitor
Dhlo_verifier.cc340 Status ShapeVerifier::HandleBatchNormTraining( in HandleBatchNormTraining() function in xla::ShapeVerifier
Dhlo_cost_analysis.cc362 Status HloCostAnalysis::HandleBatchNormTraining(const HloInstruction*) { in HandleBatchNormTraining() function in xla::HloCostAnalysis