Device

class omni.graph.core.Device(device_name: str)

Bases: object

Device type for memory location of the data types

Methods

__init__(device_name)

Initialize the device with the given name

Attributes

cpu	Is the device a CPU?
cuda	Is the device a GPU programmed with CUDA?

__init__(device_name: str)

Initialize the device with the given name

property cpu: bool

Is the device a CPU?

Type

bool

property cuda: bool

Is the device a GPU programmed with CUDA?

Type

bool