Searched refs:workspace_tensors_ (Results 1 – 2 of 2) sorted by relevance
53 std::vector<SomasTensorPtr> workspace_tensors_; variable
485 node->workspace_tensors_.push_back(tensor); in InitSomasOutputAndWorkspaceTensors()621 << "]'s Workspace size " << pre_somas_node->workspace_tensors_.size(); in InitAtomicCleanInputs()623 auto input_somas_tensor = pre_somas_node->workspace_tensors_[index]; in InitAtomicCleanInputs()904 auto workspaces = somas_node->workspace_tensors_; in UnReuseNodeProcess()1492 for (const auto &wk : node->workspace_tensors_) { in DumpNodes()1771 if (index >= somas_node->workspace_tensors_.size()) { in GetNodeWorkSpacePtr()1773 << somas_node->workspace_tensors_.size() << "]"; in GetNodeWorkSpacePtr()1775 auto workspace_tensor = somas_node->workspace_tensors_[index]; in GetNodeWorkSpacePtr()1811 for (const auto &workspace_tensor : node->workspace_tensors_) { in ConvertToProfilingNode()