Searched refs:kAttrNumSplit (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | splitv_fission.cc | 35 if (!AnfAlgo::HasNodeAttr(kAttrNumSplit, cnode)) { in Process()
38 auto num_split = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrNumSplit); in Process()
|
| D | dynamic_rnn_grad_fission_v2.cc | 91 … AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue(static_cast<int64_t>(kAttrNumSplitValue)), split_v); in CreateTLoopNode()
112 AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue(SizeToLong(num_split_x)), lstm_split); in CreateLSTMSPlitV()
309 AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue(SizeToLong(kAttrNumSplitValue)), split_v); in CreateSplitV()
|
| D | split_fission.cc | 45 AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue(num_split), splitv); in SetAttrForSplitVNode()
|
| D | dynamic_gru_v2_grad_fission.cc | 232 AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue(SizeToLong(kSplitVOutputNum)), split_v); in AddHSplitNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | all_to_all_unify_mindir.cc | 86 AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue<int64_t>(split_count), split_v); in CreateSplitNode()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 416 constexpr auto kAttrNumSplit = "num_split"; variable
|