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::*;

Structs

API

Defines the Cuda cuDNN API.

Cudnn

Provides a the high-level interface to CUDA's cuDNN.

ScalParams

Provides a convenient interface for cuDNN's scaling parameters alpha and beta.

TensorDescriptor

Describes a TensorDescriptor.

Enums

DataType

Defines the available data types for the CUDA cuDNN data representation.

Direction

Provides a convenient interface for forward/backwar functionality.

Error

Defines CUDA's cuDNN errors.

Traits

IScalParamsDefault

Provides correct default values for ScalParams.