Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc68 #define CHECK_CUDNN_OK(expr) CHECK_EQ(expr, CUDNN_STATUS_SUCCESS) macro
370 CHECK_CUDNN_OK(cudnnDestroyTensorDescriptor(descriptor)); in operator ()()
375 CHECK_CUDNN_OK(cudnnDestroyRNNDataDescriptor(descriptor)); in operator ()()
380 CHECK_CUDNN_OK(cudnnDestroyFilterDescriptor(descriptor)); in operator ()()
385 CHECK_CUDNN_OK(cudnnDestroyConvolutionDescriptor(descriptor)); in operator ()()
390 CHECK_CUDNN_OK(cudnnDestroyPoolingDescriptor(descriptor)); in operator ()()
395 CHECK_CUDNN_OK(cudnnDestroyLRNDescriptor(descriptor)); in operator ()()
401 CHECK_CUDNN_OK(cudnnDestroyActivationDescriptor(descriptor)); in operator ()()
406 CHECK_CUDNN_OK(cudnnDestroyDropoutDescriptor(descriptor)); in operator ()()
411 CHECK_CUDNN_OK(cudnnDestroyRNNDescriptor(descriptor)); in operator ()()
[all …]