chainer.backends.cuda.get_max_workspace_size

chainer.backends.cuda.get_max_workspace_size()[source]

Gets the workspace size for cuDNN.

Check “cuDNN Library User Guide” for detail.

Returns

The workspace size for cuDNN.

Return type

int