Module cuda

Source
Expand description

Defines Cuda Device Memory.

Hold a pointer and size of the cuda device memory. This is only a convenience wrapper to interact in a defined mamner with cudnn, which requires scrap/temporary memory for some operations, i.e. dropout.

Structsยง

CudaDeviceMemory
A pointer to memory existing on a nvidia GPU