Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator.h350 Status HandleBatchNormInference( in HandleBatchNormInference() function
H A Ddfs_hlo_visitor_with_default.h64 Status HandleBatchNormInference(HloInstructionPtr hlo) override { in HandleBatchNormInference() function
H A Dbatchnorm_expander.cc294 Status BatchNormExpanderVisitor::HandleBatchNormInference( in HandleBatchNormInference() function in xla::BatchNormExpanderVisitor
H A Dhlo_cost_analysis.cc535 Status HloCostAnalysis::HandleBatchNormInference(const HloInstruction*) { in HandleBatchNormInference() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc1565 Status ShapeVerifier::HandleBatchNormInference( in HandleBatchNormInference() function in xla::ShapeVerifier
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter.cc594 Status IrEmitter::HandleBatchNormInference(HloInstruction*) { in HandleBatchNormInference() function in xla::gpu::IrEmitter