chainer.device_resident.DeviceResidentsVisitor

class chainer.device_resident.DeviceResidentsVisitor[source]

Base class of visitors that visits device resident objects recursively.

Methods

visit_array(arr)[source]

Processes an array and returns a new one.

If the visitor does not create a new array, it can simply return the original array.

visit_device_resident(device_resident)[source]

Processes a DeviceResident instance.

visit_variable(param)[source]

Processes a Variable or a Parameter.