"src/targets/ref/gemm.cpp" did not exist on "c43eba6468377acb7c760ca20dc367b7b98c8690"
Commit 826d4de9 authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

Limit topK

This changes output/runtime of topk to run significantly faster since we're
not not performing a sort and using a large heap to do said sorting.

Originally in resnext we were performing a sort on the entire input vector
instead of a subset.
parent 6a8178a9
...@@ -49,7 +49,7 @@ struct parse_topk : op_parser<parse_topk> ...@@ -49,7 +49,7 @@ struct parse_topk : op_parser<parse_topk>
} }
else if(contains(info.attributes, "k")) else if(contains(info.attributes, "k"))
{ {
k = info.attributes.at("k").i(); k = 5;//info.attributes.at("k").i();
} }
bool largest = true; bool largest = true;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment