MCPcopy Create free account
hub / github.com/OAID/Tengine / ArgMaxParam

Class ArgMaxParam

operator/include/operator/argmax_param.hpp:31–40  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

29namespace TEngine {
30
31struct ArgMaxParam : public NamedParam
32{
33 int axis;
34 int dimension;
35 DECLARE_PARSER_STRUCTURE(ArgMaxParam)
36 {
37 DECLARE_PARSER_ENTRY(axis);
38 DECLARE_PARSER_ENTRY(dimension);
39 }
40};
41
42} // namespace TEngine
43

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected