Searched refs:kAttrLabelSwitchList (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | label_switch.cc | 39 if (!AnfAlgo::HasNodeAttr(kAttrLabelSwitchList, cnode)) { in Init()
44 label_list_ = GetValue<std::vector<uint32_t>>(primitive->GetAttr(kAttrLabelSwitchList)); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/debug/rdr/ |
| D | stream_exec_order_recorder.h | 77 if (AnfAlgo::HasNodeAttr(kAttrLabelSwitchList, cur_cnode_ptr)) { in StreamExecOrderRecorder()
78 …auto label_list = AnfAlgo::GetNodeAttr<std::vector<uint32_t>>(cur_cnode_ptr, kAttrLabelSwitchList); in StreamExecOrderRecorder()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_label_assign.cc | 63 …AnfAlgo::SetNodeAttr(kAttrLabelSwitchList, MakeValue<std::vector<uint32_t>>(label_list), node.get(… in UpdateLabelSwitch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_auto_monad.cc | 130 if (AnfAlgo::HasNodeAttr(kAttrLabelSwitchList, cnode)) { in DumpExecuteOrder()
131 auto labels = AnfAlgo::GetNodeAttr<std::vector<uint32_t>>(cnode, kAttrLabelSwitchList); in DumpExecuteOrder()
1368 AnfAlgo::SetNodeAttr(kAttrLabelSwitchList, label_list, cnode); in LabelSwitch()
1484 auto label_list = AnfAlgo::GetNodeAttr<std::vector<uint32_t>>(node, kAttrLabelSwitchList); in HandleLabelSwitch()
1502 AnfAlgo::SetNodeAttr(kAttrLabelSwitchList, MakeValue(new_labels), node); in HandleLabelSwitch()
1827 auto labels = AnfAlgo::GetNodeAttr<std::vector<uint32_t>>(node, kAttrLabelSwitchList); in EraseLabel()
|
| D | kernel_graph.cc | 1012 if (AnfAlgo::HasNodeAttr(kAttrLabelSwitchList, cur_cnode_ptr)) { in PrintGraphExecuteOrder()
1013 …auto label_list = AnfAlgo::GetNodeAttr<std::vector<uint32_t>>(cur_cnode_ptr, kAttrLabelSwitchList); in PrintGraphExecuteOrder()
|
| D | ascend_session.cc | 1434 …o label_list = AnfAlgo::GetNodeAttr<std::vector<uint32_t>>(kernel_cnodes[i], kAttrLabelSwitchList); in InitMemReuseExecOrder()
|
| D | anf_runtime_algorithm.cc | 1483 auto label_list = AnfAlgo::GetNodeAttr<std::vector<uint32_t>>(cnode, kAttrLabelSwitchList); in IsLabelIndexInNode()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 403 constexpr auto kAttrLabelSwitchList = "label_switch_list"; variable
|