Searched refs:kAttrSizeSplits (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | dynamic_rnn_grad_fission_v2.cc | 85 AnfAlgo::SetNodeAttr(kAttrSizeSplits, in CreateTLoopNode()
110 AnfAlgo::SetNodeAttr(kAttrSizeSplits, MakeValue(size_split), lstm_split); in CreateLSTMSPlitV()
310 …AnfAlgo::SetNodeAttr(kAttrSizeSplits, MakeValue(std::vector<int64_t>{SizeToLong(origin_input6_shap… in CreateSplitV()
|
| D | split_fission.cc | 43 AnfAlgo::SetNodeAttr(kAttrSizeSplits, MakeValue(size_splits), splitv); in SetAttrForSplitVNode()
|
| D | dynamic_gru_v2_grad_fission.cc | 233 AnfAlgo::SetNodeAttr(kAttrSizeSplits, MakeValue(split_list), split_v); in AddHSplitNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | all_to_all_unify_mindir.cc | 87 …AnfAlgo::SetNodeAttr(kAttrSizeSplits, MakeValue(std::vector<int64_t>(split_count, shape[LongToSize… in CreateSplitNode()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 419 constexpr auto kAttrSizeSplits = "size_splits"; variable
|