Searched refs:workspace_indexs (Results 1 – 1 of 1) sorted by relevance
148 …auto workspace_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorkspaceIn… in CalCleanZerosSize() local150 for (const auto &index : workspace_indexs) { in CalCleanZerosSize()272 std::vector<size_t> workspace_indexs = {}; in IsAtomicNode() local276 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()