Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_info.h58 bool WorkspaceAddrExist(size_t index) const;
Dkernel_info.cc84 bool KernelInfo::WorkspaceAddrExist(size_t index) const { in WorkspaceAddrExist() function in mindspore::device::KernelInfo
Dkernel_runtime.cc1110 if (AnfAlgo::WorkspaceAddrExist(node, index)) { in AssignWorkSpaceMem()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.h193 static bool WorkspaceAddrExist(const AnfNodePtr &node, size_t output_idx);
Danf_runtime_algorithm.cc1095 bool AnfRuntimeAlgorithm::WorkspaceAddrExist(const AnfNodePtr &node, size_t output_idx) { in WorkspaceAddrExist() function in mindspore::session::AnfRuntimeAlgorithm
1100 return kernel_info->WorkspaceAddrExist(output_idx); in WorkspaceAddrExist()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/
Dsomas.cc476 if (AnfAlgo::WorkspaceAddrExist(kernel, IntToSize(index))) { in InitSomasOutputAndWorkspaceTensors()