grain.cuda.cudnn

High-level wrapper of cudnn library

Members

Functions

isDeterministic auto isDeterministic()
isNanProp auto isNanProp(): return global cudnn option
makeCudnnTensor TensorDesc makeCudnnTensor(Tensor!(dim, T, Storage) x): convert variable to cudnn tensor discriptor object
transform void transform(Tensor!(dim, T, Storage) src, ref Tensor!(dim, T, Storage) dst, T alpha = 1, T beta = 0): copy src to dst with broadcasting

Structs

TensorDesc struct TensorDesc: cudnn data type of variable like struct

Templates

cudnnDataType template cudnnDataType(T, bool allowSameSize): convert floating point types (float, double) into cudnn enum

Meta

Source

See Source File

grain cuda modules