Fork me on GitHub

src/arraymancer/nn_primitives/backend/cudnn

Search:
Group by:
  Source Edit

Lets

cudnnHandle0 = initCudnnHandle()
  Source Edit

Procs

proc newCudnn4DTensorDesc[T: SomeFloat](t: CudaTensor[T]): cudnnTensorDescriptor_t {.
    inline, noinit.}
  Source Edit

Templates

template asCudnnType[T: SomeFloat](typ: typedesc[T]): cudnnDataType_t
  Source Edit

Made with Nim. Generated: 2024-04-17 08:45:57 UTC
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood