Searched refs:ObtainKernelWorkspaces (Results 1 – 4 of 4) sorted by relevance
66 std::vector<kernel::AddressPtr> LaunchKernel::ObtainKernelWorkspaces(const std::vector<size_t> &wor… in ObtainKernelWorkspaces() function in mindspore::device::LaunchKernel84 auto kernel_workspaces = ObtainKernelWorkspaces(kernel_mod_->GetWorkspaceSizeList()); in LaunchSingleKernel()
54 …std::vector<kernel::AddressPtr> ObtainKernelWorkspaces(const std::vector<size_t> &workspaces_list);
64 auto kernel_workspaces = ObtainKernelWorkspaces(kernel_mod_->GetWorkspaceSizeList()); in LaunchOpKernel()
66 auto kernel_workspace = ObtainKernelWorkspaces(kernel_mod_->GetWorkspaceSizeList()); in LaunchOpKernel()