Searched refs:kAttrTrueBranchStream (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | stream_switch.cc | 48 if (!AnfAlgo::HasNodeAttr(kAttrTrueBranchStream, anf_node->cast<CNodePtr>())) { in Init()
51 true_stream_index_ = GetValue<uint32_t>(primitive->GetAttr(kAttrTrueBranchStream)); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_stream_assign.cc | 958 auto true_stream_id = AnfAlgo::GetNodeAttr<uint32_t>(cur_cnode_ptr, kAttrTrueBranchStream); in ActiveRootGraphHcom()
1002 …AnfAlgo::SetNodeAttr(kAttrTrueBranchStream, MakeValue<uint32_t>(first_independent_stream), cur_cno… in ActiveRootGraphIndependent()
1135 if (AnfAlgo::HasNodeAttr(kAttrTrueBranchStream, cur_cnode_ptr)) { in GetProcessedStream()
1136 auto true_stream_id = AnfAlgo::GetNodeAttr<uint32_t>(cur_cnode_ptr, kAttrTrueBranchStream); in GetProcessedStream()
2168 auto true_stream_id = AnfAlgo::GetNodeAttr<uint32_t>(node_ptr, kAttrTrueBranchStream); in GetStreamSwitchStreamRelation()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_adjust.cc | 169 …AnfAlgo::SetNodeAttr(kAttrTrueBranchStream, MakeValue<uint32_t>(*fpbp_stream_id), fpbp_switch_app); in InsertFpBpLoopStreamSwitch()
307 …AnfAlgo::SetNodeAttr(kAttrTrueBranchStream, MakeValue<uint32_t>(*getnext_stream_id), getnext_switc… in InsertGetNextLoopStreamSwitch()
372 AnfAlgo::SetNodeAttr(kAttrTrueBranchStream, MakeValue<uint32_t>(*eos_stream_id), eos_switch_app); in InsertEosStreamSwitch()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 367 constexpr auto kAttrTrueBranchStream = "true_branch_stream"; variable
|