Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc39 Status HandleBatchNormInference(HloInstruction* batch_norm) override;
54 Status Visitor::HandleBatchNormInference(HloInstruction* batch_norm) { in HandleBatchNormInference() function in xla::gpu::__anondf8c4db30111::Visitor
Dir_emitter.h100 Status HandleBatchNormInference(HloInstruction* batch_norm) override;
Dir_emitter.cc758 Status IrEmitter::HandleBatchNormInference(HloInstruction*) { in HandleBatchNormInference() function in xla::gpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h89 Status HandleBatchNormInference(
Ddfs_hlo_visitor_with_default.h65 Status HandleBatchNormInference(HloInstructionPtr hlo) override { in HandleBatchNormInference() function
Dbatchnorm_expander.cc58 Status HandleBatchNormInference(HloInstruction* batch_norm) override;
326 Status BatchNormExpanderVisitor::HandleBatchNormInference( in HandleBatchNormInference() function in xla::BatchNormExpanderVisitor
Dhlo_evaluator.h250 Status HandleBatchNormInference( in HandleBatchNormInference() function
Dhlo_verifier.h94 Status HandleBatchNormInference(
Ddfs_hlo_visitor.h259 virtual Status HandleBatchNormInference(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc401 Status HloCostAnalysis::HandleBatchNormInference(const HloInstruction*) { in HandleBatchNormInference() function in xla::HloCostAnalysis
Dhlo_verifier.cc731 Status ShapeVerifier::HandleBatchNormInference( in HandleBatchNormInference() function in xla::ShapeVerifier
Dhlo_instruction.cc2488 return visitor->HandleBatchNormInference(this); in Visit()