HomeSort by relevance Sort by last modified time
    Searched refs:HandleBatchNormGrad (Results 1 - 11 of 11) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_rewriter.cc 40 Status HandleBatchNormGrad(HloInstruction* batch_norm) override;
151 Status Visitor::HandleBatchNormGrad(HloInstruction* batch_norm) {
ir_emitter.h 97 Status HandleBatchNormGrad(HloInstruction* batch_norm) override;
ir_emitter.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
dfs_hlo_visitor_with_default.h 63 Status HandleBatchNormGrad(HloInstructionPtr hlo) override {
hlo_cost_analysis.h 84 Status HandleBatchNormGrad(const HloInstruction* batch_norm_grad) override;
hlo_verifier.h 82 Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override;
batchnorm_expander.cc 61 Status HandleBatchNormGrad(HloInstruction* batch_norm) override;
402 Status BatchNormExpanderVisitor::HandleBatchNormGrad(
dfs_hlo_visitor.h 231 virtual Status HandleBatchNormGrad(HloInstructionPtr hlo) = 0;
hlo_cost_analysis.cc 372 Status HloCostAnalysis::HandleBatchNormGrad(const HloInstruction*) {
hlo_verifier.cc 362 Status ShapeVerifier::HandleBatchNormGrad(HloInstruction* batch_norm_grad) {
    [all...]
hlo_instruction.cc     [all...]

Completed in 414 milliseconds