qonnx.transformation.insert_topk (module)
- class qonnx.transformation.insert_topk.InsertTopK(k=5, axis=- 1, largest=1, sorted=1)
Bases:
qonnx.transformation.base.TransformationAdd TopK node at the network output and replace the graph output with the TopK indices.
- apply(model)