Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc148 …auto workspace_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorkspaceIn… in CalCleanZerosSize() local
150 for (const auto &index : workspace_indexs) { in CalCleanZerosSize()
272 std::vector<size_t> workspace_indexs = {}; in IsAtomicNode() local
276 workspace_indexs.emplace_back(k); in IsAtomicNode()
280 if (!workspace_indexs.empty()) { in IsAtomicNode()
281 AnfAlgo::SetNodeAttr(kAttrAtomicWorkspaceIndexs, MakeValue(workspace_indexs), kernel_node); in IsAtomicNode()
283 return !(workspace_indexs.empty() && output_indexs.empty()); in IsAtomicNode()