Searched refs:GetMutableWorkspaceAddr (Results 1 – 5 of 5) sorted by relevance
57 DeviceAddressPtr GetMutableWorkspaceAddr(size_t index) const;
76 DeviceAddressPtr KernelInfo::GetMutableWorkspaceAddr(size_t index) const { in GetMutableWorkspaceAddr() function in mindspore::device::KernelInfo
707 auto device_address = AnfAlgo::GetMutableWorkspaceAddr(kernel, i); in ClearKernelWorkspaceAddress()1171 auto device_address = AnfAlgo::GetMutableWorkspaceAddr(kernel, i); in AllocKernelWorkspaceDynamicRes()1331 auto device_address = AnfAlgo::GetMutableWorkspaceAddr(kernel, i); in FreeKernelDynamicRes()
206 static DeviceAddressPtr GetMutableWorkspaceAddr(const AnfNodePtr &node, size_t index);
1152 DeviceAddressPtr AnfRuntimeAlgorithm::GetMutableWorkspaceAddr(const AnfNodePtr &node, size_t index)… in GetMutableWorkspaceAddr() function in mindspore::session::AnfRuntimeAlgorithm1156 auto addr = kernel_info->GetMutableWorkspaceAddr(index); in GetMutableWorkspaceAddr()