cerebras.modelzoo.common.utils.run.utils.DeviceType#

class cerebras.modelzoo.common.utils.run.utils.DeviceType[source]#

Bases: object

Supported Devices for Running Modelzoo Scripts

Methods

devices

Valid strategies

Attributes

ANY

CPU

CSX

GPU

classmethod devices()[source]#

Valid strategies