Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h280 Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override { in HandleBatchNormGrad() function
Ddfs_hlo_visitor_with_default.h70 Status HandleBatchNormGrad(HloInstructionPtr hlo) override { in HandleBatchNormGrad() function
Dbatchnorm_expander.cc380 Status BatchNormExpanderVisitor::HandleBatchNormGrad( in HandleBatchNormGrad() function in xla::BatchNormExpanderVisitor
Dhlo_cost_analysis.cc531 Status HloCostAnalysis::HandleBatchNormGrad(const HloInstruction*) { in HandleBatchNormGrad() function in xla::HloCostAnalysis
Dhlo_verifier.cc1396 Status ShapeVerifier::HandleBatchNormGrad(HloInstruction* batch_norm_grad) { in HandleBatchNormGrad() function in xla::ShapeVerifier
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc210 Status Visitor::HandleBatchNormGrad(HloInstruction* batch_norm) { in HandleBatchNormGrad() function in xla::gpu::__anon4073e8b60111::Visitor
Dir_emitter.cc623 Status IrEmitter::HandleBatchNormGrad(HloInstruction*) { in HandleBatchNormGrad() function in xla::gpu::IrEmitter