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
..
onnx_import Loading commit data...
ops Loading commit data...
passes Loading commit data...
runtime Loading commit data...
types Loading commit data...
axis_set.cpp Loading commit data...
axis_set.hpp Loading commit data...
axis_vector.cpp Loading commit data...
axis_vector.hpp Loading commit data...
coordinate.cpp Loading commit data...
coordinate.hpp Loading commit data...
coordinate_diff.cpp Loading commit data...
coordinate_diff.hpp Loading commit data...
function.cpp Loading commit data...
function.hpp Loading commit data...
node.cpp Loading commit data...
node.hpp Loading commit data...
node_vector.cpp Loading commit data...
node_vector.hpp Loading commit data...
pyngraph.cpp Loading commit data...
serializer.cpp Loading commit data...
serializer.hpp Loading commit data...
shape.cpp Loading commit data...
shape.hpp Loading commit data...
strides.cpp Loading commit data...
strides.hpp Loading commit data...
util.cpp Loading commit data...
util.hpp Loading commit data...
util.py Loading commit data...