Searched defs:kernel_workspaces (Results 1 – 13 of 13) sorted by relevance
46 AddressPtrList kernel_workspaces; in UpdateArgs() local84 AddressPtrList kernel_workspaces; in StaticShapeExecute() local
111 AddressPtrList kernel_workspaces; in UpdateArgs() local
84 auto kernel_workspaces = ObtainKernelWorkspaces(kernel_mod_->GetWorkspaceSizeList()); in LaunchSingleKernel() local
1121 … AddressPtrList *kernel_inputs, AddressPtrList *const kernel_workspaces, in GenLaunchArgs()1299 … AddressPtrList *kernel_inputs, AddressPtrList *kernel_workspaces, in AssignKernelAddress()1419 AddressPtrList kernel_workspaces; in LaunchKernel() local
64 auto kernel_workspaces = ObtainKernelWorkspaces(kernel_mod_->GetWorkspaceSizeList()); in LaunchOpKernel() local
239 AddressPtrList kernel_workspaces; in HashSwapOut() local279 AddressPtrList kernel_workspaces; in HashSwapIn() local
137 AddressPtrList kernel_workspaces; in GenDynamicKernel() local
136 const std::vector<mindspore::kernel::AddressPtr> &kernel_workspaces, in LoadKernelData()774 AddressPtrList kernel_workspaces; in LaunchKernelDynamic() local894 AddressPtrList kernel_workspaces; in RunOpLaunchKernelDynamic() local1082 … AddressPtrList *kernel_workspaces, AddressPtrList *kernel_outputs, in AllocKernelDynamicRes()1162 … AddressPtrList *kernel_workspaces, bool mock) { in AllocKernelWorkspaceDynamicRes()
208 AddressPtrList kernel_workspaces; in GenDynamicKernel() local
115 AddressPtrList kernel_workspaces; in GenDynamicKernel() local
149 AddressPtrList kernel_workspaces; in LaunchKernel() local
435 std::vector<kernel::AddressPtr> kernel_workspaces; in Run() local
2380 std::vector<AddressPtr> kernel_workspaces; in CacheAddrForKernel() local