finn.transformation.infer_data_layouts (module)

class finn.transformation.infer_data_layouts.InferDataLayouts

Bases: finn.transformation.base.Transformation

Try to infer data layout annotations info for all input/intermediate/output tensors based on inputs and node type.

apply(model)