Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc41 Status HandleBatchNormGrad(HloInstruction* batch_norm) override;
159 Status Visitor::HandleBatchNormGrad(HloInstruction* batch_norm) { in HandleBatchNormGrad() function in xla::gpu::__anondf8c4db30111::Visitor
Dir_emitter.h102 Status HandleBatchNormGrad(HloInstruction* batch_norm) override;
Dir_emitter.cc774 Status IrEmitter::HandleBatchNormGrad(HloInstruction*) { in HandleBatchNormGrad() function in xla::gpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h91 Status HandleBatchNormGrad(const HloInstruction* batch_norm_grad) override;
Ddfs_hlo_visitor_with_default.h69 Status HandleBatchNormGrad(HloInstructionPtr hlo) override { in HandleBatchNormGrad() function
Dbatchnorm_expander.cc60 Status HandleBatchNormGrad(HloInstruction* batch_norm) override;
431 Status BatchNormExpanderVisitor::HandleBatchNormGrad( in HandleBatchNormGrad() function in xla::BatchNormExpanderVisitor
Dhlo_evaluator.h247 Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override { in HandleBatchNormGrad() function
Dhlo_verifier.h96 Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override;
Ddfs_hlo_visitor.h261 virtual Status HandleBatchNormGrad(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc406 Status HloCostAnalysis::HandleBatchNormGrad(const HloInstruction*) { in HandleBatchNormGrad() function in xla::HloCostAnalysis
Dhlo_verifier.cc743 Status ShapeVerifier::HandleBatchNormGrad(HloInstruction* batch_norm_grad) { in HandleBatchNormGrad() function in xla::ShapeVerifier
Dhlo_instruction.cc2490 return visitor->HandleBatchNormGrad(this); in Visit()