Skip to content

N
ngraph
Switch branch/tag
History Find file
  • Robert Kimball's avatar
    Performance optimize reference TopK op (#3552) · 78c145e6
    Robert Kimball authored 
    * wip

* r50 test

* update test

* R50 topk calls

* Faster topk

* Finish topk implementation

* cleanup

* checkpoint

* new test working

* more unit test

* style

* wip

* fix tests

* add needed header

* Add new TopK test to plaid manifest

* Fix windows build error:
    78c145e6
Name
Last commit
 Last update
..
all.cpp Loading commit data...
any.cpp Loading commit data...
avg_pool.cpp Loading commit data...
batch_mat_mul.cpp Loading commit data...
batch_norm.cpp Loading commit data...
binary_elementwise.cpp Loading commit data...
broadcast.cpp Loading commit data...
clamp.cpp Loading commit data...
concat.cpp Loading commit data...
constant.cpp Loading commit data...
convert.cpp Loading commit data...
convolution.cpp Loading commit data...
convolution_bias.cpp Loading commit data...
depth_to_space.cpp Loading commit data...
dequantize.cpp Loading commit data...
dot.cpp Loading commit data...
dyn_broadcast.cpp Loading commit data...
dyn_pad.cpp Loading commit data...
dyn_replace_slice.cpp Loading commit data...
dyn_reshape.cpp Loading commit data...
dyn_slice.cpp Loading commit data...
elu.cpp Loading commit data...
embedding_lookup.cpp Loading commit data...
fake_quantize.cpp Loading commit data...
gather.cpp Loading commit data...
gather_nd.cpp Loading commit data...
gemm.cpp Loading commit data...
get_output_element.cpp Loading commit data...
grn.cpp Loading commit data...
group_convolution.cpp Loading commit data...
group_convolution_transpose.cpp Loading commit data...
gru_cell.cpp Loading commit data...
hard_sigmoid.cpp Loading commit data...
index_reduction.cpp Loading commit data...
lrn.cpp Loading commit data...
lstm_cell.cpp Loading commit data...
max_pool.cpp Loading commit data...
mvn.cpp Loading commit data...
normalize.cpp Loading commit data...
one_hot.cpp Loading commit data...
pad.cpp Loading commit data...
parameter.cpp Loading commit data...
prelu.cpp Loading commit data...
proposal.cpp Loading commit data...
quantize.cpp Loading commit data...
quantized_convolution.cpp Loading commit data...
quantized_dot.cpp Loading commit data...
range.cpp Loading commit data...
replace_slice.cpp Loading commit data...
reshape.cpp Loading commit data...
reverse.cpp Loading commit data...
reverse_sequence.cpp Loading commit data...
rnn_cell.cpp Loading commit data...
scale_shift.cpp Loading commit data...
scatter_add.cpp Loading commit data...
scatter_nd.cpp Loading commit data...
select.cpp Loading commit data...
shape_of.cpp Loading commit data...
shuffle_channels.cpp Loading commit data...
slice.cpp Loading commit data...
space_to_depth.cpp Loading commit data...
split.cpp Loading commit data...
squared_difference.cpp Loading commit data...
squeeze.cpp Loading commit data...
sum.cpp Loading commit data...
tile.cpp Loading commit data...
top_k.cpp Loading commit data...
transpose.cpp Loading commit data...
unary_elementwise.cpp Loading commit data...
unsqueeze.cpp Loading commit data...