Searched refs:kAttrStreamSwitchKind (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_stream_assign.cc | 947 if (!AnfAlgo::HasNodeAttr(kAttrStreamSwitchKind, cur_cnode_ptr)) { in ActiveRootGraphHcom()
952 auto kind = AnfAlgo::GetNodeAttr<uint32_t>(cur_cnode_ptr, kAttrStreamSwitchKind); in ActiveRootGraphHcom()
989 if (!AnfAlgo::HasNodeAttr(kAttrStreamSwitchKind, cur_cnode_ptr)) { in ActiveRootGraphIndependent()
994 auto kind = AnfAlgo::GetNodeAttr<uint32_t>(cur_cnode_ptr, kAttrStreamSwitchKind); in ActiveRootGraphIndependent()
1164 if (!AnfAlgo::HasNodeAttr(kAttrStreamSwitchKind, switch_ptr)) { in CheckStreamSwitch()
1168 auto kind = AnfAlgo::GetNodeAttr<uint32_t>(switch_ptr, kAttrStreamSwitchKind); in CheckStreamSwitch()
1737 if (!AnfAlgo::HasNodeAttr(kAttrStreamSwitchKind, item)) { in GetIndependentStreamSwitchStreamId()
1740 auto kind = AnfAlgo::GetNodeAttr<uint32_t>(item, kAttrStreamSwitchKind); in GetIndependentStreamSwitchStreamId()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_adjust.cc | 148 …AnfAlgo::SetNodeAttr(kAttrStreamSwitchKind, MakeValue<uint32_t>(kIndependentStreamSwitch), indepen… in InsertIndepentParallel()
170 …AnfAlgo::SetNodeAttr(kAttrStreamSwitchKind, MakeValue<uint32_t>(kFpBpStreamSwitch), fpbp_switch_ap… in InsertFpBpLoopStreamSwitch()
308 …AnfAlgo::SetNodeAttr(kAttrStreamSwitchKind, MakeValue<uint32_t>(kGetNextStreamSwitch), getnext_swi… in InsertGetNextLoopStreamSwitch()
373 …AnfAlgo::SetNodeAttr(kAttrStreamSwitchKind, MakeValue<uint32_t>(kEosStreamSwitch), eos_switch_app); in InsertEosStreamSwitch()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 368 constexpr auto kAttrStreamSwitchKind = "stream_switch_kind"; variable
|