qonnx.transformation.change_datalayout (module)

class qonnx.transformation.change_datalayout.ChangeDataLayoutQuantAvgPool2d

Bases: qonnx.transformation.base.Transformation

Replace QuantAvgPool2d with datalayout (N,C,H,W) with Transpose nodes and QuantAvgPool2dNHWC with datalayout (N,H,W,C)

apply(model)