Searched refs:device_ptr (Results 1 – 7 of 7) sorted by relevance
| /external/llvm-project/openmp/libomptarget/src/ |
| D | api.cpp | 64 EXTERN void omp_target_free(void *device_ptr, int device_num) { in omp_target_free() argument
66 device_num, DPxPTR(device_ptr)); in omp_target_free()
68 if (!device_ptr) { in omp_target_free()
74 free(device_ptr); in omp_target_free()
84 PM->Devices[device_num].deleteData(device_ptr); in omp_target_free()
245 EXTERN int omp_target_associate_ptr(void *host_ptr, void *device_ptr, in omp_target_associate_ptr() argument
249 DPxPTR(host_ptr), DPxPTR(device_ptr), size, device_offset, device_num); in omp_target_associate_ptr()
251 if (!host_ptr || !device_ptr || size <= 0) { in omp_target_associate_ptr()
267 void *device_addr = (void *)((uint64_t)device_ptr + (uint64_t)device_offset); in omp_target_associate_ptr()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_mem_allocator.h | 65 se::DeviceMemoryBase device_ptr(ptr); in Free()
66 stream_exec_->Deallocate(&device_ptr); in Free()
|
| /external/llvm-project/openmp/libomptarget/include/ |
| D | omptarget.h | 142 void omp_target_free(void *device_ptr, int device_num);
150 int omp_target_associate_ptr(void *host_ptr, void *device_ptr, size_t size,
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_client.h | 245 void* device_ptr, const Shape& shape, PjRtDevice* device,
|
| D | pjrt_stream_executor_client.h | 201 void* device_ptr, const Shape& shape, PjRtDevice* device,
|
| D | pjrt_stream_executor_client.cc | 959 void* device_ptr, const Shape& shape, PjRtDevice* device, in CreateViewOfDeviceBuffer() argument
961 se::DeviceMemoryBase buffer(device_ptr, ShapeUtil::ByteSizeOf(shape)); in CreateViewOfDeviceBuffer()
|
| /external/llvm-project/openmp/runtime/src/ |
| D | kmp_ftn_entry.h | 987 void FTN_STDCALL FTN_TARGET_FREE(void *device_ptr, int device_num) {} in FTN_TARGET_FREE() argument
1005 int FTN_STDCALL FTN_TARGET_ASSOCIATE_PTR(void *host_ptr, void *device_ptr, in FTN_TARGET_ASSOCIATE_PTR() argument
|