Searched defs:HandleBatchNormTraining (Results 1 – 7 of 7) sorted by relevance
279 Status HandleBatchNormTraining(HloInstruction* batch_norm_training) override { in HandleBatchNormTraining() function
62 Status HandleBatchNormTraining(HloInstructionPtr hlo) override { in HandleBatchNormTraining() function
152 Status BatchNormExpanderVisitor::HandleBatchNormTraining( in HandleBatchNormTraining() function in xla::BatchNormExpanderVisitor
520 Status HloCostAnalysis::HandleBatchNormTraining(const HloInstruction*) { in HandleBatchNormTraining() function in xla::HloCostAnalysis
1037 Status ShapeVerifier::HandleBatchNormTraining( in HandleBatchNormTraining() function in xla::ShapeVerifier
121 Status Visitor::HandleBatchNormTraining(HloInstruction* batch_norm) { in HandleBatchNormTraining() function in xla::gpu::__anon8335950a0111::Visitor
620 Status IrEmitter::HandleBatchNormTraining(HloInstruction*) { in HandleBatchNormTraining() function in xla::gpu::IrEmitter