Home
last modified time | relevance | path

Searched defs:device_ptr (Results 1 – 19 of 19) sorted by relevance

/third_party/boost/libs/compute/include/boost/compute/detail/
Ddevice_ptr.hpp61 class device_ptr class
71 device_ptr() in device_ptr() function in boost::compute::detail::device_ptr
76 device_ptr(const buffer &buffer, size_t index = 0) in device_ptr() function in boost::compute::detail::device_ptr
82 device_ptr(const device_ptr<T> &other) in device_ptr() function in boost::compute::detail::device_ptr
99 device_ptr(device_ptr<T>&& other) BOOST_NOEXCEPT in device_ptr() function in boost::compute::detail::device_ptr
/third_party/boost/boost/compute/detail/
Ddevice_ptr.hpp61 class device_ptr class
71 device_ptr() in device_ptr() function in boost::compute::detail::device_ptr
76 device_ptr(const buffer &buffer, size_t index = 0) in device_ptr() function in boost::compute::detail::device_ptr
82 device_ptr(const device_ptr<T> &other) in device_ptr() function in boost::compute::detail::device_ptr
99 device_ptr(device_ptr<T>&& other) BOOST_NOEXCEPT in device_ptr() function in boost::compute::detail::device_ptr
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_subgraph.cc319 auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(tensor, tensor.DataSize()); in Prepare() local
345 auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(tensor, tensor.DataSize()); in Prepare() local
389 …auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(trt_in_tensor_name_[i], inputs_[i].Dat… in ReSize() local
415 …auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(trt_out_tensor_name_[i], outputs_[i].D… in ReSize() local
Dtensorrt_allocator.cc36 void *device_ptr = nullptr; in MallocDeviceMem() local
107 auto device_ptr = current_cuda_tensor.data; in SyncMemInHostAndDevice() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dmemory_scheduler.cc108 auto device_ptr = mem_handler_->MallocDevice(event->mem_size); in PreCompute() local
118 auto device_ptr = mem_handler_->MallocDevice(event->mem_size); in PreCompute() local
130 auto device_ptr = mem_handler_->MallocDevice(event->mem_size); in PreCompute() local
168 auto device_ptr = mem_result_[event->key]; in PostCompute() local
Dmemory_manager.h99 void SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *stream) override { in SwapIn()
102 void SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *stream) override { in SwapOut()
Dmemory_manager.cc146 auto device_ptr = MallocMemFromMemPool(size); in MallocMemFromMemPool() local
171 void MemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_memory_manager.cc31 void GPUMemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool()
109 auto device_ptr = MallocMemFromMemPool(size); in MallocStaticMem() local
Dgpu_memory_copy_manager.cc37 DeviceMemPtr device_ptr = const_cast<DeviceMemPtr>(device_address->GetPtr()); in AddMemSwapOutTask() local
64 DeviceMemPtr device_ptr = const_cast<DeviceMemPtr>(device_address->GetPtr()); in AddMemSwapInTask() local
Dgpu_kernel_runtime.cc227 DeviceAddressPtr GPUKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, const … in CreateDeviceAddress()
232 DeviceAddressPtr GPUKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, const … in CreateDeviceAddress()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_memory_manager.cc131 void AscendMemoryManager::FreeMemFromMemPool(void *device_ptr) { in FreeMemFromMemPool()
225 void AscendMemoryManager::SwapIn(const void *host_ptr, void *device_ptr, size_t mem_size, void *str… in SwapIn()
242 void AscendMemoryManager::SwapOut(const void *device_ptr, void *host_ptr, size_t mem_size, void *st… in SwapOut()
Dascend_kernel_runtime.cc403 DeviceAddressPtr AscendKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, con… in CreateDeviceAddress()
408 DeviceAddressPtr AscendKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, con… in CreateDeviceAddress()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_memory_manager.h50 …void FreeMemFromMemPool(void *device_ptr) override { CPUMemoryPool::GetInstance().FreeTensorMem(de… in FreeMemFromMemPool()
Dcpu_kernel_runtime.cc176 DeviceAddressPtr CPUKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, const … in CreateDeviceAddress()
181 DeviceAddressPtr CPUKernelRuntime::CreateDeviceAddress(void *device_ptr, size_t device_size, const … in CreateDeviceAddress()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.cc81 auto device_ptr = mem_manager_->MallocMemFromMemPool(size); in AllocateMemory() local
102 DeviceAddressPtr CPUDeviceContext::CreateDeviceAddress(void *const device_ptr, size_t device_size, … in CreateDeviceAddress()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Dai_core_dynamic_kernel.cc161 auto device_ptr = runtime_instance->MallocMem(MemType::kDynamicMem, size, device_address_ptr); in AllocateWorkspace() local
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc171 auto device_ptr = mem_manager_->MallocMemFromMemPool(size); in AllocateMemory() local
199 DeviceAddressPtr GPUDeviceContext::CreateDeviceAddress(void *const device_ptr, size_t device_size, … in CreateDeviceAddress()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/
DDvppCommon.cc1215 uint8_t *device_ptr = nullptr; in TransferYuvDataH2D() local
/third_party/gstreamer/gstplugins_bad/sys/nvcodec/
Dcuda-converter.c726 CUdeviceptr device_ptr; member