Searched refs:kAttrAtomicWorkspaceIndexs (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_build_ascend.cc | 147 if (AnfAlgo::HasNodeAttr(kAttrAtomicWorkspaceIndexs, pre_node)) { in CalCleanZerosSize()
148 …workspace_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorkspaceIndexs); in CalCleanZerosSize()
281 AnfAlgo::SetNodeAttr(kAttrAtomicWorkspaceIndexs, MakeValue(workspace_indexs), kernel_node); in IsAtomicNode()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 123 if (AnfAlgo::HasNodeAttr(kAttrAtomicWorkspaceIndexs, pre_node)) { in LaunchAddrCleanKernel()
124 …workspace_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorkspaceIndexs); in LaunchAddrCleanKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 1213 if (AnfAlgo::HasNodeAttr(kAttrAtomicWorkspaceIndexs, pre_node)) { in GenAddrCleanLaunchArgs()
1215 …_workspaces_indexes = AnfAlgo::GetNodeAttr<std::vector<int>>(pre_node, kAttrAtomicWorkspaceIndexs); in GenAddrCleanLaunchArgs()
1217 …rkspaces_indexes = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorkspaceIndexs); in GenAddrCleanLaunchArgs()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 360 constexpr auto kAttrAtomicWorkspaceIndexs = "atomic_workspace_clean_indexs"; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 616 if (AnfAlgo::HasNodeAttr(kAttrAtomicWorkspaceIndexs, pre_node)) { in InitAtomicCleanInputs()
617 …workspace_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorkspaceIndexs); in InitAtomicCleanInputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 2464 if (HasNodeAttr(kAttrAtomicWorkspaceIndexs, pre_node)) { in CacheAddrForAtomicClean()
2466 …uto clean_workspaces_indexes = GetNodeAttr<std::vector<int>>(pre_node, kAttrAtomicWorkspaceIndexs); in CacheAddrForAtomicClean()
2468 … clean_workspaces_indexes = GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorkspaceIndexs); in CacheAddrForAtomicClean()
|