qonnx.transformation.extend_partition (module)
- class qonnx.transformation.extend_partition.ExtendPartition(extend_index)
Bases:
qonnx.transformation.base.TransformationExtends GenericPartition type nodes by inserting the graph pointed to by the model attribute. Argument 0: extend_index * List that contains the node indices of the GenericPartition nodes
- apply(model)