Home
last modified time | relevance | path

Searched refs:CopyGPUTensorToSameGPU (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_util_platform_specific.cc46 GPUUtil::CopyGPUTensorToSameGPU(device, this, input_tensor, output_tensor, in CopyTensorInSameDevice()
H A Dgpu_util.h103 static void CopyGPUTensorToSameGPU(Device* gpu_device,
H A Dgpu_util.cc468 void GPUUtil::CopyGPUTensorToSameGPU(Device* gpu_device, in CopyGPUTensorToSameGPU() function in tensorflow::GPUUtil
H A Dgpu_device.cc895 GPUUtil::CopyGPUTensorToSameGPU(static_cast<Device*>(this), device_context, in CopyTensorInSameDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddebug_ops.h93 GPUUtil::CopyGPUTensorToSameGPU( in Compute()