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>
}
else if(contains(info.attributes, "k"))
{
k = info.attributes.at("k").i();
k = 5;//info.attributes.at("k").i();
}
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