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
..
arithmetic_reduction.cpp Loading commit data...
arithmetic_reduction.hpp Loading commit data...
binary_elementwise_arithmetic.cpp Loading commit data...
binary_elementwise_arithmetic.hpp Loading commit data...
binary_elementwise_comparison.cpp Loading commit data...
binary_elementwise_comparison.hpp Loading commit data...
binary_elementwise_logical.cpp Loading commit data...
binary_elementwise_logical.hpp Loading commit data...
index_reduction.cpp Loading commit data...
index_reduction.hpp Loading commit data...
op_annotations.cpp Loading commit data...
op_annotations.hpp Loading commit data...
regmodule_pyngraph_op_util.cpp Loading commit data...
regmodule_pyngraph_op_util.hpp Loading commit data...
unary_elementwise_arithmetic.cpp Loading commit data...
unary_elementwise_arithmetic.hpp Loading commit data...