Searched refs:kAttrN (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | addn_fission.cc | 36 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(offset)), new_addn); in CreateNewAddn()
75 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(base_addn_inputs.size() - 1)), base_addn); in Process()
|
| D | pack_fission.cc | 38 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(offset)), new_pack); in CreateNewPack()
98 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(base_concat_inputs.size() - 1)), base_concat); in Process()
|
| D | concat_fission.cc | 43 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(offset)), new_concat); in CreateNewConcat()
115 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(base_concat_inputs.size() - 1)), base_concat); in Process()
|
| D | dynamic_rnn_grad_fission_v2.cc | 271 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(num_split_x)), lstm_x_concat); in AddLSTMInputGradNode()
282 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(num_split_x)), lstm_gage_concat); in AddLSTMInputGradNode()
349 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(kAttrNValue)), concat); in CreateHConcat()
372 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(kAttrNValue)), concat); in CreateConcat()
407 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(kAttrNValue)), concat); in CreateConcatNodeT1()
|
| D | dynamic_gru_v2_grad_fission.cc | 182 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(t_size)), concat_t_node); in AddTConcatNode()
279 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(kConcatNum)), concat); in AddHConcatNode()
353 AnfAlgo::SetNodeAttr(kAttrN, MakeValue(SizeToLong(kConcatNum)), concat_op); in CreateDgateXConcatDNode()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 383 constexpr auto kAttrN = "n"; variable
|