Searched refs:src_gpu_tensor (Results 1 – 2 of 2) sorted by relevance
424 const Tensor* src_gpu_tensor, in CopyGPUTensorToSameGPU() argument430 Status s = PrepareCopy(gpu_device, device_context, *src_gpu_tensor, in CopyGPUTensorToSameGPU()437 const int64 total_bytes = src_gpu_tensor->TotalBytes(); in CopyGPUTensorToSameGPU()439 void* src_ptr = GetBase(src_gpu_tensor); in CopyGPUTensorToSameGPU()
105 const Tensor* src_gpu_tensor,