Searched refs:workspaces_ (Results 1 – 7 of 7) sorted by relevance
61 (void)launch_info_.workspaces_.emplace_back(std::make_shared<Address>()); in Init()215 …to ret = device_contexts_[0]->LaunchKernel(kernel_, launch_info_.inputs_, launch_info_.workspaces_, in OnMemoryAllocFinish()394 MS_EXCEPTION_IF_NULL(launch_info_.workspaces_[i]); in PreLaunchKernel()395 launch_info_.workspaces_[i]->addr = workspace_device_tensors_[i]->GetMutablePtr(); in PreLaunchKernel()396 launch_info_.workspaces_[i]->size = workspace_device_tensors_[i]->GetSize(); in PreLaunchKernel()
170 …t = device_contexts_[0]->LaunchKernel(data_kernel_, launch_info_.inputs_, launch_info_.workspaces_, in OnMemoryAllocFinish()
36 AddressPtrList *workspaces_; member
49 mem_info_stream_ << MemInfo2String("kernel_workspaces", mem_info.workspaces_); in SaveMemInfo()
56 std::vector<AddressPtr> workspaces_; variable
28 workspaces_.push_back(workspace); in AddWorkspace()33 const std::vector<AddressPtr> &OptimizerInfo::workspaces() const { return workspaces_; } in workspaces()
174 std::vector<AddressPtr> workspaces_; member