qonnx.transformation.infer_data_layouts (module)

class qonnx.transformation.infer_data_layouts.InferDataLayouts

Bases: qonnx.transformation.base.Transformation

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

apply(model)