decent_array.types
-
class decent_array.types.DTypes(*values)
Bases: Enum
mypyc filler docstring
-
BOOL: str = 'bool'
-
COMPLEX128: str = 'complex128'
-
COMPLEX64: str = 'complex64'
-
FLOAT16: str = 'float16'
-
FLOAT32: str = 'float32'
-
FLOAT64: str = 'float64'
-
INT16: str = 'int16'
-
INT32: str = 'int32'
-
INT64: str = 'int64'
-
INT8: str = 'int8'
-
UINT16: str = 'uint16'
-
UINT32: str = 'uint32'
-
UINT64: str = 'uint64'
-
UINT8: str = 'uint8'
-
class decent_array.types.SupportedDevices(*values)
Bases: Enum
mypyc filler docstring
-
CPU: str = 'cpu'
-
GPU: str = 'gpu'
-
MPS: str = 'mps'
-
class decent_array.types.SupportedFrameworks(*values)
Bases: Enum
mypyc filler docstring
-
JAX: str = 'jax'
-
NUMPY: str = 'numpy'
-
PYTORCH: str = 'pytorch'
-
TENSORFLOW: str = 'tensorflow'