chainerx.using_device

chainerx.using_device(device)

Creates a context manager to temporarily set the default device.

Parameters

device (Device or str) – Device object or device name to set as the default device during the context. See chainerx.Device.name for the specification of device names.