MCPcopy Create free account
hub / github.com/alibaba/MNN / add_topK

Method add_topK

schema/current/CaffeOp_generated.h:2386–2388  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

2384 fbb_.AddElement<int32_t>(4, outMaxVal, 0);
2385 }
2386 void add_topK(int32_t topK) {
2387 fbb_.AddElement<int32_t>(6, topK, 0);
2388 }
2389 void add_axis(int32_t axis) {
2390 fbb_.AddElement<int32_t>(8, axis, 0);
2391 }

Callers 1

CreateArgMaxFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected