Searched refs:TF_OP_REQUIRES_CUDA_SUCCESS (Results 1 – 2 of 2) sorted by relevance
356 TF_OP_REQUIRES_CUDA_SUCCESS(context, cuda_ret); in Compute()361 TF_OP_REQUIRES_CUDA_SUCCESS( in Compute()396 TF_OP_REQUIRES_CUDA_SUCCESS( in Compute()494 TF_OP_REQUIRES_CUDA_SUCCESS( in Compute()501 TF_OP_REQUIRES_CUDA_SUCCESS( in Compute()508 TF_OP_REQUIRES_CUDA_SUCCESS(context, in Compute()510 TF_OP_REQUIRES_CUDA_SUCCESS(context, gpuEventSynchronize(copy_done)); in Compute()535 TF_OP_REQUIRES_CUDA_SUCCESS(context, cudaGetLastError()); in Compute()
87 #define TF_OP_REQUIRES_CUDA_SUCCESS(context, result) \ macro