Skip to content

Add TopK operator

Olivier BICHLER requested to merge topk into dev

Add multipes features required from aidge#259 (closed).

  • 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