Searched refs:src_gpu_tensor (Results 1 – 2 of 2) sorted by relevance
470 const Tensor* src_gpu_tensor, in CopyGPUTensorToSameGPU() argument476 Status s = PrepareCopy(gpu_device, device_context, *src_gpu_tensor, in CopyGPUTensorToSameGPU()483 const int64_t total_bytes = src_gpu_tensor->TotalBytes(); in CopyGPUTensorToSameGPU()485 void* src_ptr = GetBase(src_gpu_tensor); in CopyGPUTensorToSameGPU()
105 const Tensor* src_gpu_tensor,