Home
last modified time | relevance | path

Searched refs:gpuSuccess (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcustom_call_test.cc35 #define gpuSuccess cudaSuccess macro
42 #define gpuSuccess hipSuccess macro
85 ASSERT_EQ(err, gpuSuccess); in Callback_Memcpy()
/external/tensorflow/tensorflow/core/kernels/
Dhistogram_op_gpu.cu.cc82 if (err != gpuSuccess) { in Compute()
106 if (err != gpuSuccess) { in Compute()
Dwhere_op_gpu.cu.h155 if (first_success != gpuSuccess) {
174 if (second_success != gpuSuccess) {
291 if (first_success != gpuSuccess) {
312 if (second_success != gpuSuccess) {
Dbincount_op_gpu.cu.cc73 if (err != gpuSuccess) { in Compute()
96 if (err != gpuSuccess) { in Compute()
/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper.h47 #define gpuSuccess cudaSuccess macro
51 #define gpuSuccess hipSuccess macro
/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels_gpu.cu.cc87 if (first_success != gpuSuccess) { in operator ()()
111 if (second_success != gpuSuccess) { in operator ()()