Factory methods for cuDNN types.
| 492 | |
| 493 | // Factory methods for cuDNN types. |
| 494 | TensorDescriptor CreateTensorDescriptor() { |
| 495 | cudnnTensorDescriptor_t result; |
| 496 | CHECK_CUDNN_OK(cudnnCreateTensorDescriptor(&result)); |
| 497 | return TensorDescriptor(result); |
| 498 | } |
| 499 | #if CUDNN_VERSION >= 7201 |
| 500 | RNNDataDescriptor CreateRNNDataDescriptor() { |
| 501 | cudnnRNNDataDescriptor_t result; |
no test coverage detected