Searched defs:HandleBatchNormInference (Results 1 – 7 of 7) sorted by relevance
262 Status HandleBatchNormInference( in HandleBatchNormInference() function
66 Status HandleBatchNormInference(HloInstructionPtr hlo) override { in HandleBatchNormInference() function
293 Status BatchNormExpanderVisitor::HandleBatchNormInference( in HandleBatchNormInference() function in xla::BatchNormExpanderVisitor
491 Status HloCostAnalysis::HandleBatchNormInference(const HloInstruction*) { in HandleBatchNormInference() function in xla::HloCostAnalysis
912 Status ShapeVerifier::HandleBatchNormInference( in HandleBatchNormInference() function in xla::ShapeVerifier
69 Status Visitor::HandleBatchNormInference(HloInstruction* batch_norm) { in HandleBatchNormInference() function in xla::gpu::__anonbbe443b50111::Visitor
846 Status IrEmitter::HandleBatchNormInference(HloInstruction*) { in HandleBatchNormInference() function in xla::gpu::IrEmitter