Searched refs:gpu_ptr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator.cc | 44 se::DeviceMemory<int64> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in CheckMask() local
47 Status result = exec->SynchronousMemcpyD2H(gpu_ptr, MASK_BYTES, tmp); in CheckMask()
66 se::DeviceMemory<int64> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in InitMask() local
67 Status result = exec->SynchronousMemcpyH2D(mask, MASK_BYTES, &gpu_ptr); in InitMask()
|
| D | gpu_util.cc | 383 DeviceMemoryBase gpu_ptr(ptr, num_bytes); in MemoryDebugString() local
385 gpu_ptr, num_bytes, &*buf.begin()); in MemoryDebugString()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | dynamic_device_mgr_test.cc | 204 Device* gpu_ptr = gpu.get(); in TEST() local
230 std::vector<Device*> removed{gpu_ptr, cpu0_ptr}; in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | metal_delegate.mm | 474 void* gpu_ptr = [input_output_buffers_[input.id] contents];
475 std::memcpy(gpu_ptr, tensor->data.f, input.shape.DimensionsProduct() * sizeof(float));
564 const void* gpu_ptr = [input_output_buffers_[output.id] contents];
565 std::memcpy(tensor->data.f, gpu_ptr, output.shape.DimensionsProduct() * sizeof(float));
|