HomeSort by relevance Sort by last modified time
    Searched defs:cudaSuccess (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_managed_allocator.cc 27 CHECK_EQ(cudaMallocManaged(&ptr, num_bytes), cudaSuccess);
35 CHECK_EQ(cudaFree(ptr), cudaSuccess);
  /external/tensorflow/tensorflow/core/util/
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);
275 ASSERT_EQ(cudaDeviceSynchronize(), cudaSuccess);
  /external/tensorflow/tensorflow/contrib/nccl/kernels/
nccl_manager.cc 275 CHECK_EQ(cudaGetDevice(&savedDevice), cudaSuccess);

Completed in 172 milliseconds