Skip to content
Snippets Groups Projects

Add TopK operator

Merged Olivier BICHLER requested to merge topk into dev

Add multipes features required from aidge#259.

  • Add TopK operator;
  • Add TopK operator CPU implementation;
  • Add Python binding for Fold and Unfold;
  • Fix binding for Expand (which was named expand, not following operator naming conventions);
  • Fix missing clamping on ends input for Slice 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

Loading
Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
Please register or sign in to reply
Loading