Searched refs:PrepareCopy (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 63 Status PrepareCopy(Device* device, const DeviceContext* ctx, const Tensor& src, in PrepareCopy() function
122 Status s = PrepareCopy(dev, device_context, tensor, nullptr, &dev_info, in SetProtoFromGPU()
195 Status s = PrepareCopy(src, send_dev_context, *input, output, &dev_info, in DeviceToDeviceCopy()
262 Status s = PrepareCopy(gpu_device, device_context, *gpu_tensor, cpu_tensor, in CopyGPUTensorToCPU()
307 Status s = PrepareCopy(gpu_device, device_context, *cpu_tensor, gpu_tensor, in CopyCPUTensorToGPU()
430 Status s = PrepareCopy(gpu_device, device_context, *src_gpu_tensor, in CopyGPUTensorToSameGPU()
|