Searched refs:SetWorkspaceSizeList (Results 1 – 10 of 10) sorted by relevance
43 kernel_mod_ptr->SetWorkspaceSizeList(kernel_json_info.workspaces); in AkgSetKernelMod()
34 void SetWorkspaceSizeList(const std::vector<size_t> &size_list);
44 void AkgKernelMod::SetWorkspaceSizeList(const std::vector<size_t> &size_list) { workspace_size_list… in SetWorkspaceSizeList() function in mindspore::kernel::AkgKernelMod
45 kernel_mod_ptr->SetWorkspaceSizeList(kernel_json_info.workspaces); in AkgSetKernelMod()
63 void SetWorkspaceSizeList(const std::vector<size_t> &size_list);
94 void GpuKernelMod::SetWorkspaceSizeList(const std::vector<size_t> &size_list) { workspace_size_list… in SetWorkspaceSizeList() function in mindspore::kernel::GpuKernelMod
36 …void SetWorkspaceSizeList(const std::vector<size_t> &size_list) { workspace_size_list_ = size_list… in SetWorkspaceSizeList() function
295 kernel_mod_ptr->SetWorkspaceSizeList(kernel_json_info.workspaces); in GenKernelMod()
53 void SetWorkspaceSizeList(const std::vector<size_t> &size_list);
59 void AicpuOpKernelMod::SetWorkspaceSizeList(const std::vector<size_t> &size_list) { workspace_size_… in SetWorkspaceSizeList() function in mindspore::kernel::AicpuOpKernelMod