megengine-schema.js 80 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423
  1. var $root = flatbuffers.get('megengine');
  2. $root.mgb = $root.mgb || {};
  3. $root.mgb.serialization = $root.mgb.serialization || {};
  4. $root.mgb.serialization.fbs = $root.mgb.serialization.fbs || {};
  5. $root.mgb.serialization.fbs.DTypeEnum = {
  6. Float32: 0,
  7. Uint8: 1,
  8. Int8: 2,
  9. Int16: 3,
  10. Int32: 4,
  11. IntB1: 5,
  12. IntB2: 6,
  13. IntB4: 7,
  14. Byte: 8,
  15. Float16: 9,
  16. UintB4: 10,
  17. Quantized8Asymm: 11,
  18. QuantizedS32: 12,
  19. QuantizedS8: 13,
  20. Quantized4Asymm: 14,
  21. QuantizedS4: 15,
  22. QuantizedS16: 16,
  23. BFloat16: 17,
  24. Bool: 18,
  25. Uint16: 19,
  26. QuantizedS1: 20
  27. };
  28. $root.mgb.serialization.fbs.LinearQuantizationParam = class LinearQuantizationParam {
  29. static decode(reader, position) {
  30. const $ = new $root.mgb.serialization.fbs.LinearQuantizationParam();
  31. $.scale = reader.float32_(position, 4, 0);
  32. $.zero_point = reader.uint8_(position, 6, 0);
  33. return $;
  34. }
  35. };
  36. $root.mgb.serialization.fbs.DTypeParam = class {
  37. static decode(reader, position, type) {
  38. switch (type) {
  39. case 1: return $root.mgb.serialization.fbs.LinearQuantizationParam.decode(reader, position);
  40. default: return undefined;
  41. }
  42. }
  43. };
  44. $root.mgb.serialization.fbs.DType = class DType {
  45. static decode(reader, position) {
  46. const $ = new $root.mgb.serialization.fbs.DType();
  47. $.type = reader.int8_(position, 4, 0);
  48. $.param = reader.union(position, 6, $root.mgb.serialization.fbs.DTypeParam.decode);
  49. return $;
  50. }
  51. };
  52. $root.mgb = $root.mgb || {};
  53. $root.mgb.serialization = $root.mgb.serialization || {};
  54. $root.mgb.serialization.fbs = $root.mgb.serialization.fbs || {};
  55. $root.mgb.serialization.fbs.param = $root.mgb.serialization.fbs.param || {};
  56. $root.mgb.serialization.fbs.param.ArgsortOrder = {
  57. ASCENDING: 0,
  58. DESCENDING: 1
  59. };
  60. $root.mgb.serialization.fbs.param.BNFwdMode = {
  61. TRAINING: 0,
  62. INFERENCE: 1
  63. };
  64. $root.mgb.serialization.fbs.param.BNParamDim = {
  65. DIM_11HW: 0,
  66. DIM_1CHW: 1,
  67. DIM_1C11: 2,
  68. DIM_111C: 3
  69. };
  70. $root.mgb.serialization.fbs.param.CondTakeMode = {
  71. EQ: 0,
  72. NEQ: 1,
  73. LT: 2,
  74. LEQ: 3,
  75. GT: 4,
  76. GEQ: 5
  77. };
  78. $root.mgb.serialization.fbs.param.Conv3DBiasNonlineMode = {
  79. IDENTITY: 0,
  80. RELU: 1,
  81. SIGMOID: 2
  82. };
  83. $root.mgb.serialization.fbs.param.ConvBiasV0NonlineMode = {
  84. IDENTITY: 0,
  85. RELU: 1,
  86. SIGMOID: 2,
  87. H_SWISH: 3
  88. };
  89. $root.mgb.serialization.fbs.param.ConvPoolingMethod = {
  90. WITH_TEXTURE_OBJ: 0,
  91. WITH_SHARED_MEM: 1
  92. };
  93. $root.mgb.serialization.fbs.param.ConvPoolingNonlineMode = {
  94. IDENTITY: 0,
  95. RELU: 1,
  96. SIGMOID: 2
  97. };
  98. $root.mgb.serialization.fbs.param.ConvPoolingPoolMode = {
  99. AVERAGE: 0,
  100. MAX_: 1
  101. };
  102. $root.mgb.serialization.fbs.param.ConvolutionFormat = {
  103. NCHW: 0,
  104. NHWC: 1,
  105. NHWCD4: 2,
  106. NCHW4: 3,
  107. NCHW8: 4,
  108. NCHW32: 5,
  109. NCHW88: 6,
  110. NCHW44: 7,
  111. NCHW44_DOT: 8,
  112. NCHW4_NCHW32: 9,
  113. NCHW32_NCHW4: 10,
  114. NCHW4_NCHW: 11,
  115. NHWC_NCHW: 12,
  116. NHWC_NCHW4_IC_SMALL: 13,
  117. NCHW_NCHW4_IC_SMALL: 14,
  118. CHWN4: 15,
  119. NCHW64: 16,
  120. NCHW4_NHWC: 17
  121. };
  122. $root.mgb.serialization.fbs.param.Convolution3DDataType = {
  123. FLOAT: 0,
  124. FLOAT_IO16xC32: 1
  125. };
  126. $root.mgb.serialization.fbs.param.Convolution3DFormat = {
  127. NCDHW: 0,
  128. NDHWC: 1
  129. };
  130. $root.mgb.serialization.fbs.param.Convolution3DMode = {
  131. CROSS_CORRELATION: 0,
  132. CONVOLUTION: 1
  133. };
  134. $root.mgb.serialization.fbs.param.Convolution3DSparse = {
  135. DENSE: 0,
  136. GROUP: 1
  137. };
  138. $root.mgb.serialization.fbs.param.ConvolutionV0DataType = {
  139. FLOAT: 0,
  140. INT8x8x16: 1,
  141. INT8x8x32: 2,
  142. FLOAT_IO16xC32: 3,
  143. QUINT8x8x32: 4,
  144. INT8x8xX: 5,
  145. QUINT4x4x32: 6
  146. };
  147. $root.mgb.serialization.fbs.param.ConvolutionV0Format = {
  148. NCHW: 0,
  149. NHWC: 1,
  150. NHWCD4: 2,
  151. NCHW4: 3,
  152. NCHW8: 4,
  153. NCHW32: 5,
  154. NCHW88: 6,
  155. NCHW44: 7,
  156. NCHW44_DOT: 8,
  157. NCHW_WINOGRAD: 9,
  158. NCHW88_WINOGRAD: 10,
  159. NCHW44_WINOGRAD: 11,
  160. NCHW4_NCHW32: 12,
  161. NCHW32_NCHW4: 13,
  162. NCHW4_NCHW: 14,
  163. NHWC_NCHW: 15,
  164. NHWC_NCHW4_IC_SMALL: 16,
  165. NCHW_NCHW4_IC_SMALL: 17,
  166. CHWN4: 18,
  167. NCHW4_NHWC: 19
  168. };
  169. $root.mgb.serialization.fbs.param.ConvolutionV0Mode = {
  170. CROSS_CORRELATION: 0,
  171. CONVOLUTION: 1
  172. };
  173. $root.mgb.serialization.fbs.param.ConvolutionV0Sparse = {
  174. DENSE: 0,
  175. GROUP: 1
  176. };
  177. $root.mgb.serialization.fbs.param.ConvolutionV1ComputeMode = {
  178. DEFAULT: 0,
  179. FLOAT32: 1
  180. };
  181. $root.mgb.serialization.fbs.param.CvtColorMode = {
  182. RGB2GRAY: 0,
  183. RGB2YUV: 1,
  184. YUV2RGB: 2,
  185. GRAY2RGB: 3,
  186. RGBA2RGB: 4,
  187. RGBA2BGR: 5,
  188. RGBA2GRAY: 6,
  189. RGB2BGR: 7,
  190. BGR2GRAY: 8,
  191. BGR2RGB: 9,
  192. YUV2GRAY_NV21: 10,
  193. YUV2RGB_NV21: 11,
  194. YUV2BGR_NV21: 12,
  195. YUV2GRAY_NV12: 13,
  196. YUV2RGB_NV12: 14,
  197. YUV2BGR_NV12: 15,
  198. YUV2GRAY_YV12: 16,
  199. YUV2RGB_YV12: 17,
  200. YUV2BGR_YV12: 18,
  201. YUV2GRAY_YU12: 19,
  202. YUV2RGB_YU12: 20,
  203. YUV2BGR_YU12: 21,
  204. YCrCb2RGB: 22,
  205. YCrCb2BGR: 23,
  206. BT601_YUV2RGB_NV21: 24,
  207. BT601_YUV2BGR_NV21: 25,
  208. BT601_YUV2RGB_NV12: 26,
  209. BT601_YUV2BGR_NV12: 27,
  210. BT601_YUV2RGB_YV12: 28,
  211. BT601_YUV2BGR_YV12: 29,
  212. BT601_YUV2RGB_YU12: 30,
  213. BT601_YUV2BGR_YU12: 31
  214. };
  215. $root.mgb.serialization.fbs.param.DctChannelSelectV0FastImpl = {
  216. NONE: 0,
  217. FIX_32_MASK: 1
  218. };
  219. $root.mgb.serialization.fbs.param.ElemwiseMode = {
  220. RELU: 0,
  221. ABS: 1,
  222. ACOS: 2,
  223. ASIN: 3,
  224. CEIL: 4,
  225. COS: 5,
  226. EXP: 6,
  227. EXPM1: 7,
  228. FLOOR: 8,
  229. LOG: 9,
  230. LOG1P: 10,
  231. NEGATE: 11,
  232. SIGMOID: 12,
  233. SIN: 13,
  234. TANH: 14,
  235. ABS_GRAD: 15,
  236. ADD: 16,
  237. FLOOR_DIV: 17,
  238. MAX_: 18,
  239. MIN_: 19,
  240. MOD: 20,
  241. MUL: 21,
  242. POW: 22,
  243. SIGMOID_GRAD: 23,
  244. SUB: 24,
  245. SWITCH_GT0: 25,
  246. TANH_GRAD: 26,
  247. TRUE_DIV: 27,
  248. LOG_SUM_EXP: 28,
  249. LT: 29,
  250. LEQ: 30,
  251. EQ: 31,
  252. SHL: 32,
  253. SHR: 33,
  254. COND_LEQ_MOV: 34,
  255. FUSE_MUL_ADD3: 35,
  256. FUSE_MUL_ADD4: 36,
  257. FUSE_ADD_RELU: 37,
  258. FUSE_ADD_SIGMOID: 38,
  259. FUSE_ADD_TANH: 39,
  260. FAST_TANH: 40,
  261. FAST_TANH_GRAD: 41,
  262. ROUND: 42,
  263. RMULH: 43,
  264. ATAN2: 44,
  265. ERF: 45,
  266. ERFINV: 46,
  267. ERFC: 47,
  268. ERFCINV: 48,
  269. H_SWISH: 49,
  270. H_SWISH_GRAD: 50,
  271. FUSE_ADD_H_SWISH: 51,
  272. NOT: 52,
  273. AND: 53,
  274. OR: 54,
  275. XOR: 55,
  276. SILU: 56,
  277. SILU_GRAD: 57,
  278. GELU: 58,
  279. GELU_GRAD: 59
  280. };
  281. $root.mgb.serialization.fbs.param.ElemwiseMultiTypeMode = {
  282. FUSE_MUL_ADD3_INT16x32x32x32: 0,
  283. FUSE_MUL_ADD3_IXxF32xF32xI8: 1,
  284. ROUND_SHR_SATURATE_IXxI8xI8: 2,
  285. FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT16x16x16x8: 3,
  286. FUSE_ADD_RMULH_ROUND_SHR_SATURATE_INT32x32x32x8: 4,
  287. ROUND_SHR_SATURATE_IXxI8xI16: 5,
  288. QADD: 6,
  289. QFUSE_ADD_RELU: 7,
  290. QMUL: 8,
  291. QMIN: 9,
  292. QMAX: 10,
  293. QSUB: 11,
  294. QTRUE_DIV: 12,
  295. QFUSE_ADD_SIGMOID: 13,
  296. QFUSE_ADD_TANH: 14,
  297. QRELU: 15,
  298. QABS: 16,
  299. QSIGMOID: 17,
  300. QEXP: 18,
  301. QTANH: 19,
  302. QFUSE_MUL_ADD3: 20,
  303. QFAST_TANH: 21,
  304. QNEGATE: 22,
  305. QACOS: 23,
  306. QASIN: 24,
  307. QCEIL: 25,
  308. QCOS: 26,
  309. QEXPM1: 27,
  310. QFLOOR: 28,
  311. QLOG: 29,
  312. QLOG1P: 30,
  313. QSIN: 31,
  314. QROUND: 32,
  315. QERF: 33,
  316. QERFINV: 34,
  317. QERFC: 35,
  318. QERFCINV: 36,
  319. QABS_GRAD: 37,
  320. QFLOOR_DIV: 38,
  321. QMOD: 39,
  322. QSIGMOID_GRAD: 40,
  323. QSWITCH_GT0: 41,
  324. QTANH_GRAD: 42,
  325. QLT: 43,
  326. QLEQ: 44,
  327. QEQ: 45,
  328. QPOW: 46,
  329. QLOG_SUM_EXP: 47,
  330. QFAST_TANH_GRAD: 48,
  331. QATAN2: 49,
  332. QCOND_LEQ_MOV: 50,
  333. QH_SWISH: 51,
  334. QFUSE_ADD_H_SWISH: 52,
  335. QH_SWISH_GRAD: 53,
  336. FUSE_MUL_ADD3_INT16xF32xF32xF32: 54,
  337. MUL_INT16xF32xF32: 55,
  338. FUSE_MUL_ADD3_UINT8xF32xF32xF32: 56
  339. };
  340. $root.mgb.serialization.fbs.param.MatrixMulFormat = {
  341. DEFAULT: 0,
  342. MK4: 1,
  343. MK8: 2,
  344. MK4_DOT: 3
  345. };
  346. $root.mgb.serialization.fbs.param.MatrixMulV0DataType = {
  347. FLOAT: 0,
  348. INT8x8x16: 1,
  349. INT8x8x32: 2,
  350. FLOAT_IO16xC32: 3,
  351. QUINT8x8x32: 4,
  352. QUINT4x4x32: 5
  353. };
  354. $root.mgb.serialization.fbs.param.MatrixMulV1ComputeMode = {
  355. DEFAULT: 0,
  356. FLOAT32: 1
  357. };
  358. $root.mgb.serialization.fbs.param.PaddingPaddingMode = {
  359. REPLICATE: 0,
  360. REFLECT: 1,
  361. CONSTANT: 2
  362. };
  363. $root.mgb.serialization.fbs.param.PoolingV0Mode = {
  364. MAX_: 0,
  365. AVERAGE: 1,
  366. AVERAGE_COUNT_EXCLUDE_PADDING: 2
  367. };
  368. $root.mgb.serialization.fbs.param.RNNCellNonlineMode = {
  369. IDENTITY: 0,
  370. RELU: 1,
  371. TANH: 2
  372. };
  373. $root.mgb.serialization.fbs.param.ROIAlignV0Mode = {
  374. MAX_: 0,
  375. AVERAGE: 1
  376. };
  377. $root.mgb.serialization.fbs.param.ROIPoolingMode = {
  378. MAX_: 0,
  379. AVERAGE: 1
  380. };
  381. $root.mgb.serialization.fbs.param.ReduceDataType = {
  382. DEFAULT: 0,
  383. FLOAT_IO16xC32: 1,
  384. FLOAT_O32xC32: 2,
  385. FLOAT_O16xC32: 3,
  386. QUINT_I8xO32: 4,
  387. QINT_I8xO32: 5
  388. };
  389. $root.mgb.serialization.fbs.param.ReduceMode = {
  390. SUM: 0,
  391. SUM_SQR: 1,
  392. PRODUCT: 2,
  393. MIN_: 3,
  394. MAX_: 4,
  395. MEAN: 5
  396. };
  397. $root.mgb.serialization.fbs.param.ReduceV0Mode = {
  398. SUM: 0,
  399. SUM_SQR: 1,
  400. PRODUCT: 2,
  401. MIN_: 3,
  402. MAX_: 4
  403. };
  404. $root.mgb.serialization.fbs.param.ReduceV1DataType = {
  405. DEFAULT: 0,
  406. FLOAT_IO16xC32: 1,
  407. FLOAT_O32xC32: 2,
  408. FLOAT_O16xC32: 3,
  409. QUINT_I8xO32: 4,
  410. QINT_I8xO32: 5
  411. };
  412. $root.mgb.serialization.fbs.param.ReduceV1Mode = {
  413. SUM: 0,
  414. SUM_SQR: 1,
  415. PRODUCT: 2,
  416. MIN_: 3,
  417. MAX_: 4,
  418. MEAN: 5
  419. };
  420. $root.mgb.serialization.fbs.param.RelayoutFormatV0Mode = {
  421. NHWC_NHWCD4: 0,
  422. NHWCD4_NHWC: 1,
  423. NHWC_NHWCD4I: 2,
  424. NCHW_NHWCD4: 3,
  425. NCHW_NHWCD4I: 4,
  426. NHWCD4I_NCHW: 5,
  427. NHWCD4_NCHW: 6,
  428. INTER_WEIGHT_DENSE: 7,
  429. INTER_WEIGHT_DENSEI: 8,
  430. INTER_WEIGHT_GROUP: 9,
  431. INTER_WEIGHT_GROUPI: 10,
  432. INTER_WEIGHT_CHAN: 11,
  433. INTER_WEIGHT_CHANI: 12,
  434. INTER_WEIGHT_DENSEI_DOT: 13,
  435. INTER_WEIGHT_GROUPI_DOT: 14,
  436. NCHW4_CHWN4: 15,
  437. CHWN4_NCHW4: 16,
  438. NCHW_NCHW88_CONV_DENSE_WEIGHT: 17,
  439. NCHW_NCHW88_CONV_CHAN_WEIGHT: 18,
  440. NCHW_NCHW88_CONV_GROUP_WEIGHT: 19,
  441. NCHW_NCHW88: 20,
  442. NCHW88_NCHW: 21,
  443. NCHW_NCHW4_IC_SMALL: 22,
  444. NCHW_NCHW4_IC_SMALL_CONV_DENSE_WEIGHT: 23,
  445. NCHW_NCHW4: 24,
  446. NCHW4_NCHW: 25,
  447. NCHW_NCHW4_WEIGHT: 26,
  448. NCHW_NCHW64: 27,
  449. NCHW64_NCHW: 28,
  450. NCHW_NHWC: 29,
  451. NHWC_NCHW: 30,
  452. NHWCD4I_NHWC: 31
  453. };
  454. $root.mgb.serialization.fbs.param.SeparableConvBorderMode = {
  455. BORDER_REPLICATE: 0,
  456. BORDER_REFLECT: 1,
  457. BORDER_REFLECT_101: 2,
  458. BORDER_WRAP: 3,
  459. BORDER_CONSTANT: 4,
  460. BORDER_TRANSPARENT: 5,
  461. BORDER_ISOLATED: 6
  462. };
  463. $root.mgb.serialization.fbs.param.SeparableConv3DBorderMode = {
  464. BORDER_REPLICATE: 0,
  465. BORDER_REFLECT: 1,
  466. BORDER_REFLECT_101: 2,
  467. BORDER_WRAP: 3,
  468. BORDER_CONSTANT: 4,
  469. BORDER_TRANSPARENT: 5,
  470. BORDER_ISOLATED: 6
  471. };
  472. $root.mgb.serialization.fbs.param.SpatialTfGridGeneratorMode = {
  473. AFFINE: 0
  474. };
  475. $root.mgb.serialization.fbs.param.SpatialTfSamplerMode = {
  476. BILINEAR: 0
  477. };
  478. $root.mgb.serialization.fbs.param.TopKMode = {
  479. KTH_ONLY: 0,
  480. VALUE_IDX_NOSORT: 1,
  481. VALUE_IDX_SORTED: 2
  482. };
  483. $root.mgb.serialization.fbs.param.WarpPerspectiveV1BorderMode = {
  484. REPLICATE: 0,
  485. REFLECT: 1,
  486. REFLECT_101: 2,
  487. WRAP: 3,
  488. CONSTANT: 4,
  489. TRANSPARENT: 5,
  490. ISOLATED: 6
  491. };
  492. $root.mgb.serialization.fbs.param.WarpPerspectiveV1InterpolationMode = {
  493. NEAREST: 0,
  494. LINEAR: 1,
  495. AREA: 2,
  496. CUBIC: 3,
  497. LANCZOS4: 4
  498. };
  499. $root.mgb.serialization.fbs.param.Empty = class Empty {
  500. static decode(/* reader, position */) {
  501. const $ = new $root.mgb.serialization.fbs.param.Empty();
  502. return $;
  503. }
  504. };
  505. $root.mgb.serialization.fbs.param.Axis = class Axis {
  506. static decode(reader, position) {
  507. const $ = new $root.mgb.serialization.fbs.param.Axis();
  508. $.axis = reader.int32_(position, 4, 0);
  509. return $;
  510. }
  511. };
  512. $root.mgb.serialization.fbs.param.ConvolutionV0 = class ConvolutionV0 {
  513. static decode(reader, position) {
  514. const $ = new $root.mgb.serialization.fbs.param.ConvolutionV0();
  515. $.mode = reader.uint32_(position, 4, 0);
  516. $.pad_h = reader.uint32_(position, 6, 0);
  517. $.pad_w = reader.uint32_(position, 8, 0);
  518. $.stride_h = reader.uint32_(position, 10, 1);
  519. $.stride_w = reader.uint32_(position, 12, 1);
  520. $.dilate_h = reader.uint32_(position, 14, 1);
  521. $.dilate_w = reader.uint32_(position, 16, 1);
  522. $.data_type = reader.uint32_(position, 18, 0);
  523. $.sparse = reader.uint32_(position, 20, 0);
  524. $.format = reader.uint32_(position, 22, 0);
  525. return $;
  526. }
  527. };
  528. $root.mgb.serialization.fbs.param.ConvolutionV1 = class ConvolutionV1 {
  529. static decode(reader, position) {
  530. const $ = new $root.mgb.serialization.fbs.param.ConvolutionV1();
  531. $.mode = reader.uint32_(position, 4, 0);
  532. $.pad_h = reader.uint32_(position, 6, 0);
  533. $.pad_w = reader.uint32_(position, 8, 0);
  534. $.stride_h = reader.uint32_(position, 10, 1);
  535. $.stride_w = reader.uint32_(position, 12, 1);
  536. $.dilate_h = reader.uint32_(position, 14, 1);
  537. $.dilate_w = reader.uint32_(position, 16, 1);
  538. $.sparse = reader.uint32_(position, 18, 0);
  539. $.format = reader.uint32_(position, 20, 0);
  540. $.compute_mode = reader.uint32_(position, 22, 0);
  541. return $;
  542. }
  543. };
  544. $root.mgb.serialization.fbs.param.Convolution = class Convolution {
  545. static decode(reader, position) {
  546. const $ = new $root.mgb.serialization.fbs.param.Convolution();
  547. $.mode = reader.uint32_(position, 4, 0);
  548. $.pad_h = reader.uint32_(position, 6, 0);
  549. $.pad_w = reader.uint32_(position, 8, 0);
  550. $.stride_h = reader.uint32_(position, 10, 1);
  551. $.stride_w = reader.uint32_(position, 12, 1);
  552. $.dilate_h = reader.uint32_(position, 14, 1);
  553. $.dilate_w = reader.uint32_(position, 16, 1);
  554. $.sparse = reader.uint32_(position, 18, 0);
  555. $.format = reader.uint32_(position, 20, 0);
  556. $.compute_mode = reader.uint32_(position, 22, 0);
  557. return $;
  558. }
  559. };
  560. $root.mgb.serialization.fbs.param.MaskPropagate = class MaskPropagate {
  561. static decode(reader, position) {
  562. const $ = new $root.mgb.serialization.fbs.param.MaskPropagate();
  563. $.pad_h = reader.uint32_(position, 4, 0);
  564. $.pad_w = reader.uint32_(position, 6, 0);
  565. $.stride_h = reader.uint32_(position, 8, 1);
  566. $.stride_w = reader.uint32_(position, 10, 1);
  567. $.kernel_h = reader.uint32_(position, 12, 1);
  568. $.kernel_w = reader.uint32_(position, 14, 1);
  569. $.dilate_h = reader.uint32_(position, 16, 1);
  570. $.dilate_w = reader.uint32_(position, 18, 1);
  571. return $;
  572. }
  573. };
  574. $root.mgb.serialization.fbs.param.ConvPooling = class ConvPooling {
  575. static decode(reader, position) {
  576. const $ = new $root.mgb.serialization.fbs.param.ConvPooling();
  577. $.method = reader.uint32_(position, 4, 0);
  578. $.convMode = reader.uint32_(position, 6, 0);
  579. $.poolMode = reader.uint32_(position, 8, 0);
  580. $.nonlineMode = reader.uint32_(position, 10, 0);
  581. $.pool_shape_h = reader.uint32_(position, 12, 1);
  582. $.pool_shape_w = reader.uint32_(position, 14, 1);
  583. $.pool_stride_h = reader.uint32_(position, 16, 1);
  584. $.pool_stride_w = reader.uint32_(position, 18, 1);
  585. $.pool_pad_h = reader.uint32_(position, 20, 0);
  586. $.pool_pad_w = reader.uint32_(position, 22, 0);
  587. $.conv_stride_h = reader.uint32_(position, 24, 1);
  588. $.conv_stride_w = reader.uint32_(position, 26, 1);
  589. $.conv_pad_h = reader.uint32_(position, 28, 0);
  590. $.conv_pad_w = reader.uint32_(position, 30, 0);
  591. return $;
  592. }
  593. };
  594. $root.mgb.serialization.fbs.param.ConvBiasV0 = class ConvBiasV0 {
  595. static decode(reader, position) {
  596. const $ = new $root.mgb.serialization.fbs.param.ConvBiasV0();
  597. $.nonlineMode = reader.uint32_(position, 4, 0);
  598. $.mode = reader.uint32_(position, 6, 0);
  599. $.pad_h = reader.uint32_(position, 8, 0);
  600. $.pad_w = reader.uint32_(position, 10, 0);
  601. $.stride_h = reader.uint32_(position, 12, 1);
  602. $.stride_w = reader.uint32_(position, 14, 1);
  603. return $;
  604. }
  605. };
  606. $root.mgb.serialization.fbs.param.ConvBiasV1 = class ConvBiasV1 {
  607. static decode(reader, position) {
  608. const $ = new $root.mgb.serialization.fbs.param.ConvBiasV1();
  609. $.nonlineMode = reader.uint32_(position, 4, 0);
  610. $.mode = reader.uint32_(position, 6, 0);
  611. $.data_type = reader.uint32_(position, 8, 0);
  612. $.sparse = reader.uint32_(position, 10, 0);
  613. $.format = reader.uint32_(position, 12, 0);
  614. $.pad_h = reader.uint32_(position, 14, 0);
  615. $.pad_w = reader.uint32_(position, 16, 0);
  616. $.stride_h = reader.uint32_(position, 18, 1);
  617. $.stride_w = reader.uint32_(position, 20, 1);
  618. $.dilate_h = reader.uint32_(position, 22, 1);
  619. $.dilate_w = reader.uint32_(position, 24, 1);
  620. return $;
  621. }
  622. };
  623. $root.mgb.serialization.fbs.param.ConvBiasV2 = class ConvBiasV2 {
  624. static decode(reader, position) {
  625. const $ = new $root.mgb.serialization.fbs.param.ConvBiasV2();
  626. $.nonlineMode = reader.uint32_(position, 4, 0);
  627. $.mode = reader.uint32_(position, 6, 0);
  628. $.sparse = reader.uint32_(position, 8, 0);
  629. $.format = reader.uint32_(position, 10, 0);
  630. $.pad_h = reader.uint32_(position, 12, 0);
  631. $.pad_w = reader.uint32_(position, 14, 0);
  632. $.stride_h = reader.uint32_(position, 16, 1);
  633. $.stride_w = reader.uint32_(position, 18, 1);
  634. $.dilate_h = reader.uint32_(position, 20, 1);
  635. $.dilate_w = reader.uint32_(position, 22, 1);
  636. $.compute_mode = reader.uint32_(position, 24, 0);
  637. return $;
  638. }
  639. };
  640. $root.mgb.serialization.fbs.param.ConvBiasV3 = class ConvBiasV3 {
  641. static decode(reader, position) {
  642. const $ = new $root.mgb.serialization.fbs.param.ConvBiasV3();
  643. $.nonlineMode = reader.uint32_(position, 4, 0);
  644. $.mode = reader.uint32_(position, 6, 0);
  645. $.sparse = reader.uint32_(position, 8, 0);
  646. $.format = reader.uint32_(position, 10, 0);
  647. $.pad_h = reader.uint32_(position, 12, 0);
  648. $.pad_w = reader.uint32_(position, 14, 0);
  649. $.stride_h = reader.uint32_(position, 16, 1);
  650. $.stride_w = reader.uint32_(position, 18, 1);
  651. $.dilate_h = reader.uint32_(position, 20, 1);
  652. $.dilate_w = reader.uint32_(position, 22, 1);
  653. $.output_block_size = reader.uint32_(position, 24, 0);
  654. $.compute_mode = reader.uint32_(position, 26, 0);
  655. return $;
  656. }
  657. };
  658. $root.mgb.serialization.fbs.param.ConvBias = class ConvBias {
  659. static decode(reader, position) {
  660. const $ = new $root.mgb.serialization.fbs.param.ConvBias();
  661. $.nonlineMode = reader.uint32_(position, 4, 0);
  662. $.mode = reader.uint32_(position, 6, 0);
  663. $.sparse = reader.uint32_(position, 8, 0);
  664. $.format = reader.uint32_(position, 10, 0);
  665. $.pad_h = reader.uint32_(position, 12, 0);
  666. $.pad_w = reader.uint32_(position, 14, 0);
  667. $.stride_h = reader.uint32_(position, 16, 1);
  668. $.stride_w = reader.uint32_(position, 18, 1);
  669. $.dilate_h = reader.uint32_(position, 20, 1);
  670. $.dilate_w = reader.uint32_(position, 22, 1);
  671. $.compute_mode = reader.uint32_(position, 24, 0);
  672. return $;
  673. }
  674. };
  675. $root.mgb.serialization.fbs.param.SeparableConv = class SeparableConv {
  676. static decode(reader, position) {
  677. const $ = new $root.mgb.serialization.fbs.param.SeparableConv();
  678. $.mode = reader.uint32_(position, 4, 0);
  679. $.borderMode = reader.uint32_(position, 6, 0);
  680. $.is_symm_kernel = reader.bool_(position, 8, true);
  681. $.pad_h = reader.uint32_(position, 10, 0);
  682. $.pad_w = reader.uint32_(position, 12, 0);
  683. $.stride_h = reader.uint32_(position, 14, 1);
  684. $.stride_w = reader.uint32_(position, 16, 1);
  685. $.ksize_h = reader.uint32_(position, 18, 3);
  686. $.ksize_w = reader.uint32_(position, 20, 3);
  687. $.anchor_h = reader.uint32_(position, 22, 1);
  688. $.anchor_w = reader.uint32_(position, 24, 1);
  689. return $;
  690. }
  691. };
  692. $root.mgb.serialization.fbs.param.Images2Neibs = class Images2Neibs {
  693. static decode(reader, position) {
  694. const $ = new $root.mgb.serialization.fbs.param.Images2Neibs();
  695. $.pad_h = reader.uint32_(position, 4, 0);
  696. $.pad_w = reader.uint32_(position, 6, 0);
  697. $.stride_h = reader.uint32_(position, 8, 1);
  698. $.stride_w = reader.uint32_(position, 10, 1);
  699. $.dilate_h = reader.uint32_(position, 12, 1);
  700. $.dilate_w = reader.uint32_(position, 14, 1);
  701. $.window_h = reader.uint32_(position, 16, 3);
  702. $.window_w = reader.uint32_(position, 18, 3);
  703. return $;
  704. }
  705. };
  706. $root.mgb.serialization.fbs.param.SlidingWindowTranspose = class SlidingWindowTranspose {
  707. static decode(reader, position) {
  708. const $ = new $root.mgb.serialization.fbs.param.SlidingWindowTranspose();
  709. $.out_h = reader.uint32_(position, 4, 0);
  710. $.out_w = reader.uint32_(position, 6, 0);
  711. $.pad_h = reader.uint32_(position, 8, 0);
  712. $.pad_w = reader.uint32_(position, 10, 0);
  713. $.stride_h = reader.uint32_(position, 12, 1);
  714. $.stride_w = reader.uint32_(position, 14, 1);
  715. $.dilate_h = reader.uint32_(position, 16, 1);
  716. $.dilate_w = reader.uint32_(position, 18, 1);
  717. $.window_h = reader.uint32_(position, 20, 3);
  718. $.window_w = reader.uint32_(position, 22, 3);
  719. return $;
  720. }
  721. };
  722. $root.mgb.serialization.fbs.param.PoolingV0 = class PoolingV0 {
  723. static decode(reader, position) {
  724. const $ = new $root.mgb.serialization.fbs.param.PoolingV0();
  725. $.mode = reader.uint32_(position, 4, 0);
  726. $.pad_h = reader.uint32_(position, 6, 0);
  727. $.pad_w = reader.uint32_(position, 8, 0);
  728. $.stride_h = reader.uint32_(position, 10, 2);
  729. $.stride_w = reader.uint32_(position, 12, 2);
  730. $.window_h = reader.uint32_(position, 14, 2);
  731. $.window_w = reader.uint32_(position, 16, 2);
  732. $.format = reader.uint32_(position, 18, 0);
  733. return $;
  734. }
  735. };
  736. $root.mgb.serialization.fbs.param.Pooling = class Pooling {
  737. static decode(reader, position) {
  738. const $ = new $root.mgb.serialization.fbs.param.Pooling();
  739. $.mode = reader.uint32_(position, 4, 0);
  740. $.pad_h = reader.uint32_(position, 6, 0);
  741. $.pad_w = reader.uint32_(position, 8, 0);
  742. $.stride_h = reader.uint32_(position, 10, 2);
  743. $.stride_w = reader.uint32_(position, 12, 2);
  744. $.window_h = reader.uint32_(position, 14, 2);
  745. $.window_w = reader.uint32_(position, 16, 2);
  746. $.format = reader.uint32_(position, 18, 0);
  747. return $;
  748. }
  749. };
  750. $root.mgb.serialization.fbs.param.Softmax = class Softmax {
  751. static decode(reader, position) {
  752. const $ = new $root.mgb.serialization.fbs.param.Softmax();
  753. $.axis = reader.int32_(position, 4, -1);
  754. return $;
  755. }
  756. };
  757. $root.mgb.serialization.fbs.param.AdaptivePoolingV0 = class AdaptivePoolingV0 {
  758. static decode(reader, position) {
  759. const $ = new $root.mgb.serialization.fbs.param.AdaptivePoolingV0();
  760. $.mode = reader.uint32_(position, 4, 0);
  761. $.format = reader.uint32_(position, 6, 0);
  762. return $;
  763. }
  764. };
  765. $root.mgb.serialization.fbs.param.AdaptivePooling = class AdaptivePooling {
  766. static decode(reader, position) {
  767. const $ = new $root.mgb.serialization.fbs.param.AdaptivePooling();
  768. $.mode = reader.uint32_(position, 4, 0);
  769. $.format = reader.uint32_(position, 6, 0);
  770. return $;
  771. }
  772. };
  773. $root.mgb.serialization.fbs.param.LRN = class LRN {
  774. static decode(reader, position) {
  775. const $ = new $root.mgb.serialization.fbs.param.LRN();
  776. $.n = reader.uint32_(position, 4, 5);
  777. $.k = reader.float32_(position, 6, 2);
  778. $.alpha = reader.float32_(position, 8, 0.0001);
  779. $.beta = reader.float32_(position, 10, 0.75);
  780. return $;
  781. }
  782. };
  783. $root.mgb.serialization.fbs.param.BN = class BN {
  784. static decode(reader, position) {
  785. const $ = new $root.mgb.serialization.fbs.param.BN();
  786. $.param_dim = reader.uint32_(position, 4, 0);
  787. $.fwd_mode = reader.uint32_(position, 6, 0);
  788. $.epsilon = reader.float64_(position, 8, 0.0001);
  789. $.avg_factor = reader.float64_(position, 10, 1);
  790. $.scale = reader.float32_(position, 12, 1);
  791. $.bias = reader.float32_(position, 14, 0);
  792. return $;
  793. }
  794. };
  795. $root.mgb.serialization.fbs.param.ROIPooling = class ROIPooling {
  796. static decode(reader, position) {
  797. const $ = new $root.mgb.serialization.fbs.param.ROIPooling();
  798. $.mode = reader.uint32_(position, 4, 0);
  799. $.scale = reader.float32_(position, 6, 1);
  800. return $;
  801. }
  802. };
  803. $root.mgb.serialization.fbs.param.WarpPerspectiveV1 = class WarpPerspectiveV1 {
  804. static decode(reader, position) {
  805. const $ = new $root.mgb.serialization.fbs.param.WarpPerspectiveV1();
  806. $.imode = reader.uint32_(position, 4, 1);
  807. $.bmode = reader.uint32_(position, 6, 0);
  808. $.format = reader.uint32_(position, 8, 0);
  809. $.border_val = reader.float32_(position, 10, 0);
  810. return $;
  811. }
  812. };
  813. $root.mgb.serialization.fbs.param.WarpPerspective = class WarpPerspective {
  814. static decode(reader, position) {
  815. const $ = new $root.mgb.serialization.fbs.param.WarpPerspective();
  816. $.imode = reader.uint32_(position, 4, 1);
  817. $.bmode = reader.uint32_(position, 6, 0);
  818. $.format = reader.uint32_(position, 8, 0);
  819. $.border_val = reader.float32_(position, 10, 0);
  820. return $;
  821. }
  822. };
  823. $root.mgb.serialization.fbs.param.SpatialTfGridGenerator = class SpatialTfGridGenerator {
  824. static decode(reader, position) {
  825. const $ = new $root.mgb.serialization.fbs.param.SpatialTfGridGenerator();
  826. $.mode = reader.uint32_(position, 4, 0);
  827. return $;
  828. }
  829. };
  830. $root.mgb.serialization.fbs.param.SpatialTfSampler = class SpatialTfSampler {
  831. static decode(reader, position) {
  832. const $ = new $root.mgb.serialization.fbs.param.SpatialTfSampler();
  833. $.mode = reader.uint32_(position, 4, 0);
  834. return $;
  835. }
  836. };
  837. $root.mgb.serialization.fbs.param.AddUpdate = class AddUpdate {
  838. static decode(reader, position) {
  839. const $ = new $root.mgb.serialization.fbs.param.AddUpdate();
  840. $.alpha = reader.float32_(position, 4, 1);
  841. $.beta = reader.float32_(position, 6, 1);
  842. $.bias = reader.float32_(position, 8, 0);
  843. return $;
  844. }
  845. };
  846. $root.mgb.serialization.fbs.param.Elemwise = class Elemwise {
  847. static decode(reader, position) {
  848. const $ = new $root.mgb.serialization.fbs.param.Elemwise();
  849. $.mode = reader.uint32_(position, 4, 0);
  850. return $;
  851. }
  852. };
  853. $root.mgb.serialization.fbs.param.ElemwiseMultiType = class ElemwiseMultiType {
  854. static decode(reader, position) {
  855. const $ = new $root.mgb.serialization.fbs.param.ElemwiseMultiType();
  856. $.mode = reader.uint32_(position, 4, 0);
  857. return $;
  858. }
  859. };
  860. $root.mgb.serialization.fbs.param.PowC = class PowC {
  861. static decode(reader, position) {
  862. const $ = new $root.mgb.serialization.fbs.param.PowC();
  863. $.exp = reader.float32_(position, 4, 0);
  864. return $;
  865. }
  866. };
  867. $root.mgb.serialization.fbs.param.DctChannelSelectV0 = class DctChannelSelectV0 {
  868. static decode(reader, position) {
  869. const $ = new $root.mgb.serialization.fbs.param.DctChannelSelectV0();
  870. $.format = reader.uint32_(position, 4, 0);
  871. $.fastImpl = reader.uint32_(position, 6, 0);
  872. $.dct_block_size = reader.int32_(position, 8, 8);
  873. return $;
  874. }
  875. };
  876. $root.mgb.serialization.fbs.param.DctChannelSelect = class DctChannelSelect {
  877. static decode(reader, position) {
  878. const $ = new $root.mgb.serialization.fbs.param.DctChannelSelect();
  879. $.format = reader.uint32_(position, 4, 0);
  880. $.fastImpl = reader.uint32_(position, 6, 0);
  881. $.dct_block_size = reader.int32_(position, 8, 8);
  882. return $;
  883. }
  884. };
  885. $root.mgb.serialization.fbs.param.MatrixMulV0 = class MatrixMulV0 {
  886. static decode(reader, position) {
  887. const $ = new $root.mgb.serialization.fbs.param.MatrixMulV0();
  888. $.transposeA = reader.bool_(position, 4, false);
  889. $.transposeB = reader.bool_(position, 6, false);
  890. $.data_type = reader.uint32_(position, 8, 0);
  891. return $;
  892. }
  893. };
  894. $root.mgb.serialization.fbs.param.MatrixMulV1 = class MatrixMulV1 {
  895. static decode(reader, position) {
  896. const $ = new $root.mgb.serialization.fbs.param.MatrixMulV1();
  897. $.transposeA = reader.bool_(position, 4, false);
  898. $.transposeB = reader.bool_(position, 6, false);
  899. $.compute_mode = reader.uint32_(position, 8, 0);
  900. return $;
  901. }
  902. };
  903. $root.mgb.serialization.fbs.param.MatrixMul = class MatrixMul {
  904. static decode(reader, position) {
  905. const $ = new $root.mgb.serialization.fbs.param.MatrixMul();
  906. $.transposeA = reader.bool_(position, 4, false);
  907. $.transposeB = reader.bool_(position, 6, false);
  908. $.compute_mode = reader.uint32_(position, 8, 0);
  909. $.format = reader.uint32_(position, 10, 0);
  910. return $;
  911. }
  912. };
  913. $root.mgb.serialization.fbs.param.SVD = class SVD {
  914. static decode(reader, position) {
  915. const $ = new $root.mgb.serialization.fbs.param.SVD();
  916. $.full_matrices = reader.bool_(position, 4, false);
  917. $.compute_uv = reader.bool_(position, 6, true);
  918. return $;
  919. }
  920. };
  921. $root.mgb.serialization.fbs.param.ReduceV0 = class ReduceV0 {
  922. static decode(reader, position) {
  923. const $ = new $root.mgb.serialization.fbs.param.ReduceV0();
  924. $.mode = reader.uint32_(position, 4, 0);
  925. $.axis = reader.int32_(position, 6, -1);
  926. return $;
  927. }
  928. };
  929. $root.mgb.serialization.fbs.param.ReduceV1 = class ReduceV1 {
  930. static decode(reader, position) {
  931. const $ = new $root.mgb.serialization.fbs.param.ReduceV1();
  932. $.mode = reader.uint32_(position, 4, 0);
  933. $.axis = reader.int32_(position, 6, -1);
  934. $.data_type = reader.uint32_(position, 8, 0);
  935. return $;
  936. }
  937. };
  938. $root.mgb.serialization.fbs.param.Reduce = class Reduce {
  939. static decode(reader, position) {
  940. const $ = new $root.mgb.serialization.fbs.param.Reduce();
  941. $.mode = reader.uint32_(position, 4, 0);
  942. $.axis = reader.int32_(position, 6, 2147483647);
  943. $.data_type = reader.uint32_(position, 8, 0);
  944. return $;
  945. }
  946. };
  947. $root.mgb.serialization.fbs.param.CumsumV0 = class CumsumV0 {
  948. static decode(reader, position) {
  949. const $ = new $root.mgb.serialization.fbs.param.CumsumV0();
  950. $.axis = reader.int32_(position, 4, -1);
  951. $.exclusive = reader.bool_(position, 6, true);
  952. $.reverse = reader.bool_(position, 8, false);
  953. return $;
  954. }
  955. };
  956. $root.mgb.serialization.fbs.param.Cumsum = class Cumsum {
  957. static decode(reader, position) {
  958. const $ = new $root.mgb.serialization.fbs.param.Cumsum();
  959. $.axis = reader.int32_(position, 4, 2147483647);
  960. $.exclusive = reader.bool_(position, 6, true);
  961. $.reverse = reader.bool_(position, 8, false);
  962. return $;
  963. }
  964. };
  965. $root.mgb.serialization.fbs.param.CondTake = class CondTake {
  966. static decode(reader, position) {
  967. const $ = new $root.mgb.serialization.fbs.param.CondTake();
  968. $.mode = reader.uint32_(position, 4, 0);
  969. $.val = reader.float32_(position, 6, 0);
  970. $.eps = reader.float32_(position, 8, 0.000001);
  971. return $;
  972. }
  973. };
  974. $root.mgb.serialization.fbs.param.Argsort = class Argsort {
  975. static decode(reader, position) {
  976. const $ = new $root.mgb.serialization.fbs.param.Argsort();
  977. $.order = reader.uint32_(position, 4, 0);
  978. return $;
  979. }
  980. };
  981. $root.mgb.serialization.fbs.param.IndexingRemap = class IndexingRemap {
  982. static decode(reader, position) {
  983. const $ = new $root.mgb.serialization.fbs.param.IndexingRemap();
  984. $.is_non_overlapping = reader.bool_(position, 4, false);
  985. return $;
  986. }
  987. };
  988. $root.mgb.serialization.fbs.param.Sleep = class Sleep {
  989. static decode(reader, position) {
  990. const $ = new $root.mgb.serialization.fbs.param.Sleep();
  991. $.time = reader.float32_(position, 4, 0);
  992. return $;
  993. }
  994. };
  995. $root.mgb.serialization.fbs.param.Linspace = class Linspace {
  996. static decode(reader, position) {
  997. const $ = new $root.mgb.serialization.fbs.param.Linspace();
  998. $.endpoint = reader.bool_(position, 4, true);
  999. return $;
  1000. }
  1001. };
  1002. $root.mgb.serialization.fbs.param.LinspaceFull = class LinspaceFull {
  1003. static decode(reader, position) {
  1004. const $ = new $root.mgb.serialization.fbs.param.LinspaceFull();
  1005. $.start = reader.float64_(position, 4, 0);
  1006. $.stop = reader.float64_(position, 6, 1);
  1007. $.endpoint = reader.bool_(position, 8, true);
  1008. return $;
  1009. }
  1010. };
  1011. $root.mgb.serialization.fbs.param.Eye = class Eye {
  1012. static decode(reader, position) {
  1013. const $ = new $root.mgb.serialization.fbs.param.Eye();
  1014. $.k = reader.int32_(position, 4, 0);
  1015. $.dtype = reader.int8_(position, 6, 0);
  1016. return $;
  1017. }
  1018. };
  1019. $root.mgb.serialization.fbs.param.Diag = class Diag {
  1020. static decode(reader, position) {
  1021. const $ = new $root.mgb.serialization.fbs.param.Diag();
  1022. $.k = reader.int32_(position, 4, 0);
  1023. return $;
  1024. }
  1025. };
  1026. $root.mgb.serialization.fbs.param.UniformRNGV0 = class UniformRNGV0 {
  1027. static decode(reader, position) {
  1028. const $ = new $root.mgb.serialization.fbs.param.UniformRNGV0();
  1029. $.seed = reader.uint64_(position, 4, 0);
  1030. return $;
  1031. }
  1032. };
  1033. $root.mgb.serialization.fbs.param.UniformRNG = class UniformRNG {
  1034. static decode(reader, position) {
  1035. const $ = new $root.mgb.serialization.fbs.param.UniformRNG();
  1036. $.seed = reader.uint64_(position, 4, 0);
  1037. $.dtype = reader.int8_(position, 6, 0);
  1038. return $;
  1039. }
  1040. };
  1041. $root.mgb.serialization.fbs.param.GaussianRNGV0 = class GaussianRNGV0 {
  1042. static decode(reader, position) {
  1043. const $ = new $root.mgb.serialization.fbs.param.GaussianRNGV0();
  1044. $.seed = reader.uint64_(position, 4, 0);
  1045. $.mean = reader.float32_(position, 6, 0);
  1046. $.std = reader.float32_(position, 8, 1);
  1047. return $;
  1048. }
  1049. };
  1050. $root.mgb.serialization.fbs.param.GaussianRNG = class GaussianRNG {
  1051. static decode(reader, position) {
  1052. const $ = new $root.mgb.serialization.fbs.param.GaussianRNG();
  1053. $.seed = reader.uint64_(position, 4, 0);
  1054. $.mean = reader.float32_(position, 6, 0);
  1055. $.std = reader.float32_(position, 8, 1);
  1056. $.dtype = reader.int8_(position, 10, 0);
  1057. return $;
  1058. }
  1059. };
  1060. $root.mgb.serialization.fbs.param.GammaRNG = class GammaRNG {
  1061. static decode(reader, position) {
  1062. const $ = new $root.mgb.serialization.fbs.param.GammaRNG();
  1063. $.seed = reader.uint64_(position, 4, 0);
  1064. return $;
  1065. }
  1066. };
  1067. $root.mgb.serialization.fbs.param.BetaRNG = class BetaRNG {
  1068. static decode(reader, position) {
  1069. const $ = new $root.mgb.serialization.fbs.param.BetaRNG();
  1070. $.seed = reader.uint64_(position, 4, 0);
  1071. return $;
  1072. }
  1073. };
  1074. $root.mgb.serialization.fbs.param.PoissonRNG = class PoissonRNG {
  1075. static decode(reader, position) {
  1076. const $ = new $root.mgb.serialization.fbs.param.PoissonRNG();
  1077. $.seed = reader.uint64_(position, 4, 0);
  1078. return $;
  1079. }
  1080. };
  1081. $root.mgb.serialization.fbs.param.PermutationRNG = class PermutationRNG {
  1082. static decode(reader, position) {
  1083. const $ = new $root.mgb.serialization.fbs.param.PermutationRNG();
  1084. $.seed = reader.uint64_(position, 4, 0);
  1085. $.dtype = reader.int8_(position, 6, 4);
  1086. return $;
  1087. }
  1088. };
  1089. $root.mgb.serialization.fbs.param.ShuffleRNG = class ShuffleRNG {
  1090. static decode(reader, position) {
  1091. const $ = new $root.mgb.serialization.fbs.param.ShuffleRNG();
  1092. $.seed = reader.uint64_(position, 4, 0);
  1093. return $;
  1094. }
  1095. };
  1096. $root.mgb.serialization.fbs.param.Flip = class Flip {
  1097. static decode(reader, position) {
  1098. const $ = new $root.mgb.serialization.fbs.param.Flip();
  1099. $.vertical = reader.bool_(position, 4, false);
  1100. $.horizontal = reader.bool_(position, 6, false);
  1101. return $;
  1102. }
  1103. };
  1104. $root.mgb.serialization.fbs.param.Rotate = class Rotate {
  1105. static decode(reader, position) {
  1106. const $ = new $root.mgb.serialization.fbs.param.Rotate();
  1107. $.clockwise = reader.bool_(position, 4, true);
  1108. return $;
  1109. }
  1110. };
  1111. $root.mgb.serialization.fbs.param.ROICopy = class ROICopy {
  1112. static decode(reader, position) {
  1113. const $ = new $root.mgb.serialization.fbs.param.ROICopy();
  1114. $.row_from = reader.uint32_(position, 4, 0);
  1115. $.row_to = reader.uint32_(position, 6, 0);
  1116. $.col_from = reader.uint32_(position, 8, 0);
  1117. $.col_to = reader.uint32_(position, 10, 0);
  1118. return $;
  1119. }
  1120. };
  1121. $root.mgb.serialization.fbs.param.CvtColor = class CvtColor {
  1122. static decode(reader, position) {
  1123. const $ = new $root.mgb.serialization.fbs.param.CvtColor();
  1124. $.mode = reader.uint32_(position, 4, 0);
  1125. return $;
  1126. }
  1127. };
  1128. $root.mgb.serialization.fbs.param.WarpAffineV0 = class WarpAffineV0 {
  1129. static decode(reader, position) {
  1130. const $ = new $root.mgb.serialization.fbs.param.WarpAffineV0();
  1131. $.imode = reader.uint32_(position, 4, 1);
  1132. $.border_mode = reader.uint32_(position, 6, 0);
  1133. $.border_val = reader.float32_(position, 8, 0);
  1134. return $;
  1135. }
  1136. };
  1137. $root.mgb.serialization.fbs.param.WarpAffineV1 = class WarpAffineV1 {
  1138. static decode(reader, position) {
  1139. const $ = new $root.mgb.serialization.fbs.param.WarpAffineV1();
  1140. $.imode = reader.uint32_(position, 4, 1);
  1141. $.border_mode = reader.uint32_(position, 6, 0);
  1142. $.border_val = reader.float32_(position, 8, 0);
  1143. $.format = reader.uint32_(position, 10, 1);
  1144. return $;
  1145. }
  1146. };
  1147. $root.mgb.serialization.fbs.param.WarpAffine = class WarpAffine {
  1148. static decode(reader, position) {
  1149. const $ = new $root.mgb.serialization.fbs.param.WarpAffine();
  1150. $.imode = reader.uint32_(position, 4, 1);
  1151. $.border_mode = reader.uint32_(position, 6, 0);
  1152. $.border_val = reader.float32_(position, 8, 0);
  1153. $.format = reader.uint32_(position, 10, 1);
  1154. return $;
  1155. }
  1156. };
  1157. $root.mgb.serialization.fbs.param.GaussianBlur = class GaussianBlur {
  1158. static decode(reader, position) {
  1159. const $ = new $root.mgb.serialization.fbs.param.GaussianBlur();
  1160. $.border_mode = reader.uint32_(position, 4, 0);
  1161. $.kernel_height = reader.uint32_(position, 6, 0);
  1162. $.kernel_width = reader.uint32_(position, 8, 0);
  1163. $.sigma_x = reader.float32_(position, 10, 0);
  1164. $.sigma_y = reader.float32_(position, 12, 0);
  1165. return $;
  1166. }
  1167. };
  1168. $root.mgb.serialization.fbs.param.ResizeV0 = class ResizeV0 {
  1169. static decode(reader, position) {
  1170. const $ = new $root.mgb.serialization.fbs.param.ResizeV0();
  1171. $.imode = reader.uint32_(position, 4, 1);
  1172. return $;
  1173. }
  1174. };
  1175. $root.mgb.serialization.fbs.param.ResizeV1 = class ResizeV1 {
  1176. static decode(reader, position) {
  1177. const $ = new $root.mgb.serialization.fbs.param.ResizeV1();
  1178. $.imode = reader.uint32_(position, 4, 1);
  1179. $.format = reader.uint32_(position, 6, 1);
  1180. return $;
  1181. }
  1182. };
  1183. $root.mgb.serialization.fbs.param.Resize = class Resize {
  1184. static decode(reader, position) {
  1185. const $ = new $root.mgb.serialization.fbs.param.Resize();
  1186. $.imode = reader.uint32_(position, 4, 1);
  1187. $.format = reader.uint32_(position, 6, 1);
  1188. return $;
  1189. }
  1190. };
  1191. $root.mgb.serialization.fbs.param.RemapV0 = class RemapV0 {
  1192. static decode(reader, position) {
  1193. const $ = new $root.mgb.serialization.fbs.param.RemapV0();
  1194. $.imode = reader.uint32_(position, 4, 1);
  1195. $.border_type = reader.uint32_(position, 6, 0);
  1196. $.format = reader.uint32_(position, 8, 1);
  1197. $.scalar = reader.float32_(position, 10, 0);
  1198. return $;
  1199. }
  1200. };
  1201. $root.mgb.serialization.fbs.param.Remap = class Remap {
  1202. static decode(reader, position) {
  1203. const $ = new $root.mgb.serialization.fbs.param.Remap();
  1204. $.imode = reader.uint32_(position, 4, 1);
  1205. $.border_type = reader.uint32_(position, 6, 0);
  1206. $.format = reader.uint32_(position, 8, 1);
  1207. $.scalar = reader.float32_(position, 10, 0);
  1208. return $;
  1209. }
  1210. };
  1211. $root.mgb.serialization.fbs.param.Convolution3D = class Convolution3D {
  1212. static decode(reader, position) {
  1213. const $ = new $root.mgb.serialization.fbs.param.Convolution3D();
  1214. $.mode = reader.uint32_(position, 4, 0);
  1215. $.pad_d = reader.uint32_(position, 6, 0);
  1216. $.pad_h = reader.uint32_(position, 8, 0);
  1217. $.pad_w = reader.uint32_(position, 10, 0);
  1218. $.stride_d = reader.uint32_(position, 12, 1);
  1219. $.stride_h = reader.uint32_(position, 14, 1);
  1220. $.stride_w = reader.uint32_(position, 16, 1);
  1221. $.dilate_d = reader.uint32_(position, 18, 1);
  1222. $.dilate_h = reader.uint32_(position, 20, 1);
  1223. $.dilate_w = reader.uint32_(position, 22, 1);
  1224. $.sparse = reader.uint32_(position, 24, 0);
  1225. $.data_type = reader.uint32_(position, 26, 0);
  1226. $.format = reader.uint32_(position, 28, 0);
  1227. return $;
  1228. }
  1229. };
  1230. $root.mgb.serialization.fbs.param.Conv3DBias = class Conv3DBias {
  1231. static decode(reader, position) {
  1232. const $ = new $root.mgb.serialization.fbs.param.Conv3DBias();
  1233. $.nonlineMode = reader.uint32_(position, 4, 0);
  1234. $.mode = reader.uint32_(position, 6, 0);
  1235. $.pad_d = reader.uint32_(position, 8, 0);
  1236. $.pad_h = reader.uint32_(position, 10, 0);
  1237. $.pad_w = reader.uint32_(position, 12, 0);
  1238. $.stride_d = reader.uint32_(position, 14, 1);
  1239. $.stride_h = reader.uint32_(position, 16, 1);
  1240. $.stride_w = reader.uint32_(position, 18, 0);
  1241. return $;
  1242. }
  1243. };
  1244. $root.mgb.serialization.fbs.param.SeparableConv3D = class SeparableConv3D {
  1245. static decode(reader, position) {
  1246. const $ = new $root.mgb.serialization.fbs.param.SeparableConv3D();
  1247. $.mode = reader.uint32_(position, 4, 0);
  1248. $.borderMode = reader.uint32_(position, 6, 0);
  1249. $.is_symm_kernel = reader.bool_(position, 8, true);
  1250. $.pad_d = reader.uint32_(position, 10, 0);
  1251. $.pad_h = reader.uint32_(position, 12, 0);
  1252. $.pad_w = reader.uint32_(position, 14, 0);
  1253. $.stride_d = reader.uint32_(position, 16, 0);
  1254. $.stride_h = reader.uint32_(position, 18, 1);
  1255. $.stride_w = reader.uint32_(position, 20, 1);
  1256. $.ksize_d = reader.uint32_(position, 22, 0);
  1257. $.ksize_h = reader.uint32_(position, 24, 3);
  1258. $.ksize_w = reader.uint32_(position, 26, 3);
  1259. $.anchor_d = reader.uint32_(position, 28, 0);
  1260. $.anchor_h = reader.uint32_(position, 30, 1);
  1261. $.anchor_w = reader.uint32_(position, 32, 1);
  1262. return $;
  1263. }
  1264. };
  1265. $root.mgb.serialization.fbs.param.TopK = class TopK {
  1266. static decode(reader, position) {
  1267. const $ = new $root.mgb.serialization.fbs.param.TopK();
  1268. $.mode = reader.uint32_(position, 4, 0);
  1269. return $;
  1270. }
  1271. };
  1272. $root.mgb.serialization.fbs.param.RelayoutFormatV0 = class RelayoutFormatV0 {
  1273. static decode(reader, position) {
  1274. const $ = new $root.mgb.serialization.fbs.param.RelayoutFormatV0();
  1275. $.mode = reader.uint32_(position, 4, 0);
  1276. return $;
  1277. }
  1278. };
  1279. $root.mgb.serialization.fbs.param.RelayoutFormat = class RelayoutFormat {
  1280. static decode(reader, position) {
  1281. const $ = new $root.mgb.serialization.fbs.param.RelayoutFormat();
  1282. $.mode = reader.uint32_(position, 4, 0);
  1283. $.oc = reader.uint32_(position, 6, 0);
  1284. $.group = reader.uint32_(position, 8, 1);
  1285. return $;
  1286. }
  1287. };
  1288. $root.mgb.serialization.fbs.param.SeparableFilterV0 = class SeparableFilterV0 {
  1289. static decode(reader, position) {
  1290. const $ = new $root.mgb.serialization.fbs.param.SeparableFilterV0();
  1291. $.format = reader.uint32_(position, 4, 0);
  1292. $.borderMode = reader.uint32_(position, 6, 0);
  1293. $.is_symm_kernel = reader.bool_(position, 8, true);
  1294. $.ksize_h = reader.uint32_(position, 10, 3);
  1295. $.ksize_w = reader.uint32_(position, 12, 3);
  1296. $.anchor_h = reader.uint32_(position, 14, 1);
  1297. $.anchor_w = reader.uint32_(position, 16, 1);
  1298. return $;
  1299. }
  1300. };
  1301. $root.mgb.serialization.fbs.param.SeparableFilter = class SeparableFilter {
  1302. static decode(reader, position) {
  1303. const $ = new $root.mgb.serialization.fbs.param.SeparableFilter();
  1304. $.format = reader.uint32_(position, 4, 0);
  1305. $.borderMode = reader.uint32_(position, 6, 0);
  1306. $.is_symm_kernel = reader.bool_(position, 8, true);
  1307. $.ksize_h = reader.uint32_(position, 10, 3);
  1308. $.ksize_w = reader.uint32_(position, 12, 3);
  1309. $.anchor_h = reader.uint32_(position, 14, 1);
  1310. $.anchor_w = reader.uint32_(position, 16, 1);
  1311. return $;
  1312. }
  1313. };
  1314. $root.mgb.serialization.fbs.param.LocalShareV0 = class LocalShareV0 {
  1315. static decode(reader, position) {
  1316. const $ = new $root.mgb.serialization.fbs.param.LocalShareV0();
  1317. $.mode = reader.uint32_(position, 4, 0);
  1318. $.pad_h = reader.uint32_(position, 6, 0);
  1319. $.pad_w = reader.uint32_(position, 8, 0);
  1320. $.stride_h = reader.uint32_(position, 10, 1);
  1321. $.stride_w = reader.uint32_(position, 12, 1);
  1322. $.dilate_h = reader.uint32_(position, 14, 1);
  1323. $.dilate_w = reader.uint32_(position, 16, 1);
  1324. $.spatial_groups_h = reader.uint32_(position, 18, 1);
  1325. $.spatial_groups_w = reader.uint32_(position, 20, 1);
  1326. $.sparse = reader.uint32_(position, 22, 0);
  1327. $.format = reader.uint32_(position, 24, 0);
  1328. $.computeMode = reader.uint32_(position, 26, 0);
  1329. return $;
  1330. }
  1331. };
  1332. $root.mgb.serialization.fbs.param.LocalShare = class LocalShare {
  1333. static decode(reader, position) {
  1334. const $ = new $root.mgb.serialization.fbs.param.LocalShare();
  1335. $.mode = reader.uint32_(position, 4, 0);
  1336. $.pad_h = reader.uint32_(position, 6, 0);
  1337. $.pad_w = reader.uint32_(position, 8, 0);
  1338. $.stride_h = reader.uint32_(position, 10, 1);
  1339. $.stride_w = reader.uint32_(position, 12, 1);
  1340. $.dilate_h = reader.uint32_(position, 14, 1);
  1341. $.dilate_w = reader.uint32_(position, 16, 1);
  1342. $.spatial_groups_h = reader.uint32_(position, 18, 1);
  1343. $.spatial_groups_w = reader.uint32_(position, 20, 1);
  1344. $.sparse = reader.uint32_(position, 22, 0);
  1345. $.format = reader.uint32_(position, 24, 0);
  1346. $.computeMode = reader.uint32_(position, 26, 0);
  1347. return $;
  1348. }
  1349. };
  1350. $root.mgb.serialization.fbs.param.ROIAlignV0 = class ROIAlignV0 {
  1351. static decode(reader, position) {
  1352. const $ = new $root.mgb.serialization.fbs.param.ROIAlignV0();
  1353. $.mode = reader.uint32_(position, 4, 0);
  1354. $.format = reader.uint32_(position, 6, 0);
  1355. $.spatial_scale = reader.float32_(position, 8, 1);
  1356. $.offset = reader.float32_(position, 10, 0);
  1357. $.pooled_height = reader.uint32_(position, 12, 1);
  1358. $.pooled_width = reader.uint32_(position, 14, 1);
  1359. $.sample_height = reader.uint32_(position, 16, 2);
  1360. $.sample_width = reader.uint32_(position, 18, 2);
  1361. return $;
  1362. }
  1363. };
  1364. $root.mgb.serialization.fbs.param.ROIAlign = class ROIAlign {
  1365. static decode(reader, position) {
  1366. const $ = new $root.mgb.serialization.fbs.param.ROIAlign();
  1367. $.mode = reader.uint32_(position, 4, 0);
  1368. $.format = reader.uint32_(position, 6, 0);
  1369. $.spatial_scale = reader.float32_(position, 8, 1);
  1370. $.offset = reader.float32_(position, 10, 0);
  1371. $.pooled_height = reader.uint32_(position, 12, 1);
  1372. $.pooled_width = reader.uint32_(position, 14, 1);
  1373. $.sample_height = reader.uint32_(position, 16, 2);
  1374. $.sample_width = reader.uint32_(position, 18, 2);
  1375. return $;
  1376. }
  1377. };
  1378. $root.mgb.serialization.fbs.param.Correlation = class Correlation {
  1379. static decode(reader, position) {
  1380. const $ = new $root.mgb.serialization.fbs.param.Correlation();
  1381. $.format = reader.uint32_(position, 4, 0);
  1382. $.kernel_size = reader.uint32_(position, 6, 1);
  1383. $.max_displacement = reader.uint32_(position, 8, 1);
  1384. $.stride1 = reader.uint32_(position, 10, 1);
  1385. $.stride2 = reader.uint32_(position, 12, 1);
  1386. $.pad_size = reader.uint32_(position, 14, 0);
  1387. $.is_multiply = reader.bool_(position, 16, true);
  1388. return $;
  1389. }
  1390. };
  1391. $root.mgb.serialization.fbs.param.DeformablePSROIPooling = class DeformablePSROIPooling {
  1392. static decode(reader, position) {
  1393. const $ = new $root.mgb.serialization.fbs.param.DeformablePSROIPooling();
  1394. $.no_trans = reader.bool_(position, 4, true);
  1395. $.spatial_scale = reader.float32_(position, 6, 1);
  1396. $.trans_std = reader.float32_(position, 8, 1);
  1397. $.pooled_h = reader.uint32_(position, 10, 1);
  1398. $.pooled_w = reader.uint32_(position, 12, 1);
  1399. $.part_size = reader.uint32_(position, 14, 1);
  1400. $.sample_per_part = reader.uint32_(position, 16, 1);
  1401. return $;
  1402. }
  1403. };
  1404. $root.mgb.serialization.fbs.param.BatchConvBiasV0 = class BatchConvBiasV0 {
  1405. static decode(reader, position) {
  1406. const $ = new $root.mgb.serialization.fbs.param.BatchConvBiasV0();
  1407. $.nonlineMode = reader.uint32_(position, 4, 0);
  1408. $.mode = reader.uint32_(position, 6, 0);
  1409. $.pad_h = reader.uint32_(position, 8, 0);
  1410. $.pad_w = reader.uint32_(position, 10, 0);
  1411. $.stride_h = reader.uint32_(position, 12, 1);
  1412. $.stride_w = reader.uint32_(position, 14, 1);
  1413. $.dilate_h = reader.uint32_(position, 16, 1);
  1414. $.dilate_w = reader.uint32_(position, 18, 1);
  1415. $.sparse = reader.uint32_(position, 20, 0);
  1416. $.format = reader.uint32_(position, 22, 0);
  1417. $.compute_mode = reader.uint32_(position, 24, 0);
  1418. return $;
  1419. }
  1420. };
  1421. $root.mgb.serialization.fbs.param.BatchConvBias = class BatchConvBias {
  1422. static decode(reader, position) {
  1423. const $ = new $root.mgb.serialization.fbs.param.BatchConvBias();
  1424. $.nonlineMode = reader.uint32_(position, 4, 0);
  1425. $.mode = reader.uint32_(position, 6, 0);
  1426. $.pad_h = reader.uint32_(position, 8, 0);
  1427. $.pad_w = reader.uint32_(position, 10, 0);
  1428. $.stride_h = reader.uint32_(position, 12, 1);
  1429. $.stride_w = reader.uint32_(position, 14, 1);
  1430. $.dilate_h = reader.uint32_(position, 16, 1);
  1431. $.dilate_w = reader.uint32_(position, 18, 1);
  1432. $.sparse = reader.uint32_(position, 20, 0);
  1433. $.format = reader.uint32_(position, 22, 0);
  1434. $.compute_mode = reader.uint32_(position, 24, 0);
  1435. return $;
  1436. }
  1437. };
  1438. $root.mgb.serialization.fbs.param.FakeQuant = class FakeQuant {
  1439. static decode(reader, position) {
  1440. const $ = new $root.mgb.serialization.fbs.param.FakeQuant();
  1441. $.qmin = reader.int32_(position, 4, -2147483648);
  1442. $.qmax = reader.int32_(position, 6, 2147483647);
  1443. return $;
  1444. }
  1445. };
  1446. $root.mgb.serialization.fbs.param.TQT = class TQT {
  1447. static decode(reader, position) {
  1448. const $ = new $root.mgb.serialization.fbs.param.TQT();
  1449. $.qmin = reader.int32_(position, 4, -2147483648);
  1450. $.qmax = reader.int32_(position, 6, 2147483647);
  1451. return $;
  1452. }
  1453. };
  1454. $root.mgb.serialization.fbs.param.LSQ = class LSQ {
  1455. static decode(reader, position) {
  1456. const $ = new $root.mgb.serialization.fbs.param.LSQ();
  1457. $.qmin = reader.int32_(position, 4, -2147483648);
  1458. $.qmax = reader.int32_(position, 6, 2147483647);
  1459. return $;
  1460. }
  1461. };
  1462. $root.mgb.serialization.fbs.param.Fill = class Fill {
  1463. static decode(reader, position) {
  1464. const $ = new $root.mgb.serialization.fbs.param.Fill();
  1465. $.value = reader.float32_(position, 4, 0);
  1466. return $;
  1467. }
  1468. };
  1469. $root.mgb.serialization.fbs.param.CheckNonFinite = class CheckNonFinite {
  1470. static decode(reader, position) {
  1471. const $ = new $root.mgb.serialization.fbs.param.CheckNonFinite();
  1472. $.scale = reader.float32_(position, 4, 1);
  1473. return $;
  1474. }
  1475. };
  1476. $root.mgb.serialization.fbs.param.Padding = class Padding {
  1477. static decode(reader, position) {
  1478. const $ = new $root.mgb.serialization.fbs.param.Padding();
  1479. $.front_offset_dim0 = reader.uint32_(position, 4, 0);
  1480. $.front_offset_dim1 = reader.uint32_(position, 6, 0);
  1481. $.front_offset_dim2 = reader.uint32_(position, 8, 0);
  1482. $.front_offset_dim3 = reader.uint32_(position, 10, 0);
  1483. $.front_offset_dim4 = reader.uint32_(position, 12, 0);
  1484. $.front_offset_dim5 = reader.uint32_(position, 14, 0);
  1485. $.front_offset_dim6 = reader.uint32_(position, 16, 0);
  1486. $.back_offset_dim0 = reader.uint32_(position, 18, 0);
  1487. $.back_offset_dim1 = reader.uint32_(position, 20, 0);
  1488. $.back_offset_dim2 = reader.uint32_(position, 22, 0);
  1489. $.back_offset_dim3 = reader.uint32_(position, 24, 0);
  1490. $.back_offset_dim4 = reader.uint32_(position, 26, 0);
  1491. $.back_offset_dim5 = reader.uint32_(position, 28, 0);
  1492. $.back_offset_dim6 = reader.uint32_(position, 30, 0);
  1493. $.padding_val = reader.float32_(position, 32, 0);
  1494. $.padding_mode = reader.uint32_(position, 34, 2);
  1495. return $;
  1496. }
  1497. };
  1498. $root.mgb.serialization.fbs.param.LayerNorm = class LayerNorm {
  1499. static decode(reader, position) {
  1500. const $ = new $root.mgb.serialization.fbs.param.LayerNorm();
  1501. $.affine = reader.bool_(position, 4, true);
  1502. $.eps = reader.float32_(position, 6, 0.00001);
  1503. $.normalized_dim = reader.uint64_(position, 8, 1);
  1504. $.normalized_size = reader.uint64_(position, 10, 1);
  1505. return $;
  1506. }
  1507. };
  1508. $root.mgb.serialization.fbs.param.GroupNorm = class GroupNorm {
  1509. static decode(reader, position) {
  1510. const $ = new $root.mgb.serialization.fbs.param.GroupNorm();
  1511. $.affine = reader.bool_(position, 4, true);
  1512. $.eps = reader.float32_(position, 6, 0.00001);
  1513. $.group = reader.uint32_(position, 8, 1);
  1514. $.format = reader.uint32_(position, 10, 0);
  1515. return $;
  1516. }
  1517. };
  1518. $root.mgb.serialization.fbs.param.Dropout = class Dropout {
  1519. static decode(reader, position) {
  1520. const $ = new $root.mgb.serialization.fbs.param.Dropout();
  1521. $.drop_prob = reader.float32_(position, 4, 0);
  1522. $.seed = reader.uint64_(position, 6, 0);
  1523. return $;
  1524. }
  1525. };
  1526. $root.mgb.serialization.fbs.param.RNNCell = class RNNCell {
  1527. static decode(reader, position) {
  1528. const $ = new $root.mgb.serialization.fbs.param.RNNCell();
  1529. $.nonlineMode = reader.uint32_(position, 4, 0);
  1530. return $;
  1531. }
  1532. };
  1533. $root.mgb.serialization.fbs.param.RNN = class RNN {
  1534. static decode(reader, position) {
  1535. const $ = new $root.mgb.serialization.fbs.param.RNN();
  1536. $.num_layers = reader.uint32_(position, 4, 1);
  1537. $.bidirectional = reader.bool_(position, 6, false);
  1538. $.bias = reader.bool_(position, 8, true);
  1539. $.hidden_size = reader.uint32_(position, 10, 128);
  1540. $.dropout = reader.float32_(position, 12, 0);
  1541. $.nonlineMode = reader.uint32_(position, 14, 0);
  1542. $.fwd_mode = reader.uint32_(position, 16, 0);
  1543. return $;
  1544. }
  1545. };
  1546. $root.mgb.serialization.fbs.param.LSTM = class LSTM {
  1547. static decode(reader, position) {
  1548. const $ = new $root.mgb.serialization.fbs.param.LSTM();
  1549. $.num_layers = reader.uint32_(position, 4, 1);
  1550. $.bidirectional = reader.bool_(position, 6, false);
  1551. $.bias = reader.bool_(position, 8, true);
  1552. $.hidden_size = reader.uint32_(position, 10, 128);
  1553. $.proj_size = reader.uint32_(position, 12, 0);
  1554. $.dropout = reader.float32_(position, 14, 0);
  1555. $.fwd_mode = reader.uint32_(position, 16, 0);
  1556. return $;
  1557. }
  1558. };
  1559. $root.mgb.serialization.fbs.param.CollectiveCommMode = {
  1560. REDUCE_SUM: 0,
  1561. BROADCAST: 1,
  1562. ALL_GATHER: 2,
  1563. REDUCE_SCATTER_SUM: 3,
  1564. ALL_REDUCE_SUM: 4,
  1565. ALL_REDUCE_MAX: 5,
  1566. ALL_REDUCE_MIN: 6,
  1567. ALL_REDUCE_PROD: 7,
  1568. GATHER: 8,
  1569. SCATTER: 9,
  1570. ALL_TO_ALL: 10
  1571. };
  1572. $root.mgb.serialization.fbs.param.CondExecMarkGradMode = {
  1573. SUM: 0,
  1574. SUM_COND_OUT: 1
  1575. };
  1576. $root.mgb.serialization.fbs.param.CondExecMarkStaticInfer = {
  1577. SHAPE_VALUE: 0,
  1578. SHAPE_ONLY: 1,
  1579. NONE: 2
  1580. };
  1581. $root.mgb.serialization.fbs.param.CondExecMergeMode = {
  1582. EXACT_ONE: 0,
  1583. EXACT_ONE_SAME_SHAPE: 1,
  1584. SUM: 2,
  1585. SUM_COND_OUT: 3
  1586. };
  1587. $root.mgb.serialization.fbs.param.CondExecPredMode = {
  1588. CASE: 0,
  1589. CASE_FALLBACK: 1,
  1590. PIECEWISE: 2
  1591. };
  1592. $root.mgb.serialization.fbs.param.CondExecPredLogicalMode = {
  1593. OR: 0,
  1594. AND: 1,
  1595. XOR: 2,
  1596. NOR: 3,
  1597. NAND: 4,
  1598. XNOR: 5
  1599. };
  1600. $root.mgb.serialization.fbs.param.ExecutionPolicyStrategy = {
  1601. HEURISTIC: 0,
  1602. PROFILE: 1,
  1603. REPRODUCIBLE: 2,
  1604. OPTIMIZED: 3
  1605. };
  1606. $root.mgb.serialization.fbs.param.ExecutionPolicyV0Strategy = {
  1607. HEURISTIC: 0,
  1608. HEURISTIC_REPRODUCIBLE: 1,
  1609. PROFILE: 2,
  1610. PROFILE_REPRODUCIBLE: 3,
  1611. PROFILE_HEURISTIC: 4
  1612. };
  1613. $root.mgb.serialization.fbs.param.DType = class DType {
  1614. static decode(reader, position) {
  1615. const $ = new $root.mgb.serialization.fbs.param.DType();
  1616. $.dtype = reader.int8_(position, 4, 8);
  1617. return $;
  1618. }
  1619. };
  1620. $root.mgb.serialization.fbs.param.PersistentOutputStorage = class PersistentOutputStorage {
  1621. static decode(reader, position) {
  1622. const $ = new $root.mgb.serialization.fbs.param.PersistentOutputStorage();
  1623. $.share_key = reader.int32_(position, 4, -1);
  1624. return $;
  1625. }
  1626. };
  1627. $root.mgb.serialization.fbs.param.OptionalAxis = class OptionalAxis {
  1628. static decode(reader, position) {
  1629. const $ = new $root.mgb.serialization.fbs.param.OptionalAxis();
  1630. $.axis = reader.int32_(position, 4, -1);
  1631. return $;
  1632. }
  1633. };
  1634. $root.mgb.serialization.fbs.param.OptionalAxisV1 = class OptionalAxisV1 {
  1635. static decode(reader, position) {
  1636. const $ = new $root.mgb.serialization.fbs.param.OptionalAxisV1();
  1637. $.axis = reader.int32_(position, 4, 7);
  1638. return $;
  1639. }
  1640. };
  1641. $root.mgb.serialization.fbs.param.ExecutionPolicyV0 = class ExecutionPolicyV0 {
  1642. static decode(reader, position) {
  1643. const $ = new $root.mgb.serialization.fbs.param.ExecutionPolicyV0();
  1644. $.strategy = reader.uint32_(position, 4, 0);
  1645. $.workspace_limit = reader.uint64_(position, 6, 18446744073709552000);
  1646. return $;
  1647. }
  1648. };
  1649. $root.mgb.serialization.fbs.param.ExecutionPolicy = class ExecutionPolicy {
  1650. static decode(reader, position) {
  1651. const $ = new $root.mgb.serialization.fbs.param.ExecutionPolicy();
  1652. $.strategy = reader.uint32_(position, 4, 1);
  1653. $.workspace_limit = reader.uint64_(position, 6, 18446744073709552000);
  1654. return $;
  1655. }
  1656. };
  1657. $root.mgb.serialization.fbs.param.AssertEqual = class AssertEqual {
  1658. static decode(reader, position) {
  1659. const $ = new $root.mgb.serialization.fbs.param.AssertEqual();
  1660. $.maxerr = reader.float32_(position, 4, 0.0001);
  1661. $.verbose = reader.bool_(position, 6, false);
  1662. return $;
  1663. }
  1664. };
  1665. $root.mgb.serialization.fbs.param.FpgaConv = class FpgaConv {
  1666. static decode(reader, position) {
  1667. const $ = new $root.mgb.serialization.fbs.param.FpgaConv();
  1668. $.need_output_quantize = reader.bool_(position, 4, false);
  1669. $.need_output_threshold = reader.bool_(position, 6, false);
  1670. $.stride = reader.int32_(position, 8, 1);
  1671. $.input_bit_width = reader.int32_(position, 10, 2);
  1672. $.output_bit_width = reader.int32_(position, 12, 2);
  1673. $.weight_bit_width = reader.int32_(position, 14, 2);
  1674. $.thres0 = reader.int32_(position, 16, 0);
  1675. $.thres1 = reader.int32_(position, 18, 1);
  1676. $.unpool_size = reader.uint32_(position, 20, 4);
  1677. $.direct_size = reader.uint32_(position, 22, 4);
  1678. return $;
  1679. }
  1680. };
  1681. $root.mgb.serialization.fbs.param.CollectiveComm = class CollectiveComm {
  1682. static decode(reader, position) {
  1683. const $ = new $root.mgb.serialization.fbs.param.CollectiveComm();
  1684. $.mode = reader.uint32_(position, 4, 0);
  1685. return $;
  1686. }
  1687. };
  1688. $root.mgb.serialization.fbs.param.FakeSerializedDType = class FakeSerializedDType {
  1689. static decode(/* reader, position */) {
  1690. const $ = new $root.mgb.serialization.fbs.param.FakeSerializedDType();
  1691. return $;
  1692. }
  1693. };
  1694. $root.mgb.serialization.fbs.param.CondExecPred = class CondExecPred {
  1695. static decode(reader, position) {
  1696. const $ = new $root.mgb.serialization.fbs.param.CondExecPred();
  1697. $.mode = reader.uint32_(position, 4, 0);
  1698. $.eps = reader.float32_(position, 6, 0.0001);
  1699. return $;
  1700. }
  1701. };
  1702. $root.mgb.serialization.fbs.param.CondExecPredLogical = class CondExecPredLogical {
  1703. static decode(reader, position) {
  1704. const $ = new $root.mgb.serialization.fbs.param.CondExecPredLogical();
  1705. $.mode = reader.uint32_(position, 4, 0);
  1706. return $;
  1707. }
  1708. };
  1709. $root.mgb.serialization.fbs.param.CondExecMark = class CondExecMark {
  1710. static decode(reader, position) {
  1711. const $ = new $root.mgb.serialization.fbs.param.CondExecMark();
  1712. $.grad_mode = reader.uint32_(position, 4, 0);
  1713. $.static_infer = reader.uint32_(position, 6, 0);
  1714. return $;
  1715. }
  1716. };
  1717. $root.mgb.serialization.fbs.param.CondExecMerge = class CondExecMerge {
  1718. static decode(reader, position) {
  1719. const $ = new $root.mgb.serialization.fbs.param.CondExecMerge();
  1720. $.nr_output = reader.uint32_(position, 4, 1);
  1721. $.mode = reader.uint32_(position, 6, 0);
  1722. return $;
  1723. }
  1724. };
  1725. $root.mgb.serialization.fbs.param.NvOf = class NvOf {
  1726. static decode(reader, position) {
  1727. const $ = new $root.mgb.serialization.fbs.param.NvOf();
  1728. $.precision = reader.uint32_(position, 4, 1);
  1729. return $;
  1730. }
  1731. };
  1732. $root.mgb.serialization.fbs.param.PersistentDTypeScalar = class PersistentDTypeScalar {
  1733. static decode(reader, position) {
  1734. const $ = new $root.mgb.serialization.fbs.param.PersistentDTypeScalar();
  1735. $.dtype = reader.int8(position + 0);
  1736. $.storage = undefined; // not implemented
  1737. return $;
  1738. }
  1739. };
  1740. $root.mgb.serialization.fbs.param.MGBAddUpdate = class MGBAddUpdate {
  1741. static decode(reader, position) {
  1742. const $ = new $root.mgb.serialization.fbs.param.MGBAddUpdate();
  1743. $.alpha = reader.struct(position, 4, $root.mgb.serialization.fbs.param.PersistentDTypeScalar.decode);
  1744. $.beta = reader.struct(position, 6, $root.mgb.serialization.fbs.param.PersistentDTypeScalar.decode);
  1745. $.bias = reader.struct(position, 8, $root.mgb.serialization.fbs.param.PersistentDTypeScalar.decode);
  1746. return $;
  1747. }
  1748. };
  1749. $root.mgb.serialization.fbs.param.Host2DeviceCopy = class Host2DeviceCopy {
  1750. static decode(reader, position) {
  1751. const $ = new $root.mgb.serialization.fbs.param.Host2DeviceCopy();
  1752. $.enable_value_infer = reader.bool_(position, 4, true);
  1753. $.dump_default_value = reader.bool_(position, 6, false);
  1754. $.allow_cpu_mem_fwd = reader.bool_(position, 8, true);
  1755. return $;
  1756. }
  1757. };
  1758. $root.mgb.serialization.fbs.param.Dimshuffle = class Dimshuffle {
  1759. static decode(reader, position) {
  1760. const $ = new $root.mgb.serialization.fbs.param.Dimshuffle();
  1761. $.pattern = reader.typedArray(position, 4, Int32Array);
  1762. $.ndim = reader.uint32_(position, 6, 0);
  1763. return $;
  1764. }
  1765. };
  1766. $root.mgb.serialization.fbs.param.AxisDescMethod = {
  1767. ADD_1: 0,
  1768. REMOVE: 1
  1769. };
  1770. $root.mgb.serialization.fbs.param.AxisDesc = class AxisDesc {
  1771. static decode(reader, position) {
  1772. const $ = new $root.mgb.serialization.fbs.param.AxisDesc();
  1773. $.method = reader.int8(position + 0);
  1774. $.axis = reader.int32(position + 4);
  1775. return $;
  1776. }
  1777. };
  1778. $root.mgb.serialization.fbs.param.AxisAddRemove = class AxisAddRemove {
  1779. static decode(reader, position) {
  1780. const $ = new $root.mgb.serialization.fbs.param.AxisAddRemove();
  1781. $.desc = reader.structArray(position, 4, $root.mgb.serialization.fbs.param.AxisDesc.decode);
  1782. return $;
  1783. }
  1784. };
  1785. $root.mgb.serialization.fbs.param.MGBSleep = class MGBSleep {
  1786. static decode(reader, position) {
  1787. const $ = new $root.mgb.serialization.fbs.param.MGBSleep();
  1788. $.device = reader.bool_(position, 4, true);
  1789. $.host = reader.bool_(position, 6, false);
  1790. $.seconds = reader.float64_(position, 8, 0);
  1791. return $;
  1792. }
  1793. };
  1794. $root.mgb.serialization.fbs.param.IndexDescMaskItem = class IndexDescMaskItem {
  1795. static decode(reader, position) {
  1796. const $ = new $root.mgb.serialization.fbs.param.IndexDescMaskItem();
  1797. $.axis = reader.int8(position + 0);
  1798. $.begin = reader.bool(position + 1);
  1799. $.end = reader.bool(position + 2);
  1800. $.step = reader.bool(position + 3);
  1801. $.idx = reader.bool(position + 4);
  1802. return $;
  1803. }
  1804. };
  1805. $root.mgb.serialization.fbs.param.IndexDescMaskDump = class IndexDescMaskDump {
  1806. static decode(reader, position) {
  1807. const $ = new $root.mgb.serialization.fbs.param.IndexDescMaskDump();
  1808. $.items = reader.structArray(position, 4, $root.mgb.serialization.fbs.param.IndexDescMaskItem.decode);
  1809. return $;
  1810. }
  1811. };
  1812. $root.mgb.serialization.fbs.param.NMSKeep = class NMSKeep {
  1813. static decode(reader, position) {
  1814. const $ = new $root.mgb.serialization.fbs.param.NMSKeep();
  1815. $.iou_thresh = reader.float32_(position, 4, 0);
  1816. $.max_output = reader.uint32_(position, 6, 0);
  1817. return $;
  1818. }
  1819. };
  1820. $root.mgb = $root.mgb || {};
  1821. $root.mgb.serialization = $root.mgb.serialization || {};
  1822. $root.mgb.serialization.fbs = $root.mgb.serialization.fbs || {};
  1823. $root.mgb.serialization.fbs.v2 = $root.mgb.serialization.fbs.v2 || {};
  1824. $root.mgb.serialization.fbs.v2.CompNode = class CompNode {
  1825. static decode(reader, position) {
  1826. const $ = new $root.mgb.serialization.fbs.v2.CompNode();
  1827. $.logical_locator = reader.string_(position, 4, null);
  1828. return $;
  1829. }
  1830. };
  1831. $root.mgb.serialization.fbs.v2.DefaultTensorFormat = class DefaultTensorFormat {
  1832. static decode(/* reader, position */) {
  1833. const $ = new $root.mgb.serialization.fbs.v2.DefaultTensorFormat();
  1834. return $;
  1835. }
  1836. };
  1837. $root.mgb.serialization.fbs.v2.Image2DPackedTensorFormat = class Image2DPackedTensorFormat {
  1838. static decode(reader, position) {
  1839. const $ = new $root.mgb.serialization.fbs.v2.Image2DPackedTensorFormat();
  1840. $.align_axis = reader.uint8_(position, 4, 0);
  1841. return $;
  1842. }
  1843. };
  1844. $root.mgb.serialization.fbs.v2.LowbitsAlignedTensorFormat = class LowbitsAlignedTensorFormat {
  1845. static decode(reader, position) {
  1846. const $ = new $root.mgb.serialization.fbs.v2.LowbitsAlignedTensorFormat();
  1847. $.size_nbits = reader.uint8_(position, 4, 0);
  1848. $.align_size_in_bits = reader.uint8_(position, 6, 0);
  1849. return $;
  1850. }
  1851. };
  1852. $root.mgb.serialization.fbs.v2.TensorFormat = class {
  1853. static decode(reader, position, type) {
  1854. switch (type) {
  1855. case 1: return $root.mgb.serialization.fbs.v2.DefaultTensorFormat.decode(reader, position);
  1856. case 2: return $root.mgb.serialization.fbs.v2.Image2DPackedTensorFormat.decode(reader, position);
  1857. case 3: return $root.mgb.serialization.fbs.v2.LowbitsAlignedTensorFormat.decode(reader, position);
  1858. default: return undefined;
  1859. }
  1860. }
  1861. };
  1862. $root.mgb.serialization.fbs.v2.Blob = class Blob {
  1863. static decode(reader, position) {
  1864. const $ = new $root.mgb.serialization.fbs.v2.Blob();
  1865. $.data = reader.typedArray(position, 4, Uint8Array);
  1866. return $;
  1867. }
  1868. };
  1869. $root.mgb.serialization.fbs.v2.Tensor = class Tensor {
  1870. static decode(reader, position) {
  1871. const $ = new $root.mgb.serialization.fbs.v2.Tensor();
  1872. $.name = reader.string_(position, 4, null);
  1873. $.shape = reader.typedArray(position, 6, Uint32Array);
  1874. $.comp_node = reader.table(position, 8, $root.mgb.serialization.fbs.v2.CompNode.decode);
  1875. $.dtype = reader.table(position, 10, $root.mgb.serialization.fbs.DType.decode);
  1876. $.format = reader.union(position, 12, $root.mgb.serialization.fbs.v2.TensorFormat.decode);
  1877. $.data = reader.typedArray(position, 16, Uint8Array);
  1878. return $;
  1879. }
  1880. };
  1881. $root.mgb.serialization.fbs.v2.Reserved0 = class Reserved0 {
  1882. static decode(/* reader, position */) {
  1883. const $ = new $root.mgb.serialization.fbs.v2.Reserved0();
  1884. return $;
  1885. }
  1886. };
  1887. $root.mgb.serialization.fbs.v2.DeprecatedParam = class DeprecatedParam {
  1888. static decode(/* reader, position */) {
  1889. const $ = new $root.mgb.serialization.fbs.v2.DeprecatedParam();
  1890. return $;
  1891. }
  1892. };
  1893. $root.mgb.serialization.fbs.v2.OperatorParam = class {
  1894. static decode(reader, position, type) {
  1895. switch (type) {
  1896. case 1: return $root.mgb.serialization.fbs.param.Empty.decode(reader, position);
  1897. case 2: return $root.mgb.serialization.fbs.param.Axis.decode(reader, position);
  1898. case 3: return $root.mgb.serialization.fbs.param.Convolution.decode(reader, position);
  1899. case 4: return $root.mgb.serialization.fbs.param.MaskPropagate.decode(reader, position);
  1900. case 5: return $root.mgb.serialization.fbs.param.ConvPooling.decode(reader, position);
  1901. case 6: return $root.mgb.serialization.fbs.param.ConvBias.decode(reader, position);
  1902. case 7: return $root.mgb.serialization.fbs.param.SeparableConv.decode(reader, position);
  1903. case 8: return $root.mgb.serialization.fbs.param.Images2Neibs.decode(reader, position);
  1904. case 9: return $root.mgb.serialization.fbs.param.Pooling.decode(reader, position);
  1905. case 10: return $root.mgb.serialization.fbs.param.LRN.decode(reader, position);
  1906. case 11: return $root.mgb.serialization.fbs.param.BN.decode(reader, position);
  1907. case 12: return $root.mgb.serialization.fbs.param.ROIPooling.decode(reader, position);
  1908. case 13: return $root.mgb.serialization.fbs.param.WarpPerspective.decode(reader, position);
  1909. case 14: return $root.mgb.serialization.fbs.param.SpatialTfGridGenerator.decode(reader, position);
  1910. case 15: return $root.mgb.serialization.fbs.param.SpatialTfSampler.decode(reader, position);
  1911. case 16: return $root.mgb.serialization.fbs.param.MGBAddUpdate.decode(reader, position);
  1912. case 17: return $root.mgb.serialization.fbs.param.Elemwise.decode(reader, position);
  1913. case 18: return $root.mgb.serialization.fbs.param.ElemwiseMultiType.decode(reader, position);
  1914. case 19: return $root.mgb.serialization.fbs.param.PowC.decode(reader, position);
  1915. case 20: return $root.mgb.serialization.fbs.param.MatrixMul.decode(reader, position);
  1916. case 21: return $root.mgb.serialization.fbs.v2.DeprecatedParam.decode(reader, position);
  1917. case 22: return $root.mgb.serialization.fbs.param.SVD.decode(reader, position);
  1918. case 23: return $root.mgb.serialization.fbs.param.Reduce.decode(reader, position);
  1919. case 24: return $root.mgb.serialization.fbs.param.Cumsum.decode(reader, position);
  1920. case 25: return $root.mgb.serialization.fbs.param.CondTake.decode(reader, position);
  1921. case 26: return $root.mgb.serialization.fbs.param.Argsort.decode(reader, position);
  1922. case 27: return $root.mgb.serialization.fbs.param.IndexingRemap.decode(reader, position);
  1923. case 28: return $root.mgb.serialization.fbs.param.MGBSleep.decode(reader, position);
  1924. case 29: return $root.mgb.serialization.fbs.param.Linspace.decode(reader, position);
  1925. case 30: return $root.mgb.serialization.fbs.param.LinspaceFull.decode(reader, position);
  1926. case 31: return $root.mgb.serialization.fbs.param.Eye.decode(reader, position);
  1927. case 32: return $root.mgb.serialization.fbs.param.UniformRNG.decode(reader, position);
  1928. case 33: return $root.mgb.serialization.fbs.param.GaussianRNG.decode(reader, position);
  1929. case 34: return $root.mgb.serialization.fbs.param.Flip.decode(reader, position);
  1930. case 35: return $root.mgb.serialization.fbs.param.Rotate.decode(reader, position);
  1931. case 36: return $root.mgb.serialization.fbs.param.ROICopy.decode(reader, position);
  1932. case 37: return $root.mgb.serialization.fbs.param.CvtColor.decode(reader, position);
  1933. case 38: return $root.mgb.serialization.fbs.param.WarpAffine.decode(reader, position);
  1934. case 39: return $root.mgb.serialization.fbs.param.GaussianBlur.decode(reader, position);
  1935. case 40: return $root.mgb.serialization.fbs.param.Resize.decode(reader, position);
  1936. case 41: return $root.mgb.serialization.fbs.param.Convolution3D.decode(reader, position);
  1937. case 42: return $root.mgb.serialization.fbs.param.Conv3DBias.decode(reader, position);
  1938. case 43: return $root.mgb.serialization.fbs.param.SeparableConv3D.decode(reader, position);
  1939. case 44: return $root.mgb.serialization.fbs.param.TopK.decode(reader, position);
  1940. case 45: return $root.mgb.serialization.fbs.param.RelayoutFormat.decode(reader, position);
  1941. case 46: return $root.mgb.serialization.fbs.param.SeparableFilter.decode(reader, position);
  1942. case 47: return $root.mgb.serialization.fbs.param.LocalShare.decode(reader, position);
  1943. case 48: return $root.mgb.serialization.fbs.param.ROIAlign.decode(reader, position);
  1944. case 49: return $root.mgb.serialization.fbs.param.DeformablePSROIPooling.decode(reader, position);
  1945. case 50: return $root.mgb.serialization.fbs.param.BatchConvBias.decode(reader, position);
  1946. case 51: return $root.mgb.serialization.fbs.param.DType.decode(reader, position);
  1947. case 52: return $root.mgb.serialization.fbs.param.PersistentOutputStorage.decode(reader, position);
  1948. case 53: return $root.mgb.serialization.fbs.param.OptionalAxis.decode(reader, position);
  1949. case 54: return $root.mgb.serialization.fbs.param.OptionalAxisV1.decode(reader, position);
  1950. case 55: return $root.mgb.serialization.fbs.param.ExecutionPolicy.decode(reader, position);
  1951. case 56: return $root.mgb.serialization.fbs.param.AssertEqual.decode(reader, position);
  1952. case 57: return $root.mgb.serialization.fbs.param.FpgaConv.decode(reader, position);
  1953. case 58: return $root.mgb.serialization.fbs.param.CollectiveComm.decode(reader, position);
  1954. case 59: return $root.mgb.serialization.fbs.param.CondExecPred.decode(reader, position);
  1955. case 60: return $root.mgb.serialization.fbs.param.CondExecPredLogical.decode(reader, position);
  1956. case 61: return $root.mgb.serialization.fbs.param.CondExecMark.decode(reader, position);
  1957. case 62: return $root.mgb.serialization.fbs.param.CondExecMerge.decode(reader, position);
  1958. case 63: return $root.mgb.serialization.fbs.param.Host2DeviceCopy.decode(reader, position);
  1959. case 64: return $root.mgb.serialization.fbs.param.Dimshuffle.decode(reader, position);
  1960. case 65: return $root.mgb.serialization.fbs.param.AxisAddRemove.decode(reader, position);
  1961. case 66: return $root.mgb.serialization.fbs.param.IndexDescMaskDump.decode(reader, position);
  1962. case 67: return $root.mgb.serialization.fbs.DType.decode(reader, position);
  1963. case 68: return $root.mgb.serialization.fbs.param.Remap.decode(reader, position);
  1964. case 69: return $root.mgb.serialization.fbs.param.NMSKeep.decode(reader, position);
  1965. case 70: return $root.mgb.serialization.fbs.param.AdaptivePooling.decode(reader, position);
  1966. case 71: return $root.mgb.serialization.fbs.param.NvOf.decode(reader, position);
  1967. case 72: return $root.mgb.serialization.fbs.param.DctChannelSelect.decode(reader, position);
  1968. case 73: return $root.mgb.serialization.fbs.param.FakeQuant.decode(reader, position);
  1969. case 74: return $root.mgb.serialization.fbs.param.TQT.decode(reader, position);
  1970. case 75: return $root.mgb.serialization.fbs.param.Correlation.decode(reader, position);
  1971. case 76: return $root.mgb.serialization.fbs.param.LSQ.decode(reader, position);
  1972. case 77: return $root.mgb.serialization.fbs.param.GammaRNG.decode(reader, position);
  1973. case 78: return $root.mgb.serialization.fbs.param.PoissonRNG.decode(reader, position);
  1974. case 79: return $root.mgb.serialization.fbs.param.PermutationRNG.decode(reader, position);
  1975. case 80: return $root.mgb.serialization.fbs.param.BetaRNG.decode(reader, position);
  1976. case 81: return $root.mgb.serialization.fbs.param.SlidingWindowTranspose.decode(reader, position);
  1977. case 82: return $root.mgb.serialization.fbs.param.Padding.decode(reader, position);
  1978. case 83: return $root.mgb.serialization.fbs.param.ShuffleRNG.decode(reader, position);
  1979. case 84: return $root.mgb.serialization.fbs.param.CheckNonFinite.decode(reader, position);
  1980. case 85: return $root.mgb.serialization.fbs.param.LayerNorm.decode(reader, position);
  1981. case 86: return $root.mgb.serialization.fbs.param.Dropout.decode(reader, position);
  1982. case 87: return $root.mgb.serialization.fbs.param.RNNCell.decode(reader, position);
  1983. case 88: return $root.mgb.serialization.fbs.param.RNN.decode(reader, position);
  1984. case 89: return $root.mgb.serialization.fbs.param.LSTM.decode(reader, position);
  1985. case 90: return $root.mgb.serialization.fbs.param.Softmax.decode(reader, position);
  1986. case 91: return $root.mgb.serialization.fbs.param.Diag.decode(reader, position);
  1987. case 92: return $root.mgb.serialization.fbs.param.GroupNorm.decode(reader, position);
  1988. case 93: return $root.mgb.serialization.fbs.param.Fill.decode(reader, position);
  1989. default: return undefined;
  1990. }
  1991. }
  1992. };
  1993. $root.mgb.serialization.fbs.v2.Operator = class Operator {
  1994. static decode(reader, position) {
  1995. const $ = new $root.mgb.serialization.fbs.v2.Operator();
  1996. $.type = reader.string_(position, 4, null);
  1997. $.type_id = reader.uint64_(position, 6, 0);
  1998. $.name = reader.string_(position, 8, null);
  1999. $.param = reader.union(position, 10, $root.mgb.serialization.fbs.v2.OperatorParam.decode);
  2000. $.additional_params = reader.unionArray(position, 14, $root.mgb.serialization.fbs.v2.OperatorParam.decode);
  2001. $.inputs = reader.typedArray(position, 18, Uint32Array);
  2002. $.outputs = reader.typedArray(position, 20, Uint32Array);
  2003. $.comp_node = reader.tableArray(position, 22, $root.mgb.serialization.fbs.v2.CompNode.decode);
  2004. $.output_dtype = reader.table(position, 24, $root.mgb.serialization.fbs.DType.decode);
  2005. $.tensors = reader.tableArray(position, 26, $root.mgb.serialization.fbs.v2.Tensor.decode);
  2006. $.opr_version = reader.uint32_(position, 28, 0);
  2007. $.priority = reader.int32_(position, 30, 0);
  2008. $.custom_data = reader.tableArray(position, 32, $root.mgb.serialization.fbs.v2.Blob.decode);
  2009. return $;
  2010. }
  2011. };
  2012. $root.mgb.serialization.fbs.v2.Metadata = class Metadata {
  2013. static decode(reader, position) {
  2014. const $ = new $root.mgb.serialization.fbs.v2.Metadata();
  2015. $.is_valid = reader.bool_(position, 4, false);
  2016. $.graph_modified = reader.bool_(position, 6, false);
  2017. $.optimize_options = reader.uint64_(position, 8, 0);
  2018. $.user_info = reader.string_(position, 10, null);
  2019. return $;
  2020. }
  2021. };
  2022. $root.mgb.serialization.fbs.v2.MiddleTensor = class MiddleTensor {
  2023. static decode(reader, position) {
  2024. const $ = new $root.mgb.serialization.fbs.v2.MiddleTensor();
  2025. $.name = reader.string_(position, 4, null);
  2026. $.shape = reader.typedArray(position, 6, Uint32Array);
  2027. $.comp_node = reader.table(position, 8, $root.mgb.serialization.fbs.v2.CompNode.decode);
  2028. $.dtype = reader.table(position, 10, $root.mgb.serialization.fbs.DType.decode);
  2029. $.format = reader.union(position, 12, $root.mgb.serialization.fbs.v2.TensorFormat.decode);
  2030. return $;
  2031. }
  2032. };
  2033. $root.mgb.serialization.fbs.v2.OutputVar = class OutputVar {
  2034. static decode(reader, position) {
  2035. const $ = new $root.mgb.serialization.fbs.v2.OutputVar();
  2036. $.compact_id = reader.uint32_(position, 4, 0);
  2037. $.original_id = reader.uint32_(position, 6, 0);
  2038. return $;
  2039. }
  2040. };
  2041. $root.mgb.serialization.fbs.v2.OutputAlias = class OutputAlias {
  2042. static decode(reader, position) {
  2043. const $ = new $root.mgb.serialization.fbs.v2.OutputAlias();
  2044. $.id = reader.uint32_(position, 4, 0);
  2045. $.name = reader.string_(position, 6, null);
  2046. return $;
  2047. }
  2048. };
  2049. $root.mgb.serialization.fbs.v2.Model = class Model {
  2050. static identifier(reader) {
  2051. return reader.identifier === 'mge2';
  2052. }
  2053. static create(reader) {
  2054. return $root.mgb.serialization.fbs.v2.Model.decode(reader, reader.root);
  2055. }
  2056. static decode(reader, position) {
  2057. const $ = new $root.mgb.serialization.fbs.v2.Model();
  2058. $.mge_version = reader.uint32_(position, 4, 0);
  2059. $.model_version = reader.uint32_(position, 6, 0);
  2060. $.oprs = reader.tableArray(position, 8, $root.mgb.serialization.fbs.v2.Operator.decode);
  2061. $.middle_tensors = reader.tableArray(position, 10, $root.mgb.serialization.fbs.v2.MiddleTensor.decode);
  2062. $.output_vars_idx = reader.tableArray(position, 12, $root.mgb.serialization.fbs.v2.OutputVar.decode);
  2063. $.output_alias = reader.tableArray(position, 14, $root.mgb.serialization.fbs.v2.OutputAlias.decode);
  2064. $.nr_shared_tensor = reader.uint32_(position, 16, 0);
  2065. $.metadata = reader.table(position, 18, $root.mgb.serialization.fbs.v2.Metadata.decode);
  2066. return $;
  2067. }
  2068. };