Add TopK operator
Add multipes features required from aidge#259.
-
Add TopK operator; -
Add TopK operator CPU implementation; -
Add Python binding for Fold
andUnfold
; -
Fix binding for Expand
(which was namedexpand
, not following operator naming conventions); -
Fix missing clamping on ends
input forSlice
operator; -
Fix compilation issue with NVCC @noamzerah @bhalimi; -
Add ONNX TopK
import/export; -
Test passes vs ONNX Runtime ( test_onnx_nodes_import_forward.py
).
Edited by Olivier BICHLER
Merge request reports
Activity
Filter activity
changed milestone to %aidge v0.6.0
added Feature 🚀 label
assigned to @olivierbichler
added 1 commit
- 570068b4 - Added binding for Fold and Unfold (fixed case issue with Expand)
added 4 commits
-
bc09c6ac...2f7af8db - 3 commits from branch
dev
- 9c1832f9 - Merge branch 'dev' into topk
-
bc09c6ac...2f7af8db - 3 commits from branch
Please register or sign in to reply