Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dsingle_batch_norm_fission.cc31 auto bn_cnode = bn->cast<CNodePtr>(); in CreateBNTrainingReduce() local
56 auto bn_cnode = bn->cast<CNodePtr>(); in CreateBNTrainingUpdateV3() local
Dbatch_norm_bert_fission.cc65 auto bn_cnode = bn->cast<CNodePtr>(); in CreateBNTrainingReduce() local
91 auto bn_cnode = bn->cast<CNodePtr>(); in CreateBNTrainingUpdateV2() local
Dbn_split.cc38 bool CreateOutputsOfBNTrainingReduce(const FuncGraphPtr &graph, const CNodePtr &bn_cnode, in CreateOutputsOfBNTrainingReduce()
70 AnfNodePtr CreateOutputsOfBNTrainingUpdate(const FuncGraphPtr &graph, const CNodePtr &bn_cnode, in CreateOutputsOfBNTrainingUpdate()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dsqueeze_fusion.cc68 auto bn_cnode = bn_node->cast<CNodePtr>(); in Process() local
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dbatch_norm_bert_fission_test.cc55 auto bn_cnode = bn->cast<CNodePtr>(); in TEST_F() local