Home
last modified time | relevance | path

Searched refs:dst_gpu_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_util.cc471 Tensor* dst_gpu_tensor, in CopyGPUTensorToSameGPU() argument
477 dst_gpu_tensor, &dev_info, &send_stream); in CopyGPUTensorToSameGPU()
487 void* dst_ptr = GetBase(dst_gpu_tensor); in CopyGPUTensorToSameGPU()
H A Dgpu_util.h106 Tensor* dst_gpu_tensor,