cuda.rst

CUDA and Backend Utilities

Utilities across backends

chainer.backend

/

chainer.backend.copyto

CUDA

chainer.backends.cuda

/

Devices

chainer.backends.cuda.get_device chainer.backends.cuda.get_device_from_id chainer.backends.cuda.get_device_from_array

CuPy array allocation and copy

chainer.backends.cuda.copy chainer.backends.cuda.to_cpu chainer.backends.cuda.to_gpu

Kernel definition utilities

chainer.backends.cuda.memoize chainer.backends.cuda.clear_memo chainer.backends.cuda.elementwise chainer.backends.cuda.raw chainer.backends.cuda.reduce

CPU/GPU generic code support

chainer.backends.cuda.get_array_module

cuDNN support

chainer.backends.cuda.set_max_workspace_size chainer.backends.cuda.get_max_workspace_size

iDeep

iDeep is a module that provides NumPy-like API and DNN acceleration using MKL-DNN for Intel CPUs. See ../../tips and ../../performance for details.

chainer.backends.intel64

chainer

chainer.backends.intel64.is_ideep_available