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
..
util Loading commit data...
abs.cpp Loading commit data...
abs.hpp Loading commit data...
acos.cpp Loading commit data...
acos.hpp Loading commit data...
add.cpp Loading commit data...
add.hpp Loading commit data...
allreduce.cpp Loading commit data...
allreduce.hpp Loading commit data...
and.cpp Loading commit data...
and.hpp Loading commit data...
argmax.cpp Loading commit data...
argmax.hpp Loading commit data...
argmin.cpp Loading commit data...
argmin.hpp Loading commit data...
asin.cpp Loading commit data...
asin.hpp Loading commit data...
atan.cpp Loading commit data...
atan.hpp Loading commit data...
avg_pool.cpp Loading commit data...
avg_pool.hpp Loading commit data...
batch_norm.cpp Loading commit data...
batch_norm.hpp Loading commit data...
broadcast.cpp Loading commit data...
broadcast.hpp Loading commit data...
ceiling.cpp Loading commit data...
ceiling.hpp Loading commit data...
concat.cpp Loading commit data...
concat.hpp Loading commit data...
constant.cpp Loading commit data...
constant.hpp Loading commit data...
convert.cpp Loading commit data...
convert.hpp Loading commit data...
convolution.cpp Loading commit data...
convolution.hpp Loading commit data...
cos.cpp Loading commit data...
cos.hpp Loading commit data...
cosh.cpp Loading commit data...
cosh.hpp Loading commit data...
divide.cpp Loading commit data...
divide.hpp Loading commit data...
dot.cpp Loading commit data...
dot.hpp Loading commit data...
equal.cpp Loading commit data...
equal.hpp Loading commit data...
exp.cpp Loading commit data...
exp.hpp Loading commit data...
floor.cpp Loading commit data...
floor.hpp Loading commit data...
function_call.cpp Loading commit data...
function_call.hpp Loading commit data...
get_output_element.cpp Loading commit data...
get_output_element.hpp Loading commit data...
greater.cpp Loading commit data...
greater.hpp Loading commit data...
greater_eq.cpp Loading commit data...
greater_eq.hpp Loading commit data...
less.cpp Loading commit data...
less.hpp Loading commit data...
less_eq.cpp Loading commit data...
less_eq.hpp Loading commit data...
log.cpp Loading commit data...
log.hpp Loading commit data...
lrn.cpp Loading commit data...
lrn.hpp Loading commit data...
max.cpp Loading commit data...
max.hpp Loading commit data...
max_pool.cpp Loading commit data...
max_pool.hpp Loading commit data...
maximum.cpp Loading commit data...
maximum.hpp Loading commit data...
min.cpp Loading commit data...
min.hpp Loading commit data...
minimum.cpp Loading commit data...
minimum.hpp Loading commit data...
multiply.cpp Loading commit data...
multiply.hpp Loading commit data...
negative.cpp Loading commit data...
negative.hpp Loading commit data...
not.cpp Loading commit data...
not.hpp Loading commit data...
not_equal.cpp Loading commit data...
not_equal.hpp Loading commit data...
one_hot.cpp Loading commit data...
one_hot.hpp Loading commit data...
op.cpp Loading commit data...
op.hpp Loading commit data...
or.cpp Loading commit data...
or.hpp Loading commit data...
pad.cpp Loading commit data...
pad.hpp Loading commit data...
parameter.cpp Loading commit data...
parameter.hpp Loading commit data...
parameter_vector.cpp Loading commit data...
parameter_vector.hpp Loading commit data...
power.cpp Loading commit data...
power.hpp Loading commit data...
product.cpp Loading commit data...
product.hpp Loading commit data...
reduce.cpp Loading commit data...
reduce.hpp Loading commit data...
regmodule_pyngraph_op.cpp Loading commit data...
regmodule_pyngraph_op.hpp Loading commit data...
relu.cpp Loading commit data...
relu.hpp Loading commit data...
replace_slice.cpp Loading commit data...
replace_slice.hpp Loading commit data...
reshape.cpp Loading commit data...
reshape.hpp Loading commit data...
reverse.cpp Loading commit data...
reverse.hpp Loading commit data...
select.cpp Loading commit data...
select.hpp Loading commit data...
sign.cpp Loading commit data...
sign.hpp Loading commit data...
sin.cpp Loading commit data...
sin.hpp Loading commit data...
sinh.cpp Loading commit data...
sinh.hpp Loading commit data...
slice.cpp Loading commit data...
slice.hpp Loading commit data...
softmax.cpp Loading commit data...
softmax.hpp Loading commit data...
sqrt.cpp Loading commit data...
sqrt.hpp Loading commit data...
subtract.cpp Loading commit data...
subtract.hpp Loading commit data...
sum.cpp Loading commit data...
sum.hpp Loading commit data...
tan.cpp Loading commit data...
tan.hpp Loading commit data...
tanh.cpp Loading commit data...
tanh.hpp Loading commit data...
topk.cpp Loading commit data...
topk.hpp Loading commit data...