hubai_sdk.utils.metadata.Metadata

class documentation

class Metadata:

View In Hierarchy

Input and output tensor metadata extracted from a model.

Instance Variable input_dtypes Input element types keyed by tensor name.
Instance Variable input_shapes Input shapes keyed by tensor name.
Instance Variable output_dtypes Output element types keyed by tensor name.
Instance Variable output_shapes Output shapes keyed by tensor name.
input_dtypes: dict[str, DataType] =

Input element types keyed by tensor name.

input_shapes: dict[str, list[int]] =

Input shapes keyed by tensor name.

output_dtypes: dict[str, DataType] =

Output element types keyed by tensor name.

output_shapes: dict[str, list[int]] =

Output shapes keyed by tensor name.