D | gpu_util.cc | 116 const DeviceContext* device_context, in SetProtoFromGPU() argument 119 VLOG(1) << "SetProtoFromGPU device_context " << device_context; in SetProtoFromGPU() 122 Status s = PrepareCopy(dev, device_context, tensor, nullptr, &dev_info, in SetProtoFromGPU() 130 static_cast<const GPUDeviceContext*>(device_context) in SetProtoFromGPU() 256 const DeviceContext* device_context, in CopyGPUTensorToCPU() argument 262 Status s = PrepareCopy(gpu_device, device_context, *gpu_tensor, cpu_tensor, in CopyGPUTensorToCPU() 270 static_cast<const GPUDeviceContext*>(device_context) in CopyGPUTensorToCPU() 301 const DeviceContext* device_context, in CopyCPUTensorToGPU() argument 307 Status s = PrepareCopy(gpu_device, device_context, *cpu_tensor, gpu_tensor, in CopyCPUTensorToGPU() 315 static_cast<const GPUDeviceContext*>(device_context) in CopyCPUTensorToGPU() [all …]
|