Cross Reference: /external/tensorflow/tensorflow/core/common_runtime/gpu/gpu

Lines Matching refs:gpu_device

259 void GPUUtil::CopyGPUTensorToCPU(Device* gpu_device,
266   Status s = PrepareCopy(gpu_device, device_context, *gpu_tensor, cpu_tensor,
306                                  Device* gpu_device, Tensor* gpu_tensor,
311   Status s = PrepareCopy(gpu_device, device_context, *cpu_tensor, gpu_tensor,
349 Status GPUUtil::Sync(Device* gpu_device) {
351   auto* dev_info = gpu_device->tensorflow_gpu_device_info();
358 Status GPUUtil::SyncAll(Device* gpu_device) {
360   auto* dev_info = gpu_device->tensorflow_gpu_device_info();
397 uint64 GPUUtil::Checksum(Device* gpu_device,
403   CopyGPUTensorToCPU(gpu_device, device_context, &tensor, &copy,
426 void GPUUtil::CopyGPUTensorToSameGPU(Device* gpu_device,
434   Status s = PrepareCopy(gpu_device, device_context, *src_gpu_tensor,

OpenGrok