Home
last modified time | relevance | path

Searched defs:gpuMemcpyDeviceToHost (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_client_gpu.cc32 #define gpuMemcpyDeviceToHost hipMemcpyDeviceToHost macro
38 #define gpuMemcpyDeviceToHost cudaMemcpyDeviceToHost macro
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorGpuHipCudaDefines.h46 #define gpuMemcpyDeviceToHost hipMemcpyDeviceToHost macro
76 #define gpuMemcpyDeviceToHost cudaMemcpyDeviceToHost macro
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcustom_call_test.cc42 #define gpuMemcpyDeviceToHost cudaMemcpyDeviceToHost macro
49 #define gpuMemcpyDeviceToHost hipMemcpyDeviceToHost macro