pymoose.computation.types module#

PyMoose stubs of various Moose Value types.

class pymoose.computation.types.AesKeyType[source]#

Bases: pymoose.computation.types.ValueType

Moose value representing an AES key.

Used in conjunction with pymoose.decrypt() to decrypt elements of AesTensorType values.

This type may be removed in future versions of Moose.

class pymoose.computation.types.AesTensorType(dtype)[source]#

Bases: pymoose.computation.types.ValueType

Moose Value representing a tensor of AES-encrypted values.

This type may be removed in future versions of Moose.

Parameters

dtype (DType) – A DType type for the underlying encrypted tensor elements.

dtype: pymoose.computation.dtypes.DType#
class pymoose.computation.types.BytesType[source]#

Bases: pymoose.computation.types.ValueType

Moose value representing a collection of raw bytes.

class pymoose.computation.types.FloatType[source]#

Bases: pymoose.computation.types.ValueType

Moose value representing a Python float.

class pymoose.computation.types.IntType[source]#

Bases: pymoose.computation.types.ValueType

Moose value representing a Python integer.

class pymoose.computation.types.ShapeType[source]#

Bases: pymoose.computation.types.ValueType

Moose value representing a shape, i.e. Python tuple of ints.

class pymoose.computation.types.StringType[source]#

Bases: pymoose.computation.types.ValueType

Moose value representing a Python string.

class pymoose.computation.types.TensorType(dtype)[source]#

Bases: pymoose.computation.types.ValueType

Moose Value representing a Tensor.

This is the core type that is most commonly found throughout a Moose computation. In Moose this roughly corresponds to a Tensor in the “logical” dialect. During lowering, the compiler will eventually replace the type with more specific/concrete types based on its DType and placement.

Parameters

dtype (DType) – A DType type for the tensor.

dtype: pymoose.computation.dtypes.DType#
class pymoose.computation.types.UnitType[source]#

Bases: pymoose.computation.types.ValueType

The unit Moose type, similar to Python’s None.

class pymoose.computation.types.UnknownType[source]#

Bases: pymoose.computation.types.ValueType

Generic unknown type, to be filled in by the compiler.

Depreceated. Was used before DType was a first-class citizen in Moose.

class pymoose.computation.types.ValueType[source]#

Bases: object

Generic type representing a Moose Value.