Crate cudnn [−] [src]
Provides a safe and convenient wrapper around the CUDA cuDNN API.
This crate was developed against cuDNN v3.
Architecture
The api
folder contains all the low-level functionality. Usually there should be no
need to use these methods, which are implemented for the API struct, directly.
All the functionality should be accessible through the high-level structs exposed through
the modules in the root src
folder.
The api
folder is structured like the modules in the root folder and expose safe methods,
around the cuDNN API - including proper Rust error messages for the cuDNN status types.
The ffi.rs
file of the api
folder contains the foreign function interface of cuDNN.
Reexports
pub use ffi::*; |
Modules
utils |
Describes utility functionality for CUDA cuDNN. |
Structs
API |
Defines the Cuda cuDNN API. |
ConvolutionDescriptor |
Describes a Convolution Descriptor. |
Cudnn |
Provides a the high-level interface to CUDA's cuDNN. |
FilterDescriptor |
Describes a Filter Descriptor. |
TensorDescriptor |
Describes a TensorDescriptor. |
Enums
DataType |
Defines the available data types for the CUDA cuDNN data representation. |
Error |
Defines CUDA's cuDNN errors. |