qonnx.transformation.infer_datatypes (module)

class qonnx.transformation.infer_datatypes.InferDataTypes

Bases: qonnx.transformation.base.Transformation

Infer QONNX DataType info for all intermediate/output tensors based on inputs and node type.

apply(model)