| 2390 | fbb_.AddElement<int32_t>(8, axis, 0); |
| 2391 | } |
| 2392 | void add_softmaxThreshold(int32_t softmaxThreshold) { |
| 2393 | fbb_.AddElement<int32_t>(10, softmaxThreshold, 0); |
| 2394 | } |
| 2395 | explicit ArgMaxBuilder(flatbuffers::FlatBufferBuilder &_fbb) |
| 2396 | : fbb_(_fbb) { |
| 2397 | start_ = fbb_.StartTable(); |