Home
last modified time | relevance | path

Searched defs:HandleBatchNormInference (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h274 Status HandleBatchNormInference( in HandleBatchNormInference() function
Ddfs_hlo_visitor_with_default.h66 Status HandleBatchNormInference(HloInstructionPtr hlo) override { in HandleBatchNormInference() function
Dbatchnorm_expander.cc293 Status BatchNormExpanderVisitor::HandleBatchNormInference( in HandleBatchNormInference() function in xla::BatchNormExpanderVisitor
Dhlo_cost_analysis.cc525 Status HloCostAnalysis::HandleBatchNormInference(const HloInstruction*) { in HandleBatchNormInference() function in xla::HloCostAnalysis
Dhlo_verifier.cc1047 Status ShapeVerifier::HandleBatchNormInference( in HandleBatchNormInference() function in xla::ShapeVerifier
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc69 Status Visitor::HandleBatchNormInference(HloInstruction* batch_norm) { in HandleBatchNormInference() function in xla::gpu::__anon8b68da910111::Visitor
Dir_emitter.cc612 Status IrEmitter::HandleBatchNormInference(HloInstruction*) { in HandleBatchNormInference() function in xla::gpu::IrEmitter