| 2405 | }; |
| 2406 | |
| 2407 | inline flatbuffers::Offset<ArgMax> CreateArgMax( |
| 2408 | flatbuffers::FlatBufferBuilder &_fbb, |
| 2409 | int32_t outMaxVal = 0, |
| 2410 | int32_t topK = 0, |
| 2411 | int32_t axis = 0, |
| 2412 | int32_t softmaxThreshold = 0) { |
| 2413 | ArgMaxBuilder builder_(_fbb); |
| 2414 | builder_.add_softmaxThreshold(softmaxThreshold); |
| 2415 | builder_.add_axis(axis); |
| 2416 | builder_.add_topK(topK); |
| 2417 | builder_.add_outMaxVal(outMaxVal); |
| 2418 | return builder_.Finish(); |
| 2419 | } |
| 2420 | |
| 2421 | flatbuffers::Offset<ArgMax> CreateArgMax(flatbuffers::FlatBufferBuilder &_fbb, const ArgMaxT *_o, const flatbuffers::rehasher_function_t *_rehasher = nullptr); |
| 2422 |
no test coverage detected