finn.transformation.infer_datatypes (module)

class finn.transformation.infer_datatypes.InferDataTypes

Bases: finn.transformation.base.Transformation

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

apply(model)