Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dgpu_kernel_helper_test.cu.cc30 gpuDeviceSynchronize(); \
37 gpuDeviceSynchronize(); \
172 gpuDeviceSynchronize(); in TearDown()
323 ASSERT_EQ(gpuDeviceSynchronize(), cudaSuccess); in TEST()
H A Dgpu_device_functions.h53 #define gpuDeviceSynchronize cudaDeviceSynchronize macro
68 #define gpuDeviceSynchronize hipDeviceSynchronize macro
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorGpuHipCudaDefines.h52 #define gpuDeviceSynchronize hipDeviceSynchronize macro
82 #define gpuDeviceSynchronize cudaDeviceSynchronize macro
H A DTensorGpuHipCudaUndefines.h37 #undef gpuDeviceSynchronize
/aosp_15_r20/external/eigen/test/
H A Dgpu_common.h61 gpuDeviceSynchronize(); in run_on_gpu()
79 err = gpuDeviceSynchronize(); in run_on_gpu()