Home
last modified time | relevance | path

Searched refs:GpuGetErrorString (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dwhere_op_gpu.cu.h159 GpuGetErrorString(first_success));
178 temp_storage_bytes, ", status: ", GpuGetErrorString(second_success));
296 GpuGetErrorString(first_success));
316 GpuGetErrorString(second_success));
H A Dhistogram_op_gpu.cu.cc85 GpuGetErrorString(err), "."); in Compute()
109 "Could not launch HistogramRange: ", GpuGetErrorString(err), "."); in Compute()
H A Dbincount_op_gpu.cu.cc86 GpuGetErrorString(err), "."); in Compute()
109 "Could not launch HistogramEven: ", GpuGetErrorString(err), "."); in Compute()
H A Dsparse_fill_empty_rows_op_gpu.cu.cc559 GpuGetErrorString(gpuprim_status)); in operator ()()
579 temp_storage_bytes, ", status: ", GpuGetErrorString(gpuprim_status)); in operator ()()
H A Dreduction_gpu_kernels.cu.h702 errors::Internal("CUB reduce error ", GpuGetErrorString(success)));
746 GpuGetErrorString(success)));
976 GpuGetErrorString(success)));
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dgpu_kernel_helper.h75 __host__ __device__ inline const char* GpuGetErrorString(cudaError_t error) { in GpuGetErrorString() function
80 inline const char* GpuGetErrorString(hipError_t error) {
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dkernels_gpu.cu.cc92 GpuGetErrorString(first_success)); in operator ()()
116 temp_storage_bytes, ", status: ", GpuGetErrorString(second_success)); in operator ()()