onnx-proto.js 90 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707
  1. export const onnx = {};
  2. onnx.Version = {
  3. "_START_VERSION": 0,
  4. "IR_VERSION_2017_10_10": 1,
  5. "IR_VERSION_2017_10_30": 2,
  6. "IR_VERSION_2017_11_3": 3,
  7. "IR_VERSION_2019_1_22": 4,
  8. "IR_VERSION_2019_3_18": 5,
  9. "IR_VERSION_2019_9_19": 6,
  10. "IR_VERSION_2020_5_8": 7,
  11. "IR_VERSION_2021_7_30": 8,
  12. "IR_VERSION_2023_5_5": 9,
  13. "IR_VERSION_2024_3_25": 10,
  14. "IR_VERSION": 11
  15. };
  16. onnx.AttributeProto = class AttributeProto {
  17. constructor() {
  18. this.floats = [];
  19. this.ints = [];
  20. this.strings = [];
  21. this.tensors = [];
  22. this.graphs = [];
  23. this.sparse_tensors = [];
  24. this.type_protos = [];
  25. }
  26. static decode(reader, length) {
  27. const message = new onnx.AttributeProto();
  28. const end = length === undefined ? reader.length : reader.position + length;
  29. while (reader.position < end) {
  30. const tag = reader.uint32();
  31. switch (tag >>> 3) {
  32. case 1:
  33. message.name = reader.string();
  34. break;
  35. case 21:
  36. message.ref_attr_name = reader.string();
  37. break;
  38. case 13:
  39. message.doc_string = reader.string();
  40. break;
  41. case 20:
  42. message.type = reader.int32();
  43. break;
  44. case 2:
  45. message.f = reader.float();
  46. break;
  47. case 3:
  48. message.i = reader.int64();
  49. break;
  50. case 4:
  51. message.s = reader.bytes();
  52. break;
  53. case 5:
  54. message.t = onnx.TensorProto.decode(reader, reader.uint32());
  55. break;
  56. case 6:
  57. message.g = onnx.GraphProto.decode(reader, reader.uint32());
  58. break;
  59. case 22:
  60. message.sparse_tensor = onnx.SparseTensorProto.decode(reader, reader.uint32());
  61. break;
  62. case 14:
  63. message.tp = onnx.TypeProto.decode(reader, reader.uint32());
  64. break;
  65. case 7:
  66. message.floats = reader.floats(message.floats, tag);
  67. break;
  68. case 8:
  69. message.ints = reader.array(message.ints, () => reader.int64(), tag);
  70. break;
  71. case 9:
  72. message.strings.push(reader.bytes());
  73. break;
  74. case 10:
  75. message.tensors.push(onnx.TensorProto.decode(reader, reader.uint32()));
  76. break;
  77. case 11:
  78. message.graphs.push(onnx.GraphProto.decode(reader, reader.uint32()));
  79. break;
  80. case 23:
  81. message.sparse_tensors.push(onnx.SparseTensorProto.decode(reader, reader.uint32()));
  82. break;
  83. case 15:
  84. message.type_protos.push(onnx.TypeProto.decode(reader, reader.uint32()));
  85. break;
  86. default:
  87. reader.skipType(tag & 7);
  88. break;
  89. }
  90. }
  91. return message;
  92. }
  93. static decodeText(reader) {
  94. const message = new onnx.AttributeProto();
  95. reader.start();
  96. while (!reader.end()) {
  97. const tag = reader.tag();
  98. switch (tag) {
  99. case "name":
  100. message.name = reader.string();
  101. break;
  102. case "ref_attr_name":
  103. message.ref_attr_name = reader.string();
  104. break;
  105. case "doc_string":
  106. message.doc_string = reader.string();
  107. break;
  108. case "type":
  109. message.type = reader.enum(onnx.AttributeProto.AttributeType);
  110. break;
  111. case "f":
  112. message.f = reader.float();
  113. break;
  114. case "i":
  115. message.i = reader.int64();
  116. break;
  117. case "s":
  118. message.s = reader.bytes();
  119. break;
  120. case "t":
  121. message.t = onnx.TensorProto.decodeText(reader);
  122. break;
  123. case "g":
  124. message.g = onnx.GraphProto.decodeText(reader);
  125. break;
  126. case "sparse_tensor":
  127. message.sparse_tensor = onnx.SparseTensorProto.decodeText(reader);
  128. break;
  129. case "tp":
  130. message.tp = onnx.TypeProto.decodeText(reader);
  131. break;
  132. case "floats":
  133. reader.array(message.floats, () => reader.float());
  134. break;
  135. case "ints":
  136. reader.array(message.ints, () => reader.int64());
  137. break;
  138. case "strings":
  139. reader.array(message.strings, () => reader.bytes());
  140. break;
  141. case "tensors":
  142. message.tensors.push(onnx.TensorProto.decodeText(reader));
  143. break;
  144. case "graphs":
  145. message.graphs.push(onnx.GraphProto.decodeText(reader));
  146. break;
  147. case "sparse_tensors":
  148. message.sparse_tensors.push(onnx.SparseTensorProto.decodeText(reader));
  149. break;
  150. case "type_protos":
  151. message.type_protos.push(onnx.TypeProto.decodeText(reader));
  152. break;
  153. default:
  154. reader.field(tag, message);
  155. break;
  156. }
  157. }
  158. return message;
  159. }
  160. static decodeJson(obj) {
  161. const message = new onnx.AttributeProto();
  162. if ('name' in obj) {
  163. message.name = obj.name;
  164. }
  165. if ('refAttrName' in obj) {
  166. message.ref_attr_name = obj.refAttrName;
  167. }
  168. if ('docString' in obj) {
  169. message.doc_string = obj.docString;
  170. }
  171. if ('type' in obj) {
  172. message.type = typeof obj.type === 'string' ? onnx.AttributeProto.AttributeType[obj.type] : obj.type;
  173. }
  174. if ('f' in obj) {
  175. message.f = Number(obj.f);
  176. }
  177. if ('i' in obj) {
  178. message.i = BigInt(obj.i);
  179. }
  180. if ('s' in obj) {
  181. message.s = typeof obj.s === 'string' ? Uint8Array.from(atob(obj.s), (c) => c.charCodeAt(0)) : Uint8Array.from(obj.s);
  182. }
  183. if ('t' in obj) {
  184. message.t = onnx.TensorProto.decodeJson(obj.t);
  185. }
  186. if ('g' in obj) {
  187. message.g = onnx.GraphProto.decodeJson(obj.g);
  188. }
  189. if ('sparseTensor' in obj) {
  190. message.sparse_tensor = onnx.SparseTensorProto.decodeJson(obj.sparseTensor);
  191. }
  192. if ('tp' in obj) {
  193. message.tp = onnx.TypeProto.decodeJson(obj.tp);
  194. }
  195. if ('floats' in obj) {
  196. message.floats = obj.floats.map((obj) => Number(obj));
  197. }
  198. if ('ints' in obj) {
  199. message.ints = obj.ints.map((obj) => BigInt(obj));
  200. }
  201. if ('strings' in obj) {
  202. message.strings = obj.strings.map((obj) => typeof obj === 'string' ? Uint8Array.from(atob(obj), (c) => c.charCodeAt(0)) : Uint8Array.from(obj));
  203. }
  204. if ('tensors' in obj) {
  205. message.tensors = obj.tensors.map((obj) => onnx.TensorProto.decodeJson(obj));
  206. }
  207. if ('graphs' in obj) {
  208. message.graphs = obj.graphs.map((obj) => onnx.GraphProto.decodeJson(obj));
  209. }
  210. if ('sparseTensors' in obj) {
  211. message.sparse_tensors = obj.sparseTensors.map((obj) => onnx.SparseTensorProto.decodeJson(obj));
  212. }
  213. if ('typeProtos' in obj) {
  214. message.type_protos = obj.typeProtos.map((obj) => onnx.TypeProto.decodeJson(obj));
  215. }
  216. return message;
  217. }
  218. };
  219. onnx.AttributeProto.prototype.name = "";
  220. onnx.AttributeProto.prototype.ref_attr_name = "";
  221. onnx.AttributeProto.prototype.doc_string = "";
  222. onnx.AttributeProto.prototype.type = 0;
  223. onnx.AttributeProto.prototype.f = 0;
  224. onnx.AttributeProto.prototype.i = 0n;
  225. onnx.AttributeProto.prototype.s = new Uint8Array([]);
  226. onnx.AttributeProto.prototype.t = null;
  227. onnx.AttributeProto.prototype.g = null;
  228. onnx.AttributeProto.prototype.sparse_tensor = null;
  229. onnx.AttributeProto.prototype.tp = null;
  230. onnx.AttributeProto.AttributeType = {
  231. "UNDEFINED": 0,
  232. "FLOAT": 1,
  233. "INT": 2,
  234. "STRING": 3,
  235. "TENSOR": 4,
  236. "GRAPH": 5,
  237. "SPARSE_TENSOR": 11,
  238. "TYPE_PROTO": 13,
  239. "FLOATS": 6,
  240. "INTS": 7,
  241. "STRINGS": 8,
  242. "TENSORS": 9,
  243. "GRAPHS": 10,
  244. "SPARSE_TENSORS": 12,
  245. "TYPE_PROTOS": 14
  246. };
  247. onnx.ValueInfoProto = class ValueInfoProto {
  248. constructor() {
  249. this.metadata_props = [];
  250. }
  251. static decode(reader, length) {
  252. const message = new onnx.ValueInfoProto();
  253. const end = length === undefined ? reader.length : reader.position + length;
  254. while (reader.position < end) {
  255. const tag = reader.uint32();
  256. switch (tag >>> 3) {
  257. case 1:
  258. message.name = reader.string();
  259. break;
  260. case 2:
  261. message.type = onnx.TypeProto.decode(reader, reader.uint32());
  262. break;
  263. case 3:
  264. message.doc_string = reader.string();
  265. break;
  266. case 4:
  267. message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
  268. break;
  269. default:
  270. reader.skipType(tag & 7);
  271. break;
  272. }
  273. }
  274. return message;
  275. }
  276. static decodeText(reader) {
  277. const message = new onnx.ValueInfoProto();
  278. reader.start();
  279. while (!reader.end()) {
  280. const tag = reader.tag();
  281. switch (tag) {
  282. case "name":
  283. message.name = reader.string();
  284. break;
  285. case "type":
  286. message.type = onnx.TypeProto.decodeText(reader);
  287. break;
  288. case "doc_string":
  289. message.doc_string = reader.string();
  290. break;
  291. case "metadata_props":
  292. message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
  293. break;
  294. default:
  295. reader.field(tag, message);
  296. break;
  297. }
  298. }
  299. return message;
  300. }
  301. static decodeJson(obj) {
  302. const message = new onnx.ValueInfoProto();
  303. if ('name' in obj) {
  304. message.name = obj.name;
  305. }
  306. if ('type' in obj) {
  307. message.type = onnx.TypeProto.decodeJson(obj.type);
  308. }
  309. if ('docString' in obj) {
  310. message.doc_string = obj.docString;
  311. }
  312. if ('metadataProps' in obj) {
  313. message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
  314. }
  315. return message;
  316. }
  317. };
  318. onnx.ValueInfoProto.prototype.name = "";
  319. onnx.ValueInfoProto.prototype.type = null;
  320. onnx.ValueInfoProto.prototype.doc_string = "";
  321. onnx.NodeProto = class NodeProto {
  322. constructor() {
  323. this.input = [];
  324. this.output = [];
  325. this.attribute = [];
  326. this.metadata_props = [];
  327. this.device_configurations = [];
  328. }
  329. static decode(reader, length) {
  330. const message = new onnx.NodeProto();
  331. const end = length === undefined ? reader.length : reader.position + length;
  332. while (reader.position < end) {
  333. const tag = reader.uint32();
  334. switch (tag >>> 3) {
  335. case 1:
  336. message.input.push(reader.string());
  337. break;
  338. case 2:
  339. message.output.push(reader.string());
  340. break;
  341. case 3:
  342. message.name = reader.string();
  343. break;
  344. case 4:
  345. message.op_type = reader.string();
  346. break;
  347. case 7:
  348. message.domain = reader.string();
  349. break;
  350. case 8:
  351. message.overload = reader.string();
  352. break;
  353. case 5:
  354. message.attribute.push(onnx.AttributeProto.decode(reader, reader.uint32()));
  355. break;
  356. case 6:
  357. message.doc_string = reader.string();
  358. break;
  359. case 9:
  360. message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
  361. break;
  362. case 10:
  363. message.device_configurations.push(onnx.NodeDeviceConfigurationProto.decode(reader, reader.uint32()));
  364. break;
  365. default:
  366. reader.skipType(tag & 7);
  367. break;
  368. }
  369. }
  370. return message;
  371. }
  372. static decodeText(reader) {
  373. const message = new onnx.NodeProto();
  374. reader.start();
  375. while (!reader.end()) {
  376. const tag = reader.tag();
  377. switch (tag) {
  378. case "input":
  379. reader.array(message.input, () => reader.string());
  380. break;
  381. case "output":
  382. reader.array(message.output, () => reader.string());
  383. break;
  384. case "name":
  385. message.name = reader.string();
  386. break;
  387. case "op_type":
  388. message.op_type = reader.string();
  389. break;
  390. case "domain":
  391. message.domain = reader.string();
  392. break;
  393. case "overload":
  394. message.overload = reader.string();
  395. break;
  396. case "attribute":
  397. message.attribute.push(onnx.AttributeProto.decodeText(reader));
  398. break;
  399. case "doc_string":
  400. message.doc_string = reader.string();
  401. break;
  402. case "metadata_props":
  403. message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
  404. break;
  405. case "device_configurations":
  406. message.device_configurations.push(onnx.NodeDeviceConfigurationProto.decodeText(reader));
  407. break;
  408. default:
  409. reader.field(tag, message);
  410. break;
  411. }
  412. }
  413. return message;
  414. }
  415. static decodeJson(obj) {
  416. const message = new onnx.NodeProto();
  417. if ('input' in obj) {
  418. message.input = obj.input;
  419. }
  420. if ('output' in obj) {
  421. message.output = obj.output;
  422. }
  423. if ('name' in obj) {
  424. message.name = obj.name;
  425. }
  426. if ('opType' in obj) {
  427. message.op_type = obj.opType;
  428. }
  429. if ('domain' in obj) {
  430. message.domain = obj.domain;
  431. }
  432. if ('overload' in obj) {
  433. message.overload = obj.overload;
  434. }
  435. if ('attribute' in obj) {
  436. message.attribute = obj.attribute.map((obj) => onnx.AttributeProto.decodeJson(obj));
  437. }
  438. if ('docString' in obj) {
  439. message.doc_string = obj.docString;
  440. }
  441. if ('metadataProps' in obj) {
  442. message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
  443. }
  444. if ('deviceConfigurations' in obj) {
  445. message.device_configurations = obj.deviceConfigurations.map((obj) => onnx.NodeDeviceConfigurationProto.decodeJson(obj));
  446. }
  447. return message;
  448. }
  449. };
  450. onnx.NodeProto.prototype.name = "";
  451. onnx.NodeProto.prototype.op_type = "";
  452. onnx.NodeProto.prototype.domain = "";
  453. onnx.NodeProto.prototype.overload = "";
  454. onnx.NodeProto.prototype.doc_string = "";
  455. onnx.IntIntListEntryProto = class IntIntListEntryProto {
  456. constructor() {
  457. this.value = [];
  458. }
  459. static decode(reader, length) {
  460. const message = new onnx.IntIntListEntryProto();
  461. const end = length === undefined ? reader.length : reader.position + length;
  462. while (reader.position < end) {
  463. const tag = reader.uint32();
  464. switch (tag >>> 3) {
  465. case 1:
  466. message.key = reader.int64();
  467. break;
  468. case 2:
  469. message.value = reader.array(message.value, () => reader.int64(), tag);
  470. break;
  471. default:
  472. reader.skipType(tag & 7);
  473. break;
  474. }
  475. }
  476. return message;
  477. }
  478. static decodeText(reader) {
  479. const message = new onnx.IntIntListEntryProto();
  480. reader.start();
  481. while (!reader.end()) {
  482. const tag = reader.tag();
  483. switch (tag) {
  484. case "key":
  485. message.key = reader.int64();
  486. break;
  487. case "value":
  488. reader.array(message.value, () => reader.int64());
  489. break;
  490. default:
  491. reader.field(tag, message);
  492. break;
  493. }
  494. }
  495. return message;
  496. }
  497. static decodeJson(obj) {
  498. const message = new onnx.IntIntListEntryProto();
  499. if ('key' in obj) {
  500. message.key = BigInt(obj.key);
  501. }
  502. if ('value' in obj) {
  503. message.value = obj.value.map((obj) => BigInt(obj));
  504. }
  505. return message;
  506. }
  507. };
  508. onnx.IntIntListEntryProto.prototype.key = 0n;
  509. onnx.NodeDeviceConfigurationProto = class NodeDeviceConfigurationProto {
  510. constructor() {
  511. this.sharding_spec = [];
  512. }
  513. static decode(reader, length) {
  514. const message = new onnx.NodeDeviceConfigurationProto();
  515. const end = length === undefined ? reader.length : reader.position + length;
  516. while (reader.position < end) {
  517. const tag = reader.uint32();
  518. switch (tag >>> 3) {
  519. case 1:
  520. message.configuration_id = reader.string();
  521. break;
  522. case 2:
  523. message.sharding_spec.push(onnx.ShardingSpecProto.decode(reader, reader.uint32()));
  524. break;
  525. case 3:
  526. message.pipeline_stage = reader.int32();
  527. break;
  528. default:
  529. reader.skipType(tag & 7);
  530. break;
  531. }
  532. }
  533. return message;
  534. }
  535. static decodeText(reader) {
  536. const message = new onnx.NodeDeviceConfigurationProto();
  537. reader.start();
  538. while (!reader.end()) {
  539. const tag = reader.tag();
  540. switch (tag) {
  541. case "configuration_id":
  542. message.configuration_id = reader.string();
  543. break;
  544. case "sharding_spec":
  545. message.sharding_spec.push(onnx.ShardingSpecProto.decodeText(reader));
  546. break;
  547. case "pipeline_stage":
  548. message.pipeline_stage = reader.int32();
  549. break;
  550. default:
  551. reader.field(tag, message);
  552. break;
  553. }
  554. }
  555. return message;
  556. }
  557. static decodeJson(obj) {
  558. const message = new onnx.NodeDeviceConfigurationProto();
  559. if ('configurationId' in obj) {
  560. message.configuration_id = obj.configurationId;
  561. }
  562. if ('shardingSpec' in obj) {
  563. message.sharding_spec = obj.shardingSpec.map((obj) => onnx.ShardingSpecProto.decodeJson(obj));
  564. }
  565. if ('pipelineStage' in obj) {
  566. message.pipeline_stage = Number(obj.pipelineStage);
  567. }
  568. return message;
  569. }
  570. };
  571. onnx.NodeDeviceConfigurationProto.prototype.configuration_id = "";
  572. onnx.NodeDeviceConfigurationProto.prototype.pipeline_stage = 0;
  573. onnx.ShardingSpecProto = class ShardingSpecProto {
  574. constructor() {
  575. this.device = [];
  576. this.index_to_device_group_map = [];
  577. this.sharded_dim = [];
  578. }
  579. static decode(reader, length) {
  580. const message = new onnx.ShardingSpecProto();
  581. const end = length === undefined ? reader.length : reader.position + length;
  582. while (reader.position < end) {
  583. const tag = reader.uint32();
  584. switch (tag >>> 3) {
  585. case 1:
  586. message.tensor_name = reader.string();
  587. break;
  588. case 2:
  589. message.device = reader.array(message.device, () => reader.int64(), tag);
  590. break;
  591. case 3:
  592. message.index_to_device_group_map.push(onnx.IntIntListEntryProto.decode(reader, reader.uint32()));
  593. break;
  594. case 4:
  595. message.sharded_dim.push(onnx.ShardedDimProto.decode(reader, reader.uint32()));
  596. break;
  597. default:
  598. reader.skipType(tag & 7);
  599. break;
  600. }
  601. }
  602. return message;
  603. }
  604. static decodeText(reader) {
  605. const message = new onnx.ShardingSpecProto();
  606. reader.start();
  607. while (!reader.end()) {
  608. const tag = reader.tag();
  609. switch (tag) {
  610. case "tensor_name":
  611. message.tensor_name = reader.string();
  612. break;
  613. case "device":
  614. reader.array(message.device, () => reader.int64());
  615. break;
  616. case "index_to_device_group_map":
  617. message.index_to_device_group_map.push(onnx.IntIntListEntryProto.decodeText(reader));
  618. break;
  619. case "sharded_dim":
  620. message.sharded_dim.push(onnx.ShardedDimProto.decodeText(reader));
  621. break;
  622. default:
  623. reader.field(tag, message);
  624. break;
  625. }
  626. }
  627. return message;
  628. }
  629. static decodeJson(obj) {
  630. const message = new onnx.ShardingSpecProto();
  631. if ('tensorName' in obj) {
  632. message.tensor_name = obj.tensorName;
  633. }
  634. if ('device' in obj) {
  635. message.device = obj.device.map((obj) => BigInt(obj));
  636. }
  637. if ('indexToDeviceGroupMap' in obj) {
  638. message.index_to_device_group_map = obj.indexToDeviceGroupMap.map((obj) => onnx.IntIntListEntryProto.decodeJson(obj));
  639. }
  640. if ('shardedDim' in obj) {
  641. message.sharded_dim = obj.shardedDim.map((obj) => onnx.ShardedDimProto.decodeJson(obj));
  642. }
  643. return message;
  644. }
  645. };
  646. onnx.ShardingSpecProto.prototype.tensor_name = "";
  647. onnx.ShardedDimProto = class ShardedDimProto {
  648. constructor() {
  649. this.simple_sharding = [];
  650. }
  651. static decode(reader, length) {
  652. const message = new onnx.ShardedDimProto();
  653. const end = length === undefined ? reader.length : reader.position + length;
  654. while (reader.position < end) {
  655. const tag = reader.uint32();
  656. switch (tag >>> 3) {
  657. case 1:
  658. message.axis = reader.int64();
  659. break;
  660. case 2:
  661. message.simple_sharding.push(onnx.SimpleShardedDimProto.decode(reader, reader.uint32()));
  662. break;
  663. default:
  664. reader.skipType(tag & 7);
  665. break;
  666. }
  667. }
  668. return message;
  669. }
  670. static decodeText(reader) {
  671. const message = new onnx.ShardedDimProto();
  672. reader.start();
  673. while (!reader.end()) {
  674. const tag = reader.tag();
  675. switch (tag) {
  676. case "axis":
  677. message.axis = reader.int64();
  678. break;
  679. case "simple_sharding":
  680. message.simple_sharding.push(onnx.SimpleShardedDimProto.decodeText(reader));
  681. break;
  682. default:
  683. reader.field(tag, message);
  684. break;
  685. }
  686. }
  687. return message;
  688. }
  689. static decodeJson(obj) {
  690. const message = new onnx.ShardedDimProto();
  691. if ('axis' in obj) {
  692. message.axis = BigInt(obj.axis);
  693. }
  694. if ('simpleSharding' in obj) {
  695. message.simple_sharding = obj.simpleSharding.map((obj) => onnx.SimpleShardedDimProto.decodeJson(obj));
  696. }
  697. return message;
  698. }
  699. };
  700. onnx.ShardedDimProto.prototype.axis = 0n;
  701. onnx.SimpleShardedDimProto = class SimpleShardedDimProto {
  702. get dim() {
  703. onnx.SimpleShardedDimProto.dimSet = onnx.SimpleShardedDimProto.dimSet || new Set(["dim_value", "dim_param"]);
  704. return Object.keys(this).find((key) => onnx.SimpleShardedDimProto.dimSet.has(key) && this[key] !== null);
  705. }
  706. static decode(reader, length) {
  707. const message = new onnx.SimpleShardedDimProto();
  708. const end = length === undefined ? reader.length : reader.position + length;
  709. while (reader.position < end) {
  710. const tag = reader.uint32();
  711. switch (tag >>> 3) {
  712. case 1:
  713. message.dim_value = reader.int64();
  714. break;
  715. case 2:
  716. message.dim_param = reader.string();
  717. break;
  718. case 3:
  719. message.num_shards = reader.int64();
  720. break;
  721. default:
  722. reader.skipType(tag & 7);
  723. break;
  724. }
  725. }
  726. return message;
  727. }
  728. static decodeText(reader) {
  729. const message = new onnx.SimpleShardedDimProto();
  730. reader.start();
  731. while (!reader.end()) {
  732. const tag = reader.tag();
  733. switch (tag) {
  734. case "dim_value":
  735. message.dim_value = reader.int64();
  736. break;
  737. case "dim_param":
  738. message.dim_param = reader.string();
  739. break;
  740. case "num_shards":
  741. message.num_shards = reader.int64();
  742. break;
  743. default:
  744. reader.field(tag, message);
  745. break;
  746. }
  747. }
  748. return message;
  749. }
  750. static decodeJson(obj) {
  751. const message = new onnx.SimpleShardedDimProto();
  752. if ('dimValue' in obj) {
  753. message.dim_value = BigInt(obj.dimValue);
  754. }
  755. if ('dimParam' in obj) {
  756. message.dim_param = obj.dimParam;
  757. }
  758. if ('numShards' in obj) {
  759. message.num_shards = BigInt(obj.numShards);
  760. }
  761. return message;
  762. }
  763. };
  764. onnx.SimpleShardedDimProto.prototype.num_shards = 0n;
  765. onnx.TrainingInfoProto = class TrainingInfoProto {
  766. constructor() {
  767. this.initialization_binding = [];
  768. this.update_binding = [];
  769. }
  770. static decode(reader, length) {
  771. const message = new onnx.TrainingInfoProto();
  772. const end = length === undefined ? reader.length : reader.position + length;
  773. while (reader.position < end) {
  774. const tag = reader.uint32();
  775. switch (tag >>> 3) {
  776. case 1:
  777. message.initialization = onnx.GraphProto.decode(reader, reader.uint32());
  778. break;
  779. case 2:
  780. message.algorithm = onnx.GraphProto.decode(reader, reader.uint32());
  781. break;
  782. case 3:
  783. message.initialization_binding.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
  784. break;
  785. case 4:
  786. message.update_binding.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
  787. break;
  788. default:
  789. reader.skipType(tag & 7);
  790. break;
  791. }
  792. }
  793. return message;
  794. }
  795. static decodeText(reader) {
  796. const message = new onnx.TrainingInfoProto();
  797. reader.start();
  798. while (!reader.end()) {
  799. const tag = reader.tag();
  800. switch (tag) {
  801. case "initialization":
  802. message.initialization = onnx.GraphProto.decodeText(reader);
  803. break;
  804. case "algorithm":
  805. message.algorithm = onnx.GraphProto.decodeText(reader);
  806. break;
  807. case "initialization_binding":
  808. message.initialization_binding.push(onnx.StringStringEntryProto.decodeText(reader));
  809. break;
  810. case "update_binding":
  811. message.update_binding.push(onnx.StringStringEntryProto.decodeText(reader));
  812. break;
  813. default:
  814. reader.field(tag, message);
  815. break;
  816. }
  817. }
  818. return message;
  819. }
  820. static decodeJson(obj) {
  821. const message = new onnx.TrainingInfoProto();
  822. if ('initialization' in obj) {
  823. message.initialization = onnx.GraphProto.decodeJson(obj.initialization);
  824. }
  825. if ('algorithm' in obj) {
  826. message.algorithm = onnx.GraphProto.decodeJson(obj.algorithm);
  827. }
  828. if ('initializationBinding' in obj) {
  829. message.initialization_binding = obj.initializationBinding.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
  830. }
  831. if ('updateBinding' in obj) {
  832. message.update_binding = obj.updateBinding.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
  833. }
  834. return message;
  835. }
  836. };
  837. onnx.TrainingInfoProto.prototype.initialization = null;
  838. onnx.TrainingInfoProto.prototype.algorithm = null;
  839. onnx.ModelProto = class ModelProto {
  840. constructor() {
  841. this.opset_import = [];
  842. this.metadata_props = [];
  843. this.training_info = [];
  844. this.functions = [];
  845. this.configuration = [];
  846. }
  847. static decode(reader, length) {
  848. const message = new onnx.ModelProto();
  849. const end = length === undefined ? reader.length : reader.position + length;
  850. while (reader.position < end) {
  851. const tag = reader.uint32();
  852. switch (tag >>> 3) {
  853. case 1:
  854. message.ir_version = reader.int64();
  855. break;
  856. case 8:
  857. message.opset_import.push(onnx.OperatorSetIdProto.decode(reader, reader.uint32()));
  858. break;
  859. case 2:
  860. message.producer_name = reader.string();
  861. break;
  862. case 3:
  863. message.producer_version = reader.string();
  864. break;
  865. case 4:
  866. message.domain = reader.string();
  867. break;
  868. case 5:
  869. message.model_version = reader.int64();
  870. break;
  871. case 6:
  872. message.doc_string = reader.string();
  873. break;
  874. case 7:
  875. message.graph = onnx.GraphProto.decode(reader, reader.uint32());
  876. break;
  877. case 14:
  878. message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
  879. break;
  880. case 20:
  881. message.training_info.push(onnx.TrainingInfoProto.decode(reader, reader.uint32()));
  882. break;
  883. case 25:
  884. message.functions.push(onnx.FunctionProto.decode(reader, reader.uint32()));
  885. break;
  886. case 26:
  887. message.configuration.push(onnx.DeviceConfigurationProto.decode(reader, reader.uint32()));
  888. break;
  889. default:
  890. reader.skipType(tag & 7);
  891. break;
  892. }
  893. }
  894. return message;
  895. }
  896. static decodeText(reader) {
  897. const message = new onnx.ModelProto();
  898. reader.start();
  899. while (!reader.end()) {
  900. const tag = reader.tag();
  901. switch (tag) {
  902. case "ir_version":
  903. message.ir_version = reader.int64();
  904. break;
  905. case "opset_import":
  906. message.opset_import.push(onnx.OperatorSetIdProto.decodeText(reader));
  907. break;
  908. case "producer_name":
  909. message.producer_name = reader.string();
  910. break;
  911. case "producer_version":
  912. message.producer_version = reader.string();
  913. break;
  914. case "domain":
  915. message.domain = reader.string();
  916. break;
  917. case "model_version":
  918. message.model_version = reader.int64();
  919. break;
  920. case "doc_string":
  921. message.doc_string = reader.string();
  922. break;
  923. case "graph":
  924. message.graph = onnx.GraphProto.decodeText(reader);
  925. break;
  926. case "metadata_props":
  927. message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
  928. break;
  929. case "training_info":
  930. message.training_info.push(onnx.TrainingInfoProto.decodeText(reader));
  931. break;
  932. case "functions":
  933. message.functions.push(onnx.FunctionProto.decodeText(reader));
  934. break;
  935. case "configuration":
  936. message.configuration.push(onnx.DeviceConfigurationProto.decodeText(reader));
  937. break;
  938. default:
  939. reader.field(tag, message);
  940. break;
  941. }
  942. }
  943. return message;
  944. }
  945. static decodeJson(obj) {
  946. const message = new onnx.ModelProto();
  947. if ('irVersion' in obj) {
  948. message.ir_version = BigInt(obj.irVersion);
  949. }
  950. if ('opsetImport' in obj) {
  951. message.opset_import = obj.opsetImport.map((obj) => onnx.OperatorSetIdProto.decodeJson(obj));
  952. }
  953. if ('producerName' in obj) {
  954. message.producer_name = obj.producerName;
  955. }
  956. if ('producerVersion' in obj) {
  957. message.producer_version = obj.producerVersion;
  958. }
  959. if ('domain' in obj) {
  960. message.domain = obj.domain;
  961. }
  962. if ('modelVersion' in obj) {
  963. message.model_version = BigInt(obj.modelVersion);
  964. }
  965. if ('docString' in obj) {
  966. message.doc_string = obj.docString;
  967. }
  968. if ('graph' in obj) {
  969. message.graph = onnx.GraphProto.decodeJson(obj.graph);
  970. }
  971. if ('metadataProps' in obj) {
  972. message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
  973. }
  974. if ('trainingInfo' in obj) {
  975. message.training_info = obj.trainingInfo.map((obj) => onnx.TrainingInfoProto.decodeJson(obj));
  976. }
  977. if ('functions' in obj) {
  978. message.functions = obj.functions.map((obj) => onnx.FunctionProto.decodeJson(obj));
  979. }
  980. if ('configuration' in obj) {
  981. message.configuration = obj.configuration.map((obj) => onnx.DeviceConfigurationProto.decodeJson(obj));
  982. }
  983. return message;
  984. }
  985. };
  986. onnx.ModelProto.prototype.ir_version = 0n;
  987. onnx.ModelProto.prototype.producer_name = "";
  988. onnx.ModelProto.prototype.producer_version = "";
  989. onnx.ModelProto.prototype.domain = "";
  990. onnx.ModelProto.prototype.model_version = 0n;
  991. onnx.ModelProto.prototype.doc_string = "";
  992. onnx.ModelProto.prototype.graph = null;
  993. onnx.DeviceConfigurationProto = class DeviceConfigurationProto {
  994. constructor() {
  995. this.device = [];
  996. }
  997. static decode(reader, length) {
  998. const message = new onnx.DeviceConfigurationProto();
  999. const end = length === undefined ? reader.length : reader.position + length;
  1000. while (reader.position < end) {
  1001. const tag = reader.uint32();
  1002. switch (tag >>> 3) {
  1003. case 1:
  1004. message.name = reader.string();
  1005. break;
  1006. case 2:
  1007. message.num_devices = reader.int32();
  1008. break;
  1009. case 3:
  1010. message.device.push(reader.string());
  1011. break;
  1012. default:
  1013. reader.skipType(tag & 7);
  1014. break;
  1015. }
  1016. }
  1017. return message;
  1018. }
  1019. static decodeText(reader) {
  1020. const message = new onnx.DeviceConfigurationProto();
  1021. reader.start();
  1022. while (!reader.end()) {
  1023. const tag = reader.tag();
  1024. switch (tag) {
  1025. case "name":
  1026. message.name = reader.string();
  1027. break;
  1028. case "num_devices":
  1029. message.num_devices = reader.int32();
  1030. break;
  1031. case "device":
  1032. reader.array(message.device, () => reader.string());
  1033. break;
  1034. default:
  1035. reader.field(tag, message);
  1036. break;
  1037. }
  1038. }
  1039. return message;
  1040. }
  1041. static decodeJson(obj) {
  1042. const message = new onnx.DeviceConfigurationProto();
  1043. if ('name' in obj) {
  1044. message.name = obj.name;
  1045. }
  1046. if ('numDevices' in obj) {
  1047. message.num_devices = Number(obj.numDevices);
  1048. }
  1049. if ('device' in obj) {
  1050. message.device = obj.device;
  1051. }
  1052. return message;
  1053. }
  1054. };
  1055. onnx.DeviceConfigurationProto.prototype.name = "";
  1056. onnx.DeviceConfigurationProto.prototype.num_devices = 0;
  1057. onnx.StringStringEntryProto = class StringStringEntryProto {
  1058. static decode(reader, length) {
  1059. const message = new onnx.StringStringEntryProto();
  1060. const end = length === undefined ? reader.length : reader.position + length;
  1061. while (reader.position < end) {
  1062. const tag = reader.uint32();
  1063. switch (tag >>> 3) {
  1064. case 1:
  1065. message.key = reader.string();
  1066. break;
  1067. case 2:
  1068. message.value = reader.string();
  1069. break;
  1070. default:
  1071. reader.skipType(tag & 7);
  1072. break;
  1073. }
  1074. }
  1075. return message;
  1076. }
  1077. static decodeText(reader) {
  1078. const message = new onnx.StringStringEntryProto();
  1079. reader.start();
  1080. while (!reader.end()) {
  1081. const tag = reader.tag();
  1082. switch (tag) {
  1083. case "key":
  1084. message.key = reader.string();
  1085. break;
  1086. case "value":
  1087. message.value = reader.string();
  1088. break;
  1089. default:
  1090. reader.field(tag, message);
  1091. break;
  1092. }
  1093. }
  1094. return message;
  1095. }
  1096. static decodeJson(obj) {
  1097. const message = new onnx.StringStringEntryProto();
  1098. if ('key' in obj) {
  1099. message.key = obj.key;
  1100. }
  1101. if ('value' in obj) {
  1102. message.value = obj.value;
  1103. }
  1104. return message;
  1105. }
  1106. };
  1107. onnx.StringStringEntryProto.prototype.key = "";
  1108. onnx.StringStringEntryProto.prototype.value = "";
  1109. onnx.TensorAnnotation = class TensorAnnotation {
  1110. constructor() {
  1111. this.quant_parameter_tensor_names = [];
  1112. }
  1113. static decode(reader, length) {
  1114. const message = new onnx.TensorAnnotation();
  1115. const end = length === undefined ? reader.length : reader.position + length;
  1116. while (reader.position < end) {
  1117. const tag = reader.uint32();
  1118. switch (tag >>> 3) {
  1119. case 1:
  1120. message.tensor_name = reader.string();
  1121. break;
  1122. case 2:
  1123. message.quant_parameter_tensor_names.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
  1124. break;
  1125. default:
  1126. reader.skipType(tag & 7);
  1127. break;
  1128. }
  1129. }
  1130. return message;
  1131. }
  1132. static decodeText(reader) {
  1133. const message = new onnx.TensorAnnotation();
  1134. reader.start();
  1135. while (!reader.end()) {
  1136. const tag = reader.tag();
  1137. switch (tag) {
  1138. case "tensor_name":
  1139. message.tensor_name = reader.string();
  1140. break;
  1141. case "quant_parameter_tensor_names":
  1142. message.quant_parameter_tensor_names.push(onnx.StringStringEntryProto.decodeText(reader));
  1143. break;
  1144. default:
  1145. reader.field(tag, message);
  1146. break;
  1147. }
  1148. }
  1149. return message;
  1150. }
  1151. static decodeJson(obj) {
  1152. const message = new onnx.TensorAnnotation();
  1153. if ('tensorName' in obj) {
  1154. message.tensor_name = obj.tensorName;
  1155. }
  1156. if ('quantParameterTensorNames' in obj) {
  1157. message.quant_parameter_tensor_names = obj.quantParameterTensorNames.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
  1158. }
  1159. return message;
  1160. }
  1161. };
  1162. onnx.TensorAnnotation.prototype.tensor_name = "";
  1163. onnx.GraphProto = class GraphProto {
  1164. constructor() {
  1165. this.node = [];
  1166. this.initializer = [];
  1167. this.sparse_initializer = [];
  1168. this.input = [];
  1169. this.output = [];
  1170. this.value_info = [];
  1171. this.quantization_annotation = [];
  1172. this.metadata_props = [];
  1173. }
  1174. static decode(reader, length) {
  1175. const message = new onnx.GraphProto();
  1176. const end = length === undefined ? reader.length : reader.position + length;
  1177. while (reader.position < end) {
  1178. const tag = reader.uint32();
  1179. switch (tag >>> 3) {
  1180. case 1:
  1181. message.node.push(onnx.NodeProto.decode(reader, reader.uint32()));
  1182. break;
  1183. case 2:
  1184. message.name = reader.string();
  1185. break;
  1186. case 5:
  1187. message.initializer.push(onnx.TensorProto.decode(reader, reader.uint32()));
  1188. break;
  1189. case 15:
  1190. message.sparse_initializer.push(onnx.SparseTensorProto.decode(reader, reader.uint32()));
  1191. break;
  1192. case 10:
  1193. message.doc_string = reader.string();
  1194. break;
  1195. case 11:
  1196. message.input.push(onnx.ValueInfoProto.decode(reader, reader.uint32()));
  1197. break;
  1198. case 12:
  1199. message.output.push(onnx.ValueInfoProto.decode(reader, reader.uint32()));
  1200. break;
  1201. case 13:
  1202. message.value_info.push(onnx.ValueInfoProto.decode(reader, reader.uint32()));
  1203. break;
  1204. case 14:
  1205. message.quantization_annotation.push(onnx.TensorAnnotation.decode(reader, reader.uint32()));
  1206. break;
  1207. case 16:
  1208. message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
  1209. break;
  1210. default:
  1211. reader.skipType(tag & 7);
  1212. break;
  1213. }
  1214. }
  1215. return message;
  1216. }
  1217. static decodeText(reader) {
  1218. const message = new onnx.GraphProto();
  1219. reader.start();
  1220. while (!reader.end()) {
  1221. const tag = reader.tag();
  1222. switch (tag) {
  1223. case "node":
  1224. message.node.push(onnx.NodeProto.decodeText(reader));
  1225. break;
  1226. case "name":
  1227. message.name = reader.string();
  1228. break;
  1229. case "initializer":
  1230. message.initializer.push(onnx.TensorProto.decodeText(reader));
  1231. break;
  1232. case "sparse_initializer":
  1233. message.sparse_initializer.push(onnx.SparseTensorProto.decodeText(reader));
  1234. break;
  1235. case "doc_string":
  1236. message.doc_string = reader.string();
  1237. break;
  1238. case "input":
  1239. message.input.push(onnx.ValueInfoProto.decodeText(reader));
  1240. break;
  1241. case "output":
  1242. message.output.push(onnx.ValueInfoProto.decodeText(reader));
  1243. break;
  1244. case "value_info":
  1245. message.value_info.push(onnx.ValueInfoProto.decodeText(reader));
  1246. break;
  1247. case "quantization_annotation":
  1248. message.quantization_annotation.push(onnx.TensorAnnotation.decodeText(reader));
  1249. break;
  1250. case "metadata_props":
  1251. message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
  1252. break;
  1253. default:
  1254. reader.field(tag, message);
  1255. break;
  1256. }
  1257. }
  1258. return message;
  1259. }
  1260. static decodeJson(obj) {
  1261. const message = new onnx.GraphProto();
  1262. if ('node' in obj) {
  1263. message.node = obj.node.map((obj) => onnx.NodeProto.decodeJson(obj));
  1264. }
  1265. if ('name' in obj) {
  1266. message.name = obj.name;
  1267. }
  1268. if ('initializer' in obj) {
  1269. message.initializer = obj.initializer.map((obj) => onnx.TensorProto.decodeJson(obj));
  1270. }
  1271. if ('sparseInitializer' in obj) {
  1272. message.sparse_initializer = obj.sparseInitializer.map((obj) => onnx.SparseTensorProto.decodeJson(obj));
  1273. }
  1274. if ('docString' in obj) {
  1275. message.doc_string = obj.docString;
  1276. }
  1277. if ('input' in obj) {
  1278. message.input = obj.input.map((obj) => onnx.ValueInfoProto.decodeJson(obj));
  1279. }
  1280. if ('output' in obj) {
  1281. message.output = obj.output.map((obj) => onnx.ValueInfoProto.decodeJson(obj));
  1282. }
  1283. if ('valueInfo' in obj) {
  1284. message.value_info = obj.valueInfo.map((obj) => onnx.ValueInfoProto.decodeJson(obj));
  1285. }
  1286. if ('quantizationAnnotation' in obj) {
  1287. message.quantization_annotation = obj.quantizationAnnotation.map((obj) => onnx.TensorAnnotation.decodeJson(obj));
  1288. }
  1289. if ('metadataProps' in obj) {
  1290. message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
  1291. }
  1292. return message;
  1293. }
  1294. };
  1295. onnx.GraphProto.prototype.name = "";
  1296. onnx.GraphProto.prototype.doc_string = "";
  1297. onnx.TensorProto = class TensorProto {
  1298. constructor() {
  1299. this.dims = [];
  1300. this.float_data = [];
  1301. this.int32_data = [];
  1302. this.string_data = [];
  1303. this.int64_data = [];
  1304. this.external_data = [];
  1305. this.double_data = [];
  1306. this.uint64_data = [];
  1307. this.metadata_props = [];
  1308. }
  1309. static decode(reader, length) {
  1310. const message = new onnx.TensorProto();
  1311. const end = length === undefined ? reader.length : reader.position + length;
  1312. while (reader.position < end) {
  1313. const tag = reader.uint32();
  1314. switch (tag >>> 3) {
  1315. case 1:
  1316. message.dims = reader.array(message.dims, () => reader.int64(), tag);
  1317. break;
  1318. case 2:
  1319. message.data_type = reader.int32();
  1320. break;
  1321. case 3:
  1322. message.segment = onnx.TensorProto.Segment.decode(reader, reader.uint32());
  1323. break;
  1324. case 4:
  1325. message.float_data = reader.floats(message.float_data, tag);
  1326. break;
  1327. case 5:
  1328. message.int32_data = reader.array(message.int32_data, () => reader.int32(), tag);
  1329. break;
  1330. case 6:
  1331. message.string_data.push(reader.bytes());
  1332. break;
  1333. case 7:
  1334. message.int64_data = reader.array(message.int64_data, () => reader.int64(), tag);
  1335. break;
  1336. case 8:
  1337. message.name = reader.string();
  1338. break;
  1339. case 12:
  1340. message.doc_string = reader.string();
  1341. break;
  1342. case 9:
  1343. message.raw_data = reader.bytes();
  1344. break;
  1345. case 13:
  1346. message.external_data.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
  1347. break;
  1348. case 14:
  1349. message.data_location = reader.int32();
  1350. break;
  1351. case 10:
  1352. message.double_data = reader.doubles(message.double_data, tag);
  1353. break;
  1354. case 11:
  1355. message.uint64_data = reader.array(message.uint64_data, () => reader.uint64(), tag);
  1356. break;
  1357. case 16:
  1358. message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
  1359. break;
  1360. default:
  1361. reader.skipType(tag & 7);
  1362. break;
  1363. }
  1364. }
  1365. return message;
  1366. }
  1367. static decodeText(reader) {
  1368. const message = new onnx.TensorProto();
  1369. reader.start();
  1370. while (!reader.end()) {
  1371. const tag = reader.tag();
  1372. switch (tag) {
  1373. case "dims":
  1374. reader.array(message.dims, () => reader.int64());
  1375. break;
  1376. case "data_type":
  1377. message.data_type = reader.int32();
  1378. break;
  1379. case "segment":
  1380. message.segment = onnx.TensorProto.Segment.decodeText(reader);
  1381. break;
  1382. case "float_data":
  1383. reader.array(message.float_data, () => reader.float());
  1384. break;
  1385. case "int32_data":
  1386. reader.array(message.int32_data, () => reader.int32());
  1387. break;
  1388. case "string_data":
  1389. reader.array(message.string_data, () => reader.bytes());
  1390. break;
  1391. case "int64_data":
  1392. reader.array(message.int64_data, () => reader.int64());
  1393. break;
  1394. case "name":
  1395. message.name = reader.string();
  1396. break;
  1397. case "doc_string":
  1398. message.doc_string = reader.string();
  1399. break;
  1400. case "raw_data":
  1401. message.raw_data = reader.bytes();
  1402. break;
  1403. case "external_data":
  1404. message.external_data.push(onnx.StringStringEntryProto.decodeText(reader));
  1405. break;
  1406. case "data_location":
  1407. message.data_location = reader.enum(onnx.TensorProto.DataLocation);
  1408. break;
  1409. case "double_data":
  1410. reader.array(message.double_data, () => reader.double());
  1411. break;
  1412. case "uint64_data":
  1413. reader.array(message.uint64_data, () => reader.uint64());
  1414. break;
  1415. case "metadata_props":
  1416. message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
  1417. break;
  1418. default:
  1419. reader.field(tag, message);
  1420. break;
  1421. }
  1422. }
  1423. return message;
  1424. }
  1425. static decodeJson(obj) {
  1426. const message = new onnx.TensorProto();
  1427. if ('dims' in obj) {
  1428. message.dims = obj.dims.map((obj) => BigInt(obj));
  1429. }
  1430. if ('dataType' in obj) {
  1431. message.data_type = Number(obj.dataType);
  1432. }
  1433. if ('segment' in obj) {
  1434. message.segment = onnx.TensorProto.Segment.decodeJson(obj.segment);
  1435. }
  1436. if ('floatData' in obj) {
  1437. message.float_data = obj.floatData.map((obj) => Number(obj));
  1438. }
  1439. if ('int32Data' in obj) {
  1440. message.int32_data = obj.int32Data.map((obj) => Number(obj));
  1441. }
  1442. if ('stringData' in obj) {
  1443. message.string_data = obj.stringData.map((obj) => typeof obj === 'string' ? Uint8Array.from(atob(obj), (c) => c.charCodeAt(0)) : Uint8Array.from(obj));
  1444. }
  1445. if ('int64Data' in obj) {
  1446. message.int64_data = obj.int64Data.map((obj) => BigInt(obj));
  1447. }
  1448. if ('name' in obj) {
  1449. message.name = obj.name;
  1450. }
  1451. if ('docString' in obj) {
  1452. message.doc_string = obj.docString;
  1453. }
  1454. if ('rawData' in obj) {
  1455. message.raw_data = typeof obj.rawData === 'string' ? Uint8Array.from(atob(obj.rawData), (c) => c.charCodeAt(0)) : Uint8Array.from(obj.rawData);
  1456. }
  1457. if ('externalData' in obj) {
  1458. message.external_data = obj.externalData.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
  1459. }
  1460. if ('dataLocation' in obj) {
  1461. message.data_location = typeof obj.dataLocation === 'string' ? onnx.TensorProto.DataLocation[obj.dataLocation] : obj.dataLocation;
  1462. }
  1463. if ('doubleData' in obj) {
  1464. message.double_data = obj.doubleData.map((obj) => Number(obj));
  1465. }
  1466. if ('uint64Data' in obj) {
  1467. message.uint64_data = obj.uint64Data.map((obj) => BigInt(obj));
  1468. }
  1469. if ('metadataProps' in obj) {
  1470. message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
  1471. }
  1472. return message;
  1473. }
  1474. };
  1475. onnx.TensorProto.prototype.data_type = 0;
  1476. onnx.TensorProto.prototype.segment = null;
  1477. onnx.TensorProto.prototype.name = "";
  1478. onnx.TensorProto.prototype.doc_string = "";
  1479. onnx.TensorProto.prototype.raw_data = new Uint8Array([]);
  1480. onnx.TensorProto.prototype.data_location = 0;
  1481. onnx.TensorProto.DataType = {
  1482. "UNDEFINED": 0,
  1483. "FLOAT": 1,
  1484. "UINT8": 2,
  1485. "INT8": 3,
  1486. "UINT16": 4,
  1487. "INT16": 5,
  1488. "INT32": 6,
  1489. "INT64": 7,
  1490. "STRING": 8,
  1491. "BOOL": 9,
  1492. "FLOAT16": 10,
  1493. "DOUBLE": 11,
  1494. "UINT32": 12,
  1495. "UINT64": 13,
  1496. "COMPLEX64": 14,
  1497. "COMPLEX128": 15,
  1498. "BFLOAT16": 16,
  1499. "FLOAT8E4M3FN": 17,
  1500. "FLOAT8E4M3FNUZ": 18,
  1501. "FLOAT8E5M2": 19,
  1502. "FLOAT8E5M2FNUZ": 20,
  1503. "UINT4": 21,
  1504. "INT4": 22,
  1505. "FLOAT4E2M1": 23
  1506. };
  1507. onnx.TensorProto.Segment = class Segment {
  1508. static decode(reader, length) {
  1509. const message = new onnx.TensorProto.Segment();
  1510. const end = length === undefined ? reader.length : reader.position + length;
  1511. while (reader.position < end) {
  1512. const tag = reader.uint32();
  1513. switch (tag >>> 3) {
  1514. case 1:
  1515. message.begin = reader.int64();
  1516. break;
  1517. case 2:
  1518. message.end = reader.int64();
  1519. break;
  1520. default:
  1521. reader.skipType(tag & 7);
  1522. break;
  1523. }
  1524. }
  1525. return message;
  1526. }
  1527. static decodeText(reader) {
  1528. const message = new onnx.TensorProto.Segment();
  1529. reader.start();
  1530. while (!reader.end()) {
  1531. const tag = reader.tag();
  1532. switch (tag) {
  1533. case "begin":
  1534. message.begin = reader.int64();
  1535. break;
  1536. case "end":
  1537. message.end = reader.int64();
  1538. break;
  1539. default:
  1540. reader.field(tag, message);
  1541. break;
  1542. }
  1543. }
  1544. return message;
  1545. }
  1546. static decodeJson(obj) {
  1547. const message = new onnx.TensorProto.Segment();
  1548. if ('begin' in obj) {
  1549. message.begin = BigInt(obj.begin);
  1550. }
  1551. if ('end' in obj) {
  1552. message.end = BigInt(obj.end);
  1553. }
  1554. return message;
  1555. }
  1556. };
  1557. onnx.TensorProto.Segment.prototype.begin = 0n;
  1558. onnx.TensorProto.Segment.prototype.end = 0n;
  1559. onnx.TensorProto.DataLocation = {
  1560. "DEFAULT": 0,
  1561. "EXTERNAL": 1
  1562. };
  1563. onnx.SparseTensorProto = class SparseTensorProto {
  1564. constructor() {
  1565. this.dims = [];
  1566. }
  1567. static decode(reader, length) {
  1568. const message = new onnx.SparseTensorProto();
  1569. const end = length === undefined ? reader.length : reader.position + length;
  1570. while (reader.position < end) {
  1571. const tag = reader.uint32();
  1572. switch (tag >>> 3) {
  1573. case 1:
  1574. message.values = onnx.TensorProto.decode(reader, reader.uint32());
  1575. break;
  1576. case 2:
  1577. message.indices = onnx.TensorProto.decode(reader, reader.uint32());
  1578. break;
  1579. case 3:
  1580. message.dims = reader.array(message.dims, () => reader.int64(), tag);
  1581. break;
  1582. default:
  1583. reader.skipType(tag & 7);
  1584. break;
  1585. }
  1586. }
  1587. return message;
  1588. }
  1589. static decodeText(reader) {
  1590. const message = new onnx.SparseTensorProto();
  1591. reader.start();
  1592. while (!reader.end()) {
  1593. const tag = reader.tag();
  1594. switch (tag) {
  1595. case "values":
  1596. message.values = onnx.TensorProto.decodeText(reader);
  1597. break;
  1598. case "indices":
  1599. message.indices = onnx.TensorProto.decodeText(reader);
  1600. break;
  1601. case "dims":
  1602. reader.array(message.dims, () => reader.int64());
  1603. break;
  1604. default:
  1605. reader.field(tag, message);
  1606. break;
  1607. }
  1608. }
  1609. return message;
  1610. }
  1611. static decodeJson(obj) {
  1612. const message = new onnx.SparseTensorProto();
  1613. if ('values' in obj) {
  1614. message.values = onnx.TensorProto.decodeJson(obj.values);
  1615. }
  1616. if ('indices' in obj) {
  1617. message.indices = onnx.TensorProto.decodeJson(obj.indices);
  1618. }
  1619. if ('dims' in obj) {
  1620. message.dims = obj.dims.map((obj) => BigInt(obj));
  1621. }
  1622. return message;
  1623. }
  1624. };
  1625. onnx.SparseTensorProto.prototype.values = null;
  1626. onnx.SparseTensorProto.prototype.indices = null;
  1627. onnx.TensorShapeProto = class TensorShapeProto {
  1628. constructor() {
  1629. this.dim = [];
  1630. }
  1631. static decode(reader, length) {
  1632. const message = new onnx.TensorShapeProto();
  1633. const end = length === undefined ? reader.length : reader.position + length;
  1634. while (reader.position < end) {
  1635. const tag = reader.uint32();
  1636. switch (tag >>> 3) {
  1637. case 1:
  1638. message.dim.push(onnx.TensorShapeProto.Dimension.decode(reader, reader.uint32()));
  1639. break;
  1640. default:
  1641. reader.skipType(tag & 7);
  1642. break;
  1643. }
  1644. }
  1645. return message;
  1646. }
  1647. static decodeText(reader) {
  1648. const message = new onnx.TensorShapeProto();
  1649. reader.start();
  1650. while (!reader.end()) {
  1651. const tag = reader.tag();
  1652. switch (tag) {
  1653. case "dim":
  1654. message.dim.push(onnx.TensorShapeProto.Dimension.decodeText(reader));
  1655. break;
  1656. default:
  1657. reader.field(tag, message);
  1658. break;
  1659. }
  1660. }
  1661. return message;
  1662. }
  1663. static decodeJson(obj) {
  1664. const message = new onnx.TensorShapeProto();
  1665. if ('dim' in obj) {
  1666. message.dim = obj.dim.map((obj) => onnx.TensorShapeProto.Dimension.decodeJson(obj));
  1667. }
  1668. return message;
  1669. }
  1670. };
  1671. onnx.TensorShapeProto.Dimension = class Dimension {
  1672. get value() {
  1673. onnx.TensorShapeProto.Dimension.valueSet = onnx.TensorShapeProto.Dimension.valueSet || new Set(["dim_value", "dim_param"]);
  1674. return Object.keys(this).find((key) => onnx.TensorShapeProto.Dimension.valueSet.has(key) && this[key] !== null);
  1675. }
  1676. static decode(reader, length) {
  1677. const message = new onnx.TensorShapeProto.Dimension();
  1678. const end = length === undefined ? reader.length : reader.position + length;
  1679. while (reader.position < end) {
  1680. const tag = reader.uint32();
  1681. switch (tag >>> 3) {
  1682. case 1:
  1683. message.dim_value = reader.int64();
  1684. break;
  1685. case 2:
  1686. message.dim_param = reader.string();
  1687. break;
  1688. case 3:
  1689. message.denotation = reader.string();
  1690. break;
  1691. default:
  1692. reader.skipType(tag & 7);
  1693. break;
  1694. }
  1695. }
  1696. return message;
  1697. }
  1698. static decodeText(reader) {
  1699. const message = new onnx.TensorShapeProto.Dimension();
  1700. reader.start();
  1701. while (!reader.end()) {
  1702. const tag = reader.tag();
  1703. switch (tag) {
  1704. case "dim_value":
  1705. message.dim_value = reader.int64();
  1706. break;
  1707. case "dim_param":
  1708. message.dim_param = reader.string();
  1709. break;
  1710. case "denotation":
  1711. message.denotation = reader.string();
  1712. break;
  1713. default:
  1714. reader.field(tag, message);
  1715. break;
  1716. }
  1717. }
  1718. return message;
  1719. }
  1720. static decodeJson(obj) {
  1721. const message = new onnx.TensorShapeProto.Dimension();
  1722. if ('dimValue' in obj) {
  1723. message.dim_value = BigInt(obj.dimValue);
  1724. }
  1725. if ('dimParam' in obj) {
  1726. message.dim_param = obj.dimParam;
  1727. }
  1728. if ('denotation' in obj) {
  1729. message.denotation = obj.denotation;
  1730. }
  1731. return message;
  1732. }
  1733. };
  1734. onnx.TensorShapeProto.Dimension.prototype.denotation = "";
  1735. onnx.TypeProto = class TypeProto {
  1736. get value() {
  1737. onnx.TypeProto.valueSet = onnx.TypeProto.valueSet || new Set(["tensor_type", "sequence_type", "map_type", "optional_type", "sparse_tensor_type", "opaque_type"]);
  1738. return Object.keys(this).find((key) => onnx.TypeProto.valueSet.has(key) && this[key] !== null);
  1739. }
  1740. static decode(reader, length) {
  1741. const message = new onnx.TypeProto();
  1742. const end = length === undefined ? reader.length : reader.position + length;
  1743. while (reader.position < end) {
  1744. const tag = reader.uint32();
  1745. switch (tag >>> 3) {
  1746. case 1:
  1747. message.tensor_type = onnx.TypeProto.Tensor.decode(reader, reader.uint32());
  1748. break;
  1749. case 4:
  1750. message.sequence_type = onnx.TypeProto.Sequence.decode(reader, reader.uint32());
  1751. break;
  1752. case 5:
  1753. message.map_type = onnx.TypeProto.Map.decode(reader, reader.uint32());
  1754. break;
  1755. case 9:
  1756. message.optional_type = onnx.TypeProto.Optional.decode(reader, reader.uint32());
  1757. break;
  1758. case 8:
  1759. message.sparse_tensor_type = onnx.TypeProto.SparseTensor.decode(reader, reader.uint32());
  1760. break;
  1761. case 7:
  1762. message.opaque_type = onnx.TypeProto.Opaque.decode(reader, reader.uint32());
  1763. break;
  1764. case 6:
  1765. message.denotation = reader.string();
  1766. break;
  1767. default:
  1768. reader.skipType(tag & 7);
  1769. break;
  1770. }
  1771. }
  1772. return message;
  1773. }
  1774. static decodeText(reader) {
  1775. const message = new onnx.TypeProto();
  1776. reader.start();
  1777. while (!reader.end()) {
  1778. const tag = reader.tag();
  1779. switch (tag) {
  1780. case "tensor_type":
  1781. message.tensor_type = onnx.TypeProto.Tensor.decodeText(reader);
  1782. break;
  1783. case "sequence_type":
  1784. message.sequence_type = onnx.TypeProto.Sequence.decodeText(reader);
  1785. break;
  1786. case "map_type":
  1787. message.map_type = onnx.TypeProto.Map.decodeText(reader);
  1788. break;
  1789. case "optional_type":
  1790. message.optional_type = onnx.TypeProto.Optional.decodeText(reader);
  1791. break;
  1792. case "sparse_tensor_type":
  1793. message.sparse_tensor_type = onnx.TypeProto.SparseTensor.decodeText(reader);
  1794. break;
  1795. case "opaque_type":
  1796. message.opaque_type = onnx.TypeProto.Opaque.decodeText(reader);
  1797. break;
  1798. case "denotation":
  1799. message.denotation = reader.string();
  1800. break;
  1801. default:
  1802. reader.field(tag, message);
  1803. break;
  1804. }
  1805. }
  1806. return message;
  1807. }
  1808. static decodeJson(obj) {
  1809. const message = new onnx.TypeProto();
  1810. if ('tensorType' in obj) {
  1811. message.tensor_type = onnx.TypeProto.Tensor.decodeJson(obj.tensorType);
  1812. }
  1813. if ('sequenceType' in obj) {
  1814. message.sequence_type = onnx.TypeProto.Sequence.decodeJson(obj.sequenceType);
  1815. }
  1816. if ('mapType' in obj) {
  1817. message.map_type = onnx.TypeProto.Map.decodeJson(obj.mapType);
  1818. }
  1819. if ('optionalType' in obj) {
  1820. message.optional_type = onnx.TypeProto.Optional.decodeJson(obj.optionalType);
  1821. }
  1822. if ('sparseTensorType' in obj) {
  1823. message.sparse_tensor_type = onnx.TypeProto.SparseTensor.decodeJson(obj.sparseTensorType);
  1824. }
  1825. if ('opaqueType' in obj) {
  1826. message.opaque_type = onnx.TypeProto.Opaque.decodeJson(obj.opaqueType);
  1827. }
  1828. if ('denotation' in obj) {
  1829. message.denotation = obj.denotation;
  1830. }
  1831. return message;
  1832. }
  1833. };
  1834. onnx.TypeProto.prototype.denotation = "";
  1835. onnx.TypeProto.Tensor = class Tensor {
  1836. static decode(reader, length) {
  1837. const message = new onnx.TypeProto.Tensor();
  1838. const end = length === undefined ? reader.length : reader.position + length;
  1839. while (reader.position < end) {
  1840. const tag = reader.uint32();
  1841. switch (tag >>> 3) {
  1842. case 1:
  1843. message.elem_type = reader.int32();
  1844. break;
  1845. case 2:
  1846. message.shape = onnx.TensorShapeProto.decode(reader, reader.uint32());
  1847. break;
  1848. default:
  1849. reader.skipType(tag & 7);
  1850. break;
  1851. }
  1852. }
  1853. return message;
  1854. }
  1855. static decodeText(reader) {
  1856. const message = new onnx.TypeProto.Tensor();
  1857. reader.start();
  1858. while (!reader.end()) {
  1859. const tag = reader.tag();
  1860. switch (tag) {
  1861. case "elem_type":
  1862. message.elem_type = reader.int32();
  1863. break;
  1864. case "shape":
  1865. message.shape = onnx.TensorShapeProto.decodeText(reader);
  1866. break;
  1867. default:
  1868. reader.field(tag, message);
  1869. break;
  1870. }
  1871. }
  1872. return message;
  1873. }
  1874. static decodeJson(obj) {
  1875. const message = new onnx.TypeProto.Tensor();
  1876. if ('elemType' in obj) {
  1877. message.elem_type = Number(obj.elemType);
  1878. }
  1879. if ('shape' in obj) {
  1880. message.shape = onnx.TensorShapeProto.decodeJson(obj.shape);
  1881. }
  1882. return message;
  1883. }
  1884. };
  1885. onnx.TypeProto.Tensor.prototype.elem_type = 0;
  1886. onnx.TypeProto.Tensor.prototype.shape = null;
  1887. onnx.TypeProto.Sequence = class Sequence {
  1888. static decode(reader, length) {
  1889. const message = new onnx.TypeProto.Sequence();
  1890. const end = length === undefined ? reader.length : reader.position + length;
  1891. while (reader.position < end) {
  1892. const tag = reader.uint32();
  1893. switch (tag >>> 3) {
  1894. case 1:
  1895. message.elem_type = onnx.TypeProto.decode(reader, reader.uint32());
  1896. break;
  1897. default:
  1898. reader.skipType(tag & 7);
  1899. break;
  1900. }
  1901. }
  1902. return message;
  1903. }
  1904. static decodeText(reader) {
  1905. const message = new onnx.TypeProto.Sequence();
  1906. reader.start();
  1907. while (!reader.end()) {
  1908. const tag = reader.tag();
  1909. switch (tag) {
  1910. case "elem_type":
  1911. message.elem_type = onnx.TypeProto.decodeText(reader);
  1912. break;
  1913. default:
  1914. reader.field(tag, message);
  1915. break;
  1916. }
  1917. }
  1918. return message;
  1919. }
  1920. static decodeJson(obj) {
  1921. const message = new onnx.TypeProto.Sequence();
  1922. if ('elemType' in obj) {
  1923. message.elem_type = onnx.TypeProto.decodeJson(obj.elemType);
  1924. }
  1925. return message;
  1926. }
  1927. };
  1928. onnx.TypeProto.Sequence.prototype.elem_type = null;
  1929. onnx.TypeProto.Map = class Map {
  1930. static decode(reader, length) {
  1931. const message = new onnx.TypeProto.Map();
  1932. const end = length === undefined ? reader.length : reader.position + length;
  1933. while (reader.position < end) {
  1934. const tag = reader.uint32();
  1935. switch (tag >>> 3) {
  1936. case 1:
  1937. message.key_type = reader.int32();
  1938. break;
  1939. case 2:
  1940. message.value_type = onnx.TypeProto.decode(reader, reader.uint32());
  1941. break;
  1942. default:
  1943. reader.skipType(tag & 7);
  1944. break;
  1945. }
  1946. }
  1947. return message;
  1948. }
  1949. static decodeText(reader) {
  1950. const message = new onnx.TypeProto.Map();
  1951. reader.start();
  1952. while (!reader.end()) {
  1953. const tag = reader.tag();
  1954. switch (tag) {
  1955. case "key_type":
  1956. message.key_type = reader.int32();
  1957. break;
  1958. case "value_type":
  1959. message.value_type = onnx.TypeProto.decodeText(reader);
  1960. break;
  1961. default:
  1962. reader.field(tag, message);
  1963. break;
  1964. }
  1965. }
  1966. return message;
  1967. }
  1968. static decodeJson(obj) {
  1969. const message = new onnx.TypeProto.Map();
  1970. if ('keyType' in obj) {
  1971. message.key_type = Number(obj.keyType);
  1972. }
  1973. if ('valueType' in obj) {
  1974. message.value_type = onnx.TypeProto.decodeJson(obj.valueType);
  1975. }
  1976. return message;
  1977. }
  1978. };
  1979. onnx.TypeProto.Map.prototype.key_type = 0;
  1980. onnx.TypeProto.Map.prototype.value_type = null;
  1981. onnx.TypeProto.Optional = class Optional {
  1982. static decode(reader, length) {
  1983. const message = new onnx.TypeProto.Optional();
  1984. const end = length === undefined ? reader.length : reader.position + length;
  1985. while (reader.position < end) {
  1986. const tag = reader.uint32();
  1987. switch (tag >>> 3) {
  1988. case 1:
  1989. message.elem_type = onnx.TypeProto.decode(reader, reader.uint32());
  1990. break;
  1991. default:
  1992. reader.skipType(tag & 7);
  1993. break;
  1994. }
  1995. }
  1996. return message;
  1997. }
  1998. static decodeText(reader) {
  1999. const message = new onnx.TypeProto.Optional();
  2000. reader.start();
  2001. while (!reader.end()) {
  2002. const tag = reader.tag();
  2003. switch (tag) {
  2004. case "elem_type":
  2005. message.elem_type = onnx.TypeProto.decodeText(reader);
  2006. break;
  2007. default:
  2008. reader.field(tag, message);
  2009. break;
  2010. }
  2011. }
  2012. return message;
  2013. }
  2014. static decodeJson(obj) {
  2015. const message = new onnx.TypeProto.Optional();
  2016. if ('elemType' in obj) {
  2017. message.elem_type = onnx.TypeProto.decodeJson(obj.elemType);
  2018. }
  2019. return message;
  2020. }
  2021. };
  2022. onnx.TypeProto.Optional.prototype.elem_type = null;
  2023. onnx.TypeProto.SparseTensor = class SparseTensor {
  2024. static decode(reader, length) {
  2025. const message = new onnx.TypeProto.SparseTensor();
  2026. const end = length === undefined ? reader.length : reader.position + length;
  2027. while (reader.position < end) {
  2028. const tag = reader.uint32();
  2029. switch (tag >>> 3) {
  2030. case 1:
  2031. message.elem_type = reader.int32();
  2032. break;
  2033. case 2:
  2034. message.shape = onnx.TensorShapeProto.decode(reader, reader.uint32());
  2035. break;
  2036. default:
  2037. reader.skipType(tag & 7);
  2038. break;
  2039. }
  2040. }
  2041. return message;
  2042. }
  2043. static decodeText(reader) {
  2044. const message = new onnx.TypeProto.SparseTensor();
  2045. reader.start();
  2046. while (!reader.end()) {
  2047. const tag = reader.tag();
  2048. switch (tag) {
  2049. case "elem_type":
  2050. message.elem_type = reader.int32();
  2051. break;
  2052. case "shape":
  2053. message.shape = onnx.TensorShapeProto.decodeText(reader);
  2054. break;
  2055. default:
  2056. reader.field(tag, message);
  2057. break;
  2058. }
  2059. }
  2060. return message;
  2061. }
  2062. static decodeJson(obj) {
  2063. const message = new onnx.TypeProto.SparseTensor();
  2064. if ('elemType' in obj) {
  2065. message.elem_type = Number(obj.elemType);
  2066. }
  2067. if ('shape' in obj) {
  2068. message.shape = onnx.TensorShapeProto.decodeJson(obj.shape);
  2069. }
  2070. return message;
  2071. }
  2072. };
  2073. onnx.TypeProto.SparseTensor.prototype.elem_type = 0;
  2074. onnx.TypeProto.SparseTensor.prototype.shape = null;
  2075. onnx.TypeProto.Opaque = class Opaque {
  2076. static decode(reader, length) {
  2077. const message = new onnx.TypeProto.Opaque();
  2078. const end = length === undefined ? reader.length : reader.position + length;
  2079. while (reader.position < end) {
  2080. const tag = reader.uint32();
  2081. switch (tag >>> 3) {
  2082. case 1:
  2083. message.domain = reader.string();
  2084. break;
  2085. case 2:
  2086. message.name = reader.string();
  2087. break;
  2088. default:
  2089. reader.skipType(tag & 7);
  2090. break;
  2091. }
  2092. }
  2093. return message;
  2094. }
  2095. static decodeText(reader) {
  2096. const message = new onnx.TypeProto.Opaque();
  2097. reader.start();
  2098. while (!reader.end()) {
  2099. const tag = reader.tag();
  2100. switch (tag) {
  2101. case "domain":
  2102. message.domain = reader.string();
  2103. break;
  2104. case "name":
  2105. message.name = reader.string();
  2106. break;
  2107. default:
  2108. reader.field(tag, message);
  2109. break;
  2110. }
  2111. }
  2112. return message;
  2113. }
  2114. static decodeJson(obj) {
  2115. const message = new onnx.TypeProto.Opaque();
  2116. if ('domain' in obj) {
  2117. message.domain = obj.domain;
  2118. }
  2119. if ('name' in obj) {
  2120. message.name = obj.name;
  2121. }
  2122. return message;
  2123. }
  2124. };
  2125. onnx.TypeProto.Opaque.prototype.domain = "";
  2126. onnx.TypeProto.Opaque.prototype.name = "";
  2127. onnx.OperatorSetIdProto = class OperatorSetIdProto {
  2128. static decode(reader, length) {
  2129. const message = new onnx.OperatorSetIdProto();
  2130. const end = length === undefined ? reader.length : reader.position + length;
  2131. while (reader.position < end) {
  2132. const tag = reader.uint32();
  2133. switch (tag >>> 3) {
  2134. case 1:
  2135. message.domain = reader.string();
  2136. break;
  2137. case 2:
  2138. message.version = reader.int64();
  2139. break;
  2140. default:
  2141. reader.skipType(tag & 7);
  2142. break;
  2143. }
  2144. }
  2145. return message;
  2146. }
  2147. static decodeText(reader) {
  2148. const message = new onnx.OperatorSetIdProto();
  2149. reader.start();
  2150. while (!reader.end()) {
  2151. const tag = reader.tag();
  2152. switch (tag) {
  2153. case "domain":
  2154. message.domain = reader.string();
  2155. break;
  2156. case "version":
  2157. message.version = reader.int64();
  2158. break;
  2159. default:
  2160. reader.field(tag, message);
  2161. break;
  2162. }
  2163. }
  2164. return message;
  2165. }
  2166. static decodeJson(obj) {
  2167. const message = new onnx.OperatorSetIdProto();
  2168. if ('domain' in obj) {
  2169. message.domain = obj.domain;
  2170. }
  2171. if ('version' in obj) {
  2172. message.version = BigInt(obj.version);
  2173. }
  2174. return message;
  2175. }
  2176. };
  2177. onnx.OperatorSetIdProto.prototype.domain = "";
  2178. onnx.OperatorSetIdProto.prototype.version = 0n;
  2179. onnx.OperatorStatus = {
  2180. "EXPERIMENTAL": 0,
  2181. "STABLE": 1
  2182. };
  2183. onnx.FunctionProto = class FunctionProto {
  2184. constructor() {
  2185. this.input = [];
  2186. this.output = [];
  2187. this.attribute = [];
  2188. this.attribute_proto = [];
  2189. this.node = [];
  2190. this.opset_import = [];
  2191. this.value_info = [];
  2192. this.metadata_props = [];
  2193. }
  2194. static decode(reader, length) {
  2195. const message = new onnx.FunctionProto();
  2196. const end = length === undefined ? reader.length : reader.position + length;
  2197. while (reader.position < end) {
  2198. const tag = reader.uint32();
  2199. switch (tag >>> 3) {
  2200. case 1:
  2201. message.name = reader.string();
  2202. break;
  2203. case 4:
  2204. message.input.push(reader.string());
  2205. break;
  2206. case 5:
  2207. message.output.push(reader.string());
  2208. break;
  2209. case 6:
  2210. message.attribute.push(reader.string());
  2211. break;
  2212. case 11:
  2213. message.attribute_proto.push(onnx.AttributeProto.decode(reader, reader.uint32()));
  2214. break;
  2215. case 7:
  2216. message.node.push(onnx.NodeProto.decode(reader, reader.uint32()));
  2217. break;
  2218. case 8:
  2219. message.doc_string = reader.string();
  2220. break;
  2221. case 9:
  2222. message.opset_import.push(onnx.OperatorSetIdProto.decode(reader, reader.uint32()));
  2223. break;
  2224. case 10:
  2225. message.domain = reader.string();
  2226. break;
  2227. case 13:
  2228. message.overload = reader.string();
  2229. break;
  2230. case 12:
  2231. message.value_info.push(onnx.ValueInfoProto.decode(reader, reader.uint32()));
  2232. break;
  2233. case 14:
  2234. message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
  2235. break;
  2236. default:
  2237. reader.skipType(tag & 7);
  2238. break;
  2239. }
  2240. }
  2241. return message;
  2242. }
  2243. static decodeText(reader) {
  2244. const message = new onnx.FunctionProto();
  2245. reader.start();
  2246. while (!reader.end()) {
  2247. const tag = reader.tag();
  2248. switch (tag) {
  2249. case "name":
  2250. message.name = reader.string();
  2251. break;
  2252. case "input":
  2253. reader.array(message.input, () => reader.string());
  2254. break;
  2255. case "output":
  2256. reader.array(message.output, () => reader.string());
  2257. break;
  2258. case "attribute":
  2259. reader.array(message.attribute, () => reader.string());
  2260. break;
  2261. case "attribute_proto":
  2262. message.attribute_proto.push(onnx.AttributeProto.decodeText(reader));
  2263. break;
  2264. case "node":
  2265. message.node.push(onnx.NodeProto.decodeText(reader));
  2266. break;
  2267. case "doc_string":
  2268. message.doc_string = reader.string();
  2269. break;
  2270. case "opset_import":
  2271. message.opset_import.push(onnx.OperatorSetIdProto.decodeText(reader));
  2272. break;
  2273. case "domain":
  2274. message.domain = reader.string();
  2275. break;
  2276. case "overload":
  2277. message.overload = reader.string();
  2278. break;
  2279. case "value_info":
  2280. message.value_info.push(onnx.ValueInfoProto.decodeText(reader));
  2281. break;
  2282. case "metadata_props":
  2283. message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
  2284. break;
  2285. default:
  2286. reader.field(tag, message);
  2287. break;
  2288. }
  2289. }
  2290. return message;
  2291. }
  2292. static decodeJson(obj) {
  2293. const message = new onnx.FunctionProto();
  2294. if ('name' in obj) {
  2295. message.name = obj.name;
  2296. }
  2297. if ('input' in obj) {
  2298. message.input = obj.input;
  2299. }
  2300. if ('output' in obj) {
  2301. message.output = obj.output;
  2302. }
  2303. if ('attribute' in obj) {
  2304. message.attribute = obj.attribute;
  2305. }
  2306. if ('attributeProto' in obj) {
  2307. message.attribute_proto = obj.attributeProto.map((obj) => onnx.AttributeProto.decodeJson(obj));
  2308. }
  2309. if ('node' in obj) {
  2310. message.node = obj.node.map((obj) => onnx.NodeProto.decodeJson(obj));
  2311. }
  2312. if ('docString' in obj) {
  2313. message.doc_string = obj.docString;
  2314. }
  2315. if ('opsetImport' in obj) {
  2316. message.opset_import = obj.opsetImport.map((obj) => onnx.OperatorSetIdProto.decodeJson(obj));
  2317. }
  2318. if ('domain' in obj) {
  2319. message.domain = obj.domain;
  2320. }
  2321. if ('overload' in obj) {
  2322. message.overload = obj.overload;
  2323. }
  2324. if ('valueInfo' in obj) {
  2325. message.value_info = obj.valueInfo.map((obj) => onnx.ValueInfoProto.decodeJson(obj));
  2326. }
  2327. if ('metadataProps' in obj) {
  2328. message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
  2329. }
  2330. return message;
  2331. }
  2332. };
  2333. onnx.FunctionProto.prototype.name = "";
  2334. onnx.FunctionProto.prototype.doc_string = "";
  2335. onnx.FunctionProto.prototype.domain = "";
  2336. onnx.FunctionProto.prototype.overload = "";
  2337. onnx.OperatorProto = class OperatorProto {
  2338. static decode(reader, length) {
  2339. const message = new onnx.OperatorProto();
  2340. const end = length === undefined ? reader.length : reader.position + length;
  2341. while (reader.position < end) {
  2342. const tag = reader.uint32();
  2343. switch (tag >>> 3) {
  2344. case 1:
  2345. message.op_type = reader.string();
  2346. break;
  2347. case 2:
  2348. message.since_version = reader.int64();
  2349. break;
  2350. case 3:
  2351. message.status = reader.int32();
  2352. break;
  2353. case 10:
  2354. message.doc_string = reader.string();
  2355. break;
  2356. default:
  2357. reader.skipType(tag & 7);
  2358. break;
  2359. }
  2360. }
  2361. return message;
  2362. }
  2363. static decodeText(reader) {
  2364. const message = new onnx.OperatorProto();
  2365. reader.start();
  2366. while (!reader.end()) {
  2367. const tag = reader.tag();
  2368. switch (tag) {
  2369. case "op_type":
  2370. message.op_type = reader.string();
  2371. break;
  2372. case "since_version":
  2373. message.since_version = reader.int64();
  2374. break;
  2375. case "status":
  2376. message.status = reader.enum(onnx.OperatorStatus);
  2377. break;
  2378. case "doc_string":
  2379. message.doc_string = reader.string();
  2380. break;
  2381. default:
  2382. reader.field(tag, message);
  2383. break;
  2384. }
  2385. }
  2386. return message;
  2387. }
  2388. static decodeJson(obj) {
  2389. const message = new onnx.OperatorProto();
  2390. if ('opType' in obj) {
  2391. message.op_type = obj.opType;
  2392. }
  2393. if ('sinceVersion' in obj) {
  2394. message.since_version = BigInt(obj.sinceVersion);
  2395. }
  2396. if ('status' in obj) {
  2397. message.status = typeof obj.status === 'string' ? onnx.OperatorStatus[obj.status] : obj.status;
  2398. }
  2399. if ('docString' in obj) {
  2400. message.doc_string = obj.docString;
  2401. }
  2402. return message;
  2403. }
  2404. };
  2405. onnx.OperatorProto.prototype.op_type = "";
  2406. onnx.OperatorProto.prototype.since_version = 0n;
  2407. onnx.OperatorProto.prototype.status = 0;
  2408. onnx.OperatorProto.prototype.doc_string = "";
  2409. onnx.OperatorSetProto = class OperatorSetProto {
  2410. constructor() {
  2411. this.operator = [];
  2412. this.functions = [];
  2413. }
  2414. static decode(reader, length) {
  2415. const message = new onnx.OperatorSetProto();
  2416. const end = length === undefined ? reader.length : reader.position + length;
  2417. while (reader.position < end) {
  2418. const tag = reader.uint32();
  2419. switch (tag >>> 3) {
  2420. case 1:
  2421. message.magic = reader.string();
  2422. break;
  2423. case 2:
  2424. message.ir_version = reader.int64();
  2425. break;
  2426. case 3:
  2427. message.ir_version_prerelease = reader.string();
  2428. break;
  2429. case 7:
  2430. message.ir_build_metadata = reader.string();
  2431. break;
  2432. case 4:
  2433. message.domain = reader.string();
  2434. break;
  2435. case 5:
  2436. message.opset_version = reader.int64();
  2437. break;
  2438. case 6:
  2439. message.doc_string = reader.string();
  2440. break;
  2441. case 8:
  2442. message.operator.push(onnx.OperatorProto.decode(reader, reader.uint32()));
  2443. break;
  2444. case 9:
  2445. message.functions.push(onnx.FunctionProto.decode(reader, reader.uint32()));
  2446. break;
  2447. default:
  2448. reader.skipType(tag & 7);
  2449. break;
  2450. }
  2451. }
  2452. return message;
  2453. }
  2454. static decodeText(reader) {
  2455. const message = new onnx.OperatorSetProto();
  2456. reader.start();
  2457. while (!reader.end()) {
  2458. const tag = reader.tag();
  2459. switch (tag) {
  2460. case "magic":
  2461. message.magic = reader.string();
  2462. break;
  2463. case "ir_version":
  2464. message.ir_version = reader.int64();
  2465. break;
  2466. case "ir_version_prerelease":
  2467. message.ir_version_prerelease = reader.string();
  2468. break;
  2469. case "ir_build_metadata":
  2470. message.ir_build_metadata = reader.string();
  2471. break;
  2472. case "domain":
  2473. message.domain = reader.string();
  2474. break;
  2475. case "opset_version":
  2476. message.opset_version = reader.int64();
  2477. break;
  2478. case "doc_string":
  2479. message.doc_string = reader.string();
  2480. break;
  2481. case "operator":
  2482. message.operator.push(onnx.OperatorProto.decodeText(reader));
  2483. break;
  2484. case "functions":
  2485. message.functions.push(onnx.FunctionProto.decodeText(reader));
  2486. break;
  2487. default:
  2488. reader.field(tag, message);
  2489. break;
  2490. }
  2491. }
  2492. return message;
  2493. }
  2494. static decodeJson(obj) {
  2495. const message = new onnx.OperatorSetProto();
  2496. if ('magic' in obj) {
  2497. message.magic = obj.magic;
  2498. }
  2499. if ('irVersion' in obj) {
  2500. message.ir_version = BigInt(obj.irVersion);
  2501. }
  2502. if ('irVersionPrerelease' in obj) {
  2503. message.ir_version_prerelease = obj.irVersionPrerelease;
  2504. }
  2505. if ('irBuildMetadata' in obj) {
  2506. message.ir_build_metadata = obj.irBuildMetadata;
  2507. }
  2508. if ('domain' in obj) {
  2509. message.domain = obj.domain;
  2510. }
  2511. if ('opsetVersion' in obj) {
  2512. message.opset_version = BigInt(obj.opsetVersion);
  2513. }
  2514. if ('docString' in obj) {
  2515. message.doc_string = obj.docString;
  2516. }
  2517. if ('operator' in obj) {
  2518. message.operator = obj.operator.map((obj) => onnx.OperatorProto.decodeJson(obj));
  2519. }
  2520. if ('functions' in obj) {
  2521. message.functions = obj.functions.map((obj) => onnx.FunctionProto.decodeJson(obj));
  2522. }
  2523. return message;
  2524. }
  2525. };
  2526. onnx.OperatorSetProto.prototype.magic = "";
  2527. onnx.OperatorSetProto.prototype.ir_version = 0n;
  2528. onnx.OperatorSetProto.prototype.ir_version_prerelease = "";
  2529. onnx.OperatorSetProto.prototype.ir_build_metadata = "";
  2530. onnx.OperatorSetProto.prototype.domain = "";
  2531. onnx.OperatorSetProto.prototype.opset_version = 0n;
  2532. onnx.OperatorSetProto.prototype.doc_string = "";