Skip to content

N
ngraph
Switch branch/tag
History Find file
  • Sang Ik Lee's avatar
    TopK (w/ArgMax, ArgMin python wrapper) (#1560) · 3548772b
    Sang Ik Lee authored 
    * Implement TopK.

* Update python wrappers for TopK, ArgMin and ArgMax.

* Address some reviewer comments.

* Add type property check tests for TopK.
Set correct TopK behavior for K==0.

* TopK: Add 1d and 3d unit tests.

* Address more reviewer comments.

* Apply code style.
    3548772b
Name
Last commit
 Last update
..
models Loading commit data...
__init__.py Loading commit data...
test_basic.py Loading commit data...
test_convolution.py Loading commit data...
test_normalization.py Loading commit data...
test_onnx_import.py Loading commit data...
test_ops_binary.py Loading commit data...
test_ops_matmul.py Loading commit data...
test_ops_reshape.py Loading commit data...
test_ops_unary.py Loading commit data...
test_pooling.py Loading commit data...
test_reduction.py Loading commit data...
util.py Loading commit data...