Searched refs:kAttrSplitDim (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | splitv_fission.cc | 42 …nc_graph, cnode, num_split, outputs_divisor_, AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrSplitDim)); in Process()
|
| D | dynamic_rnn_grad_fission_v2.cc | 90 … AnfAlgo::SetNodeAttr(kAttrSplitDim, MakeValue(static_cast<int64_t>(kAttrSplitDimValue)), split_v); in CreateTLoopNode()
111 AnfAlgo::SetNodeAttr(kAttrSplitDim, MakeValue(static_cast<int64_t>(0)), lstm_split); in CreateLSTMSPlitV()
308 AnfAlgo::SetNodeAttr(kAttrSplitDim, MakeValue(SizeToLong(0)), split_v); in CreateSplitV()
|
| D | split_fission.cc | 44 AnfAlgo::SetNodeAttr(kAttrSplitDim, MakeValue(split_dim), splitv); in SetAttrForSplitVNode()
|
| D | dynamic_gru_v2_grad_fission.cc | 231 AnfAlgo::SetNodeAttr(kAttrSplitDim, MakeValue(SizeToLong(0)), split_v); in AddHSplitNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | all_to_all_unify_mindir.cc | 61 int64_t split_dim = AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitDim); in CreateSplitNode()
85 AnfAlgo::SetNodeAttr(kAttrSplitDim, MakeValue<int64_t>(split_dim), split_v); in CreateSplitNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | split_n_optimizer.cc | 170 split_dim = AnfAlgo::GetNodeAttr<int64_t>(node, kAttrSplitDim); in NeedSkip()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 415 constexpr auto kAttrSplitDim = "split_dim"; variable
|