Home
last modified time | relevance | path

Searched refs:kBNTrainingUpdateOpName (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Drefresh_parameter_format.cc61 if (node_name == kBNTrainingUpdateOpName) { in Run()
Dfused_batch_norm_fusion.cc116 NewValueNode(std::make_shared<Primitive>(kBNTrainingUpdateOpName)), in GetBNTrainingUpdateInputs()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dbnupdate_eltwise_fusion_pass.cc36 if (bnupdate->isa<CNode>() && AnfAlgo::GetCNodeName(bnupdate) == kBNTrainingUpdateOpName && in MatchBnupdateDoubleOutputEltwise()
Dbnupdate_eltwise_eltwise_fusion_pass.cc46 if (bnupdate->isa<CNode>() && AnfAlgo::GetCNodeName(bnupdate) == kBNTrainingUpdateOpName && in MatchBnupdateAddRelu()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dbn_split.cc81 NewValueNode(std::make_shared<Primitive>(kBNTrainingUpdateOpName))}; in CreateOutputsOfBNTrainingUpdate()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_adapter.cc349 … {kBNTrainingUpdateOpName, "bn_update"}, in GetNodeFusionType()
Dtbe_kernel_build.cc1362 … {kBNTrainingUpdateOpName, "bn_update"}, in GetNodeFusionType()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h80 constexpr auto kBNTrainingUpdateOpName = "BNTrainingUpdate"; variable