Add TopK operator
Add multipes features required from aidge#259 (closed).
-
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