pymoose.computation.types module
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 ofAesTensorType
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
) – ADType
type for the underlying encrypted tensor elements.
- 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
) – ADType
type for the tensor.
- 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.