Searched refs:dst_gpu_tensor (Results 1 – 2 of 2) sorted by relevance
425 Tensor* dst_gpu_tensor, in CopyGPUTensorToSameGPU() argument431 dst_gpu_tensor, &dev_info, &send_stream); in CopyGPUTensorToSameGPU()441 void* dst_ptr = GetBase(dst_gpu_tensor); in CopyGPUTensorToSameGPU()
106 Tensor* dst_gpu_tensor,