Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_client_gpu.cc33 #define gpuMemcpyHostToDevice hipMemcpyHostToDevice macro
39 #define gpuMemcpyHostToDevice cudaMemcpyHostToDevice macro
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorGpuHipCudaDefines.h47 #define gpuMemcpyHostToDevice hipMemcpyHostToDevice macro
77 #define gpuMemcpyHostToDevice cudaMemcpyHostToDevice macro
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcustom_call_test.cc43 #define gpuMemcpyHostToDevice cudaMemcpyHostToDevice macro
50 #define gpuMemcpyHostToDevice hipMemcpyHostToDevice macro