Searched defs:cudaSuccess (Results 1 - 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_managed_allocator.cc | 27 CHECK_EQ(cudaMallocManaged(&ptr, num_bytes), cudaSuccess); local
35 CHECK_EQ(cudaFree(ptr), cudaSuccess); local
|
| /external/tensorflow/tensorflow/core/util/ |
| H A D | cuda_kernel_helper_test.cu.cc | 27 EXPECT_EQ(cudaSuccess, err) << cudaGetErrorString(err); \
34 ASSERT_EQ(cudaSuccess, err) << cudaGetErrorString(err); \
139 ASSERT_EQ(cudaSuccess, err) << cudaGetErrorString(err);
272 ASSERT_EQ(cudaMallocManaged(&failure_count, sizeof(unsigned)), cudaSuccess); local
275 ASSERT_EQ(cudaDeviceSynchronize(), cudaSuccess); local
|
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| H A D | nccl_manager.cc | 275 CHECK_EQ(cudaGetDevice(&savedDevice), cudaSuccess); local
|
Completed in 592 milliseconds