Home
last modified time | relevance | path

Searched defs:bn_cnode (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dsingle_batch_norm_fission.cc35 auto bn_cnode = bn->cast<CNodePtr>(); in CreateBNTrainingReduce() local
61 auto bn_cnode = bn->cast<CNodePtr>(); in CreateBNTrainingUpdateV3() local
Dbn_split.cc50 bool BnSplit::CreateOutputsOfBNTrainingReduce(const FuncGraphPtr &graph, const CNodePtr &bn_cnode, in CreateOutputsOfBNTrainingReduce()
85 …dePtr BnSplit::CreateOutputsOfBNTrainingUpdate(const FuncGraphPtr &graph, const CNodePtr &bn_cnode, in CreateOutputsOfBNTrainingUpdate()
164 …l SyncBnSplit::CreateOutputsOfBNTrainingReduce(const FuncGraphPtr &graph, const CNodePtr &bn_cnode, in CreateOutputsOfBNTrainingReduce()
205 …r SyncBnSplit::CreateOutputsOfBNTrainingUpdate(const FuncGraphPtr &graph, const CNodePtr &bn_cnode, in CreateOutputsOfBNTrainingUpdate()
Dbatch_norm_bert_fission.cc70 auto bn_cnode = bn->cast<CNodePtr>(); in CreateBNTrainingReduce() local
97 auto bn_cnode = bn->cast<CNodePtr>(); in CreateBNTrainingUpdateV2() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dsqueeze_fusion.cc73 auto bn_cnode = bn_node->cast<CNodePtr>(); in Process() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dbatch_norm_bert_fission_test.cc53 auto bn_cnode = bn->cast<CNodePtr>(); in TEST_F() local