Home
last modified time | relevance | path

Searched refs:kAttrLabelIndex (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_label_assign.cc30 if (AnfAlgo::HasNodeAttr(kAttrLabelIndex, node)) { in UpdateLabelGoto()
38 uint32_t goto_label_id = AnfAlgo::GetNodeAttr<uint32_t>(input, kAttrLabelIndex); in UpdateLabelGoto()
39 AnfAlgo::SetNodeAttr(kAttrLabelIndex, MakeValue<uint32_t>(goto_label_id), node.get()); in UpdateLabelGoto()
45 if (AnfAlgo::HasNodeAttr(kAttrLabelIndex, node)) { in UpdateLabelSwitch()
59 uint32_t goto_label_id = AnfAlgo::GetNodeAttr<uint32_t>(input, kAttrLabelIndex); in UpdateLabelSwitch()
86 if (node_name == kLabelSetOpName && !AnfAlgo::HasNodeAttr(kAttrLabelIndex, cnode)) { in AssignLabelForLabelSet()
87 AnfAlgo::SetNodeAttr(kAttrLabelIndex, MakeValue<uint32_t>(*label_id), node); in AssignLabelForLabelSet()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Dlabel_goto.cc35 if (!AnfAlgo::HasNodeAttr(kAttrLabelIndex, cnode)) { in Init()
40 label_ = GetValue<uint32_t>(primitive->GetAttr(kAttrLabelIndex)); in Init()
Dlabel_set.cc35 if (!AnfAlgo::HasNodeAttr(kAttrLabelIndex, cnode)) { in Init()
40 label_ = GetValue<uint32_t>(primitive->GetAttr(kAttrLabelIndex)); in Init()
/third_party/mindspore/mindspore/ccsrc/debug/rdr/
Dstream_exec_order_recorder.h73 if (AnfAlgo::HasNodeAttr(kAttrLabelIndex, cur_cnode_ptr)) { in StreamExecOrderRecorder()
74 exec_node.SetLabelId(AnfAlgo::GetNodeAttr<uint32_t>(cur_cnode_ptr, kAttrLabelIndex)); in StreamExecOrderRecorder()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_auto_monad.cc124 fout << "L" << AnfAlgo::GetNodeAttr<uint32_t>(cnode, kAttrLabelIndex) << ":\n"; in DumpExecuteOrder()
127 if (AnfAlgo::HasNodeAttr(kAttrLabelIndex, cnode)) { in DumpExecuteOrder()
128 fout << " : L" << AnfAlgo::GetNodeAttr<uint32_t>(cnode, kAttrLabelIndex); in DumpExecuteOrder()
146 auto value = kg->get_attr(kAttrLabelIndex); in GetGraphLabel()
490 kg->set_attr(kAttrLabelIndex, MakeValue(label)); in MakeGraphLabel()
1345 AnfAlgo::SetNodeAttr(kAttrLabelIndex, MakeValue(label_id), cnode); in LabelGoto()
1356 AnfAlgo::SetNodeAttr(kAttrLabelIndex, MakeValue(label_id), cnode); in LabelSet()
1474 auto label_id = AnfAlgo::GetNodeAttr<uint32_t>(node, kAttrLabelIndex); in FindMaxLabelId()
1508 auto label_id = AnfAlgo::GetNodeAttr<uint32_t>(node, kAttrLabelIndex); in HandleLabelGoto()
1514 AnfAlgo::SetNodeAttr(kAttrLabelIndex, MakeValue(++max_label_), node); in HandleLabelGoto()
[all …]
Dkernel_graph.cc97 if (AnfAlgo::HasNodeAttr(kAttrLabelIndex, left) && AnfAlgo::HasNodeAttr(kAttrLabelIndex, right)) { in IsSameLabel()
98 return AnfAlgo::GetNodeAttr<uint32_t>(left, kAttrLabelIndex) == in IsSameLabel()
99 AnfAlgo::GetNodeAttr<uint32_t>(right, kAttrLabelIndex); in IsSameLabel()
1008 if (AnfAlgo::HasNodeAttr(kAttrLabelIndex, cur_cnode_ptr)) { in PrintGraphExecuteOrder()
1009 … label id[" + std::to_string(AnfAlgo::GetNodeAttr<uint32_t>(cur_cnode_ptr, kAttrLabelIndex)) + "]"; in PrintGraphExecuteOrder()
Dascend_session.cc1398 auto label_id = AnfAlgo::GetNodeAttr<uint32_t>(kernel_cnodes[i], kAttrLabelIndex); in UnfoldRecursiveExecOrder()
1411 auto label_id = AnfAlgo::GetNodeAttr<uint32_t>(back_node, kAttrLabelIndex); in GetSubGraphExecOrder()
1447 auto label_id = AnfAlgo::GetNodeAttr<uint32_t>(kernel_cnodes[i], kAttrLabelIndex); in InitMemReuseExecOrder()
1457 auto label_id = AnfAlgo::GetNodeAttr<uint32_t>(kernel_cnodes[i], kAttrLabelIndex); in InitMemReuseExecOrder()
Danf_runtime_algorithm.cc1480 (AnfAlgo::GetNodeAttr<uint32_t>(cnode, kAttrLabelIndex) == label_index)) { in IsLabelIndexInNode()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h402 constexpr auto kAttrLabelIndex = "label_index"; variable