Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator.h347 Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override { in HandleBatchNormGrad() function
H A Ddfs_hlo_visitor_with_default.h68 Status HandleBatchNormGrad(HloInstructionPtr hlo) override { in HandleBatchNormGrad() function
H A Dbatchnorm_expander.cc383 Status BatchNormExpanderVisitor::HandleBatchNormGrad( in HandleBatchNormGrad() function in xla::BatchNormExpanderVisitor
H A Dhlo_cost_analysis.cc540 Status HloCostAnalysis::HandleBatchNormGrad(const HloInstruction*) { in HandleBatchNormGrad() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc1577 Status ShapeVerifier::HandleBatchNormGrad(HloInstruction* batch_norm_grad) { in HandleBatchNormGrad() function in xla::ShapeVerifier
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter.cc608 Status IrEmitter::HandleBatchNormGrad(HloInstruction*) { in HandleBatchNormGrad() function in xla::gpu::IrEmitter