Skip to main content

Python class

DeviceKind

DeviceKind​

class max.graph.DeviceKind(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

source

Bases: str, Enum

A device type representation.

CPU​

CPU = 'cpu'

source

GPU​

GPU = 'gpu'

source

from_string()​

static from_string(txt)

source

Parses a device kind from its string representation.

Parameters:

txt (str)

Return type:

DeviceKind