finn.transformation.insert_topk (module)

class finn.transformation.insert_topk.InsertTopK(k=5, axis=- 1, largest=1, sorted=1)

Bases: finn.transformation.base.Transformation

Add TopK node at the network output and replace the graph output with the TopK indices.

apply(model)