chainer.backend.ChainerxDevice

class chainer.backend.ChainerxDevice(device)

Device for ChainerX backend

Methods

__enter__()

A dummy definition that simply raises RuntimeError.

chainer.using_device() should be used instead.

__exit__(exc_type, exc_value, traceback)

A dummy definition that should never be called.

create_context()

Returns a context manager in which the device is made current.

See also

chainer.using_device() calls this method internally.

static from_array(array)

static from_fallback_device(device)

send(arrays)

Transfers given arrays to the device.

Parameters

arrays – Array or arrays of NumPy, CuPy, or ChainerX.

Returns

Transferred arrays.

send_array(array)

use()

Makes the device current in the current thread.

__eq__(other)

Return self==value.

__ne__(other)

Return self!=value.

Attributes

fallback_device

supported_array_types

Array types supported by the device.

Returns

tuple of array types which the device’s module functions can handle.

xp

Array module corresponding to the device.