| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707 |
- export const onnx = {};
- onnx.Version = {
- "_START_VERSION": 0,
- "IR_VERSION_2017_10_10": 1,
- "IR_VERSION_2017_10_30": 2,
- "IR_VERSION_2017_11_3": 3,
- "IR_VERSION_2019_1_22": 4,
- "IR_VERSION_2019_3_18": 5,
- "IR_VERSION_2019_9_19": 6,
- "IR_VERSION_2020_5_8": 7,
- "IR_VERSION_2021_7_30": 8,
- "IR_VERSION_2023_5_5": 9,
- "IR_VERSION_2024_3_25": 10,
- "IR_VERSION": 11
- };
- onnx.AttributeProto = class AttributeProto {
- constructor() {
- this.floats = [];
- this.ints = [];
- this.strings = [];
- this.tensors = [];
- this.graphs = [];
- this.sparse_tensors = [];
- this.type_protos = [];
- }
- static decode(reader, length) {
- const message = new onnx.AttributeProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.name = reader.string();
- break;
- case 21:
- message.ref_attr_name = reader.string();
- break;
- case 13:
- message.doc_string = reader.string();
- break;
- case 20:
- message.type = reader.int32();
- break;
- case 2:
- message.f = reader.float();
- break;
- case 3:
- message.i = reader.int64();
- break;
- case 4:
- message.s = reader.bytes();
- break;
- case 5:
- message.t = onnx.TensorProto.decode(reader, reader.uint32());
- break;
- case 6:
- message.g = onnx.GraphProto.decode(reader, reader.uint32());
- break;
- case 22:
- message.sparse_tensor = onnx.SparseTensorProto.decode(reader, reader.uint32());
- break;
- case 14:
- message.tp = onnx.TypeProto.decode(reader, reader.uint32());
- break;
- case 7:
- message.floats = reader.floats(message.floats, tag);
- break;
- case 8:
- message.ints = reader.array(message.ints, () => reader.int64(), tag);
- break;
- case 9:
- message.strings.push(reader.bytes());
- break;
- case 10:
- message.tensors.push(onnx.TensorProto.decode(reader, reader.uint32()));
- break;
- case 11:
- message.graphs.push(onnx.GraphProto.decode(reader, reader.uint32()));
- break;
- case 23:
- message.sparse_tensors.push(onnx.SparseTensorProto.decode(reader, reader.uint32()));
- break;
- case 15:
- message.type_protos.push(onnx.TypeProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.AttributeProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "name":
- message.name = reader.string();
- break;
- case "ref_attr_name":
- message.ref_attr_name = reader.string();
- break;
- case "doc_string":
- message.doc_string = reader.string();
- break;
- case "type":
- message.type = reader.enum(onnx.AttributeProto.AttributeType);
- break;
- case "f":
- message.f = reader.float();
- break;
- case "i":
- message.i = reader.int64();
- break;
- case "s":
- message.s = reader.bytes();
- break;
- case "t":
- message.t = onnx.TensorProto.decodeText(reader);
- break;
- case "g":
- message.g = onnx.GraphProto.decodeText(reader);
- break;
- case "sparse_tensor":
- message.sparse_tensor = onnx.SparseTensorProto.decodeText(reader);
- break;
- case "tp":
- message.tp = onnx.TypeProto.decodeText(reader);
- break;
- case "floats":
- reader.array(message.floats, () => reader.float());
- break;
- case "ints":
- reader.array(message.ints, () => reader.int64());
- break;
- case "strings":
- reader.array(message.strings, () => reader.bytes());
- break;
- case "tensors":
- message.tensors.push(onnx.TensorProto.decodeText(reader));
- break;
- case "graphs":
- message.graphs.push(onnx.GraphProto.decodeText(reader));
- break;
- case "sparse_tensors":
- message.sparse_tensors.push(onnx.SparseTensorProto.decodeText(reader));
- break;
- case "type_protos":
- message.type_protos.push(onnx.TypeProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.AttributeProto();
- if ('name' in obj) {
- message.name = obj.name;
- }
- if ('refAttrName' in obj) {
- message.ref_attr_name = obj.refAttrName;
- }
- if ('docString' in obj) {
- message.doc_string = obj.docString;
- }
- if ('type' in obj) {
- message.type = typeof obj.type === 'string' ? onnx.AttributeProto.AttributeType[obj.type] : obj.type;
- }
- if ('f' in obj) {
- message.f = Number(obj.f);
- }
- if ('i' in obj) {
- message.i = BigInt(obj.i);
- }
- if ('s' in obj) {
- message.s = typeof obj.s === 'string' ? Uint8Array.from(atob(obj.s), (c) => c.charCodeAt(0)) : Uint8Array.from(obj.s);
- }
- if ('t' in obj) {
- message.t = onnx.TensorProto.decodeJson(obj.t);
- }
- if ('g' in obj) {
- message.g = onnx.GraphProto.decodeJson(obj.g);
- }
- if ('sparseTensor' in obj) {
- message.sparse_tensor = onnx.SparseTensorProto.decodeJson(obj.sparseTensor);
- }
- if ('tp' in obj) {
- message.tp = onnx.TypeProto.decodeJson(obj.tp);
- }
- if ('floats' in obj) {
- message.floats = obj.floats.map((obj) => Number(obj));
- }
- if ('ints' in obj) {
- message.ints = obj.ints.map((obj) => BigInt(obj));
- }
- if ('strings' in obj) {
- message.strings = obj.strings.map((obj) => typeof obj === 'string' ? Uint8Array.from(atob(obj), (c) => c.charCodeAt(0)) : Uint8Array.from(obj));
- }
- if ('tensors' in obj) {
- message.tensors = obj.tensors.map((obj) => onnx.TensorProto.decodeJson(obj));
- }
- if ('graphs' in obj) {
- message.graphs = obj.graphs.map((obj) => onnx.GraphProto.decodeJson(obj));
- }
- if ('sparseTensors' in obj) {
- message.sparse_tensors = obj.sparseTensors.map((obj) => onnx.SparseTensorProto.decodeJson(obj));
- }
- if ('typeProtos' in obj) {
- message.type_protos = obj.typeProtos.map((obj) => onnx.TypeProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.AttributeProto.prototype.name = "";
- onnx.AttributeProto.prototype.ref_attr_name = "";
- onnx.AttributeProto.prototype.doc_string = "";
- onnx.AttributeProto.prototype.type = 0;
- onnx.AttributeProto.prototype.f = 0;
- onnx.AttributeProto.prototype.i = 0n;
- onnx.AttributeProto.prototype.s = new Uint8Array([]);
- onnx.AttributeProto.prototype.t = null;
- onnx.AttributeProto.prototype.g = null;
- onnx.AttributeProto.prototype.sparse_tensor = null;
- onnx.AttributeProto.prototype.tp = null;
- onnx.AttributeProto.AttributeType = {
- "UNDEFINED": 0,
- "FLOAT": 1,
- "INT": 2,
- "STRING": 3,
- "TENSOR": 4,
- "GRAPH": 5,
- "SPARSE_TENSOR": 11,
- "TYPE_PROTO": 13,
- "FLOATS": 6,
- "INTS": 7,
- "STRINGS": 8,
- "TENSORS": 9,
- "GRAPHS": 10,
- "SPARSE_TENSORS": 12,
- "TYPE_PROTOS": 14
- };
- onnx.ValueInfoProto = class ValueInfoProto {
- constructor() {
- this.metadata_props = [];
- }
- static decode(reader, length) {
- const message = new onnx.ValueInfoProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.name = reader.string();
- break;
- case 2:
- message.type = onnx.TypeProto.decode(reader, reader.uint32());
- break;
- case 3:
- message.doc_string = reader.string();
- break;
- case 4:
- message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.ValueInfoProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "name":
- message.name = reader.string();
- break;
- case "type":
- message.type = onnx.TypeProto.decodeText(reader);
- break;
- case "doc_string":
- message.doc_string = reader.string();
- break;
- case "metadata_props":
- message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.ValueInfoProto();
- if ('name' in obj) {
- message.name = obj.name;
- }
- if ('type' in obj) {
- message.type = onnx.TypeProto.decodeJson(obj.type);
- }
- if ('docString' in obj) {
- message.doc_string = obj.docString;
- }
- if ('metadataProps' in obj) {
- message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.ValueInfoProto.prototype.name = "";
- onnx.ValueInfoProto.prototype.type = null;
- onnx.ValueInfoProto.prototype.doc_string = "";
- onnx.NodeProto = class NodeProto {
- constructor() {
- this.input = [];
- this.output = [];
- this.attribute = [];
- this.metadata_props = [];
- this.device_configurations = [];
- }
- static decode(reader, length) {
- const message = new onnx.NodeProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.input.push(reader.string());
- break;
- case 2:
- message.output.push(reader.string());
- break;
- case 3:
- message.name = reader.string();
- break;
- case 4:
- message.op_type = reader.string();
- break;
- case 7:
- message.domain = reader.string();
- break;
- case 8:
- message.overload = reader.string();
- break;
- case 5:
- message.attribute.push(onnx.AttributeProto.decode(reader, reader.uint32()));
- break;
- case 6:
- message.doc_string = reader.string();
- break;
- case 9:
- message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
- break;
- case 10:
- message.device_configurations.push(onnx.NodeDeviceConfigurationProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.NodeProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "input":
- reader.array(message.input, () => reader.string());
- break;
- case "output":
- reader.array(message.output, () => reader.string());
- break;
- case "name":
- message.name = reader.string();
- break;
- case "op_type":
- message.op_type = reader.string();
- break;
- case "domain":
- message.domain = reader.string();
- break;
- case "overload":
- message.overload = reader.string();
- break;
- case "attribute":
- message.attribute.push(onnx.AttributeProto.decodeText(reader));
- break;
- case "doc_string":
- message.doc_string = reader.string();
- break;
- case "metadata_props":
- message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
- break;
- case "device_configurations":
- message.device_configurations.push(onnx.NodeDeviceConfigurationProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.NodeProto();
- if ('input' in obj) {
- message.input = obj.input;
- }
- if ('output' in obj) {
- message.output = obj.output;
- }
- if ('name' in obj) {
- message.name = obj.name;
- }
- if ('opType' in obj) {
- message.op_type = obj.opType;
- }
- if ('domain' in obj) {
- message.domain = obj.domain;
- }
- if ('overload' in obj) {
- message.overload = obj.overload;
- }
- if ('attribute' in obj) {
- message.attribute = obj.attribute.map((obj) => onnx.AttributeProto.decodeJson(obj));
- }
- if ('docString' in obj) {
- message.doc_string = obj.docString;
- }
- if ('metadataProps' in obj) {
- message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
- }
- if ('deviceConfigurations' in obj) {
- message.device_configurations = obj.deviceConfigurations.map((obj) => onnx.NodeDeviceConfigurationProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.NodeProto.prototype.name = "";
- onnx.NodeProto.prototype.op_type = "";
- onnx.NodeProto.prototype.domain = "";
- onnx.NodeProto.prototype.overload = "";
- onnx.NodeProto.prototype.doc_string = "";
- onnx.IntIntListEntryProto = class IntIntListEntryProto {
- constructor() {
- this.value = [];
- }
- static decode(reader, length) {
- const message = new onnx.IntIntListEntryProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.key = reader.int64();
- break;
- case 2:
- message.value = reader.array(message.value, () => reader.int64(), tag);
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.IntIntListEntryProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "key":
- message.key = reader.int64();
- break;
- case "value":
- reader.array(message.value, () => reader.int64());
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.IntIntListEntryProto();
- if ('key' in obj) {
- message.key = BigInt(obj.key);
- }
- if ('value' in obj) {
- message.value = obj.value.map((obj) => BigInt(obj));
- }
- return message;
- }
- };
- onnx.IntIntListEntryProto.prototype.key = 0n;
- onnx.NodeDeviceConfigurationProto = class NodeDeviceConfigurationProto {
- constructor() {
- this.sharding_spec = [];
- }
- static decode(reader, length) {
- const message = new onnx.NodeDeviceConfigurationProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.configuration_id = reader.string();
- break;
- case 2:
- message.sharding_spec.push(onnx.ShardingSpecProto.decode(reader, reader.uint32()));
- break;
- case 3:
- message.pipeline_stage = reader.int32();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.NodeDeviceConfigurationProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "configuration_id":
- message.configuration_id = reader.string();
- break;
- case "sharding_spec":
- message.sharding_spec.push(onnx.ShardingSpecProto.decodeText(reader));
- break;
- case "pipeline_stage":
- message.pipeline_stage = reader.int32();
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.NodeDeviceConfigurationProto();
- if ('configurationId' in obj) {
- message.configuration_id = obj.configurationId;
- }
- if ('shardingSpec' in obj) {
- message.sharding_spec = obj.shardingSpec.map((obj) => onnx.ShardingSpecProto.decodeJson(obj));
- }
- if ('pipelineStage' in obj) {
- message.pipeline_stage = Number(obj.pipelineStage);
- }
- return message;
- }
- };
- onnx.NodeDeviceConfigurationProto.prototype.configuration_id = "";
- onnx.NodeDeviceConfigurationProto.prototype.pipeline_stage = 0;
- onnx.ShardingSpecProto = class ShardingSpecProto {
- constructor() {
- this.device = [];
- this.index_to_device_group_map = [];
- this.sharded_dim = [];
- }
- static decode(reader, length) {
- const message = new onnx.ShardingSpecProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.tensor_name = reader.string();
- break;
- case 2:
- message.device = reader.array(message.device, () => reader.int64(), tag);
- break;
- case 3:
- message.index_to_device_group_map.push(onnx.IntIntListEntryProto.decode(reader, reader.uint32()));
- break;
- case 4:
- message.sharded_dim.push(onnx.ShardedDimProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.ShardingSpecProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "tensor_name":
- message.tensor_name = reader.string();
- break;
- case "device":
- reader.array(message.device, () => reader.int64());
- break;
- case "index_to_device_group_map":
- message.index_to_device_group_map.push(onnx.IntIntListEntryProto.decodeText(reader));
- break;
- case "sharded_dim":
- message.sharded_dim.push(onnx.ShardedDimProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.ShardingSpecProto();
- if ('tensorName' in obj) {
- message.tensor_name = obj.tensorName;
- }
- if ('device' in obj) {
- message.device = obj.device.map((obj) => BigInt(obj));
- }
- if ('indexToDeviceGroupMap' in obj) {
- message.index_to_device_group_map = obj.indexToDeviceGroupMap.map((obj) => onnx.IntIntListEntryProto.decodeJson(obj));
- }
- if ('shardedDim' in obj) {
- message.sharded_dim = obj.shardedDim.map((obj) => onnx.ShardedDimProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.ShardingSpecProto.prototype.tensor_name = "";
- onnx.ShardedDimProto = class ShardedDimProto {
- constructor() {
- this.simple_sharding = [];
- }
- static decode(reader, length) {
- const message = new onnx.ShardedDimProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.axis = reader.int64();
- break;
- case 2:
- message.simple_sharding.push(onnx.SimpleShardedDimProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.ShardedDimProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "axis":
- message.axis = reader.int64();
- break;
- case "simple_sharding":
- message.simple_sharding.push(onnx.SimpleShardedDimProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.ShardedDimProto();
- if ('axis' in obj) {
- message.axis = BigInt(obj.axis);
- }
- if ('simpleSharding' in obj) {
- message.simple_sharding = obj.simpleSharding.map((obj) => onnx.SimpleShardedDimProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.ShardedDimProto.prototype.axis = 0n;
- onnx.SimpleShardedDimProto = class SimpleShardedDimProto {
- get dim() {
- onnx.SimpleShardedDimProto.dimSet = onnx.SimpleShardedDimProto.dimSet || new Set(["dim_value", "dim_param"]);
- return Object.keys(this).find((key) => onnx.SimpleShardedDimProto.dimSet.has(key) && this[key] !== null);
- }
- static decode(reader, length) {
- const message = new onnx.SimpleShardedDimProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.dim_value = reader.int64();
- break;
- case 2:
- message.dim_param = reader.string();
- break;
- case 3:
- message.num_shards = reader.int64();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.SimpleShardedDimProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "dim_value":
- message.dim_value = reader.int64();
- break;
- case "dim_param":
- message.dim_param = reader.string();
- break;
- case "num_shards":
- message.num_shards = reader.int64();
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.SimpleShardedDimProto();
- if ('dimValue' in obj) {
- message.dim_value = BigInt(obj.dimValue);
- }
- if ('dimParam' in obj) {
- message.dim_param = obj.dimParam;
- }
- if ('numShards' in obj) {
- message.num_shards = BigInt(obj.numShards);
- }
- return message;
- }
- };
- onnx.SimpleShardedDimProto.prototype.num_shards = 0n;
- onnx.TrainingInfoProto = class TrainingInfoProto {
- constructor() {
- this.initialization_binding = [];
- this.update_binding = [];
- }
- static decode(reader, length) {
- const message = new onnx.TrainingInfoProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.initialization = onnx.GraphProto.decode(reader, reader.uint32());
- break;
- case 2:
- message.algorithm = onnx.GraphProto.decode(reader, reader.uint32());
- break;
- case 3:
- message.initialization_binding.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
- break;
- case 4:
- message.update_binding.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TrainingInfoProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "initialization":
- message.initialization = onnx.GraphProto.decodeText(reader);
- break;
- case "algorithm":
- message.algorithm = onnx.GraphProto.decodeText(reader);
- break;
- case "initialization_binding":
- message.initialization_binding.push(onnx.StringStringEntryProto.decodeText(reader));
- break;
- case "update_binding":
- message.update_binding.push(onnx.StringStringEntryProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TrainingInfoProto();
- if ('initialization' in obj) {
- message.initialization = onnx.GraphProto.decodeJson(obj.initialization);
- }
- if ('algorithm' in obj) {
- message.algorithm = onnx.GraphProto.decodeJson(obj.algorithm);
- }
- if ('initializationBinding' in obj) {
- message.initialization_binding = obj.initializationBinding.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
- }
- if ('updateBinding' in obj) {
- message.update_binding = obj.updateBinding.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.TrainingInfoProto.prototype.initialization = null;
- onnx.TrainingInfoProto.prototype.algorithm = null;
- onnx.ModelProto = class ModelProto {
- constructor() {
- this.opset_import = [];
- this.metadata_props = [];
- this.training_info = [];
- this.functions = [];
- this.configuration = [];
- }
- static decode(reader, length) {
- const message = new onnx.ModelProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.ir_version = reader.int64();
- break;
- case 8:
- message.opset_import.push(onnx.OperatorSetIdProto.decode(reader, reader.uint32()));
- break;
- case 2:
- message.producer_name = reader.string();
- break;
- case 3:
- message.producer_version = reader.string();
- break;
- case 4:
- message.domain = reader.string();
- break;
- case 5:
- message.model_version = reader.int64();
- break;
- case 6:
- message.doc_string = reader.string();
- break;
- case 7:
- message.graph = onnx.GraphProto.decode(reader, reader.uint32());
- break;
- case 14:
- message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
- break;
- case 20:
- message.training_info.push(onnx.TrainingInfoProto.decode(reader, reader.uint32()));
- break;
- case 25:
- message.functions.push(onnx.FunctionProto.decode(reader, reader.uint32()));
- break;
- case 26:
- message.configuration.push(onnx.DeviceConfigurationProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.ModelProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "ir_version":
- message.ir_version = reader.int64();
- break;
- case "opset_import":
- message.opset_import.push(onnx.OperatorSetIdProto.decodeText(reader));
- break;
- case "producer_name":
- message.producer_name = reader.string();
- break;
- case "producer_version":
- message.producer_version = reader.string();
- break;
- case "domain":
- message.domain = reader.string();
- break;
- case "model_version":
- message.model_version = reader.int64();
- break;
- case "doc_string":
- message.doc_string = reader.string();
- break;
- case "graph":
- message.graph = onnx.GraphProto.decodeText(reader);
- break;
- case "metadata_props":
- message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
- break;
- case "training_info":
- message.training_info.push(onnx.TrainingInfoProto.decodeText(reader));
- break;
- case "functions":
- message.functions.push(onnx.FunctionProto.decodeText(reader));
- break;
- case "configuration":
- message.configuration.push(onnx.DeviceConfigurationProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.ModelProto();
- if ('irVersion' in obj) {
- message.ir_version = BigInt(obj.irVersion);
- }
- if ('opsetImport' in obj) {
- message.opset_import = obj.opsetImport.map((obj) => onnx.OperatorSetIdProto.decodeJson(obj));
- }
- if ('producerName' in obj) {
- message.producer_name = obj.producerName;
- }
- if ('producerVersion' in obj) {
- message.producer_version = obj.producerVersion;
- }
- if ('domain' in obj) {
- message.domain = obj.domain;
- }
- if ('modelVersion' in obj) {
- message.model_version = BigInt(obj.modelVersion);
- }
- if ('docString' in obj) {
- message.doc_string = obj.docString;
- }
- if ('graph' in obj) {
- message.graph = onnx.GraphProto.decodeJson(obj.graph);
- }
- if ('metadataProps' in obj) {
- message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
- }
- if ('trainingInfo' in obj) {
- message.training_info = obj.trainingInfo.map((obj) => onnx.TrainingInfoProto.decodeJson(obj));
- }
- if ('functions' in obj) {
- message.functions = obj.functions.map((obj) => onnx.FunctionProto.decodeJson(obj));
- }
- if ('configuration' in obj) {
- message.configuration = obj.configuration.map((obj) => onnx.DeviceConfigurationProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.ModelProto.prototype.ir_version = 0n;
- onnx.ModelProto.prototype.producer_name = "";
- onnx.ModelProto.prototype.producer_version = "";
- onnx.ModelProto.prototype.domain = "";
- onnx.ModelProto.prototype.model_version = 0n;
- onnx.ModelProto.prototype.doc_string = "";
- onnx.ModelProto.prototype.graph = null;
- onnx.DeviceConfigurationProto = class DeviceConfigurationProto {
- constructor() {
- this.device = [];
- }
- static decode(reader, length) {
- const message = new onnx.DeviceConfigurationProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.name = reader.string();
- break;
- case 2:
- message.num_devices = reader.int32();
- break;
- case 3:
- message.device.push(reader.string());
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.DeviceConfigurationProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "name":
- message.name = reader.string();
- break;
- case "num_devices":
- message.num_devices = reader.int32();
- break;
- case "device":
- reader.array(message.device, () => reader.string());
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.DeviceConfigurationProto();
- if ('name' in obj) {
- message.name = obj.name;
- }
- if ('numDevices' in obj) {
- message.num_devices = Number(obj.numDevices);
- }
- if ('device' in obj) {
- message.device = obj.device;
- }
- return message;
- }
- };
- onnx.DeviceConfigurationProto.prototype.name = "";
- onnx.DeviceConfigurationProto.prototype.num_devices = 0;
- onnx.StringStringEntryProto = class StringStringEntryProto {
- static decode(reader, length) {
- const message = new onnx.StringStringEntryProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.key = reader.string();
- break;
- case 2:
- message.value = reader.string();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.StringStringEntryProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "key":
- message.key = reader.string();
- break;
- case "value":
- message.value = reader.string();
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.StringStringEntryProto();
- if ('key' in obj) {
- message.key = obj.key;
- }
- if ('value' in obj) {
- message.value = obj.value;
- }
- return message;
- }
- };
- onnx.StringStringEntryProto.prototype.key = "";
- onnx.StringStringEntryProto.prototype.value = "";
- onnx.TensorAnnotation = class TensorAnnotation {
- constructor() {
- this.quant_parameter_tensor_names = [];
- }
- static decode(reader, length) {
- const message = new onnx.TensorAnnotation();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.tensor_name = reader.string();
- break;
- case 2:
- message.quant_parameter_tensor_names.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TensorAnnotation();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "tensor_name":
- message.tensor_name = reader.string();
- break;
- case "quant_parameter_tensor_names":
- message.quant_parameter_tensor_names.push(onnx.StringStringEntryProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TensorAnnotation();
- if ('tensorName' in obj) {
- message.tensor_name = obj.tensorName;
- }
- if ('quantParameterTensorNames' in obj) {
- message.quant_parameter_tensor_names = obj.quantParameterTensorNames.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.TensorAnnotation.prototype.tensor_name = "";
- onnx.GraphProto = class GraphProto {
- constructor() {
- this.node = [];
- this.initializer = [];
- this.sparse_initializer = [];
- this.input = [];
- this.output = [];
- this.value_info = [];
- this.quantization_annotation = [];
- this.metadata_props = [];
- }
- static decode(reader, length) {
- const message = new onnx.GraphProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.node.push(onnx.NodeProto.decode(reader, reader.uint32()));
- break;
- case 2:
- message.name = reader.string();
- break;
- case 5:
- message.initializer.push(onnx.TensorProto.decode(reader, reader.uint32()));
- break;
- case 15:
- message.sparse_initializer.push(onnx.SparseTensorProto.decode(reader, reader.uint32()));
- break;
- case 10:
- message.doc_string = reader.string();
- break;
- case 11:
- message.input.push(onnx.ValueInfoProto.decode(reader, reader.uint32()));
- break;
- case 12:
- message.output.push(onnx.ValueInfoProto.decode(reader, reader.uint32()));
- break;
- case 13:
- message.value_info.push(onnx.ValueInfoProto.decode(reader, reader.uint32()));
- break;
- case 14:
- message.quantization_annotation.push(onnx.TensorAnnotation.decode(reader, reader.uint32()));
- break;
- case 16:
- message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.GraphProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "node":
- message.node.push(onnx.NodeProto.decodeText(reader));
- break;
- case "name":
- message.name = reader.string();
- break;
- case "initializer":
- message.initializer.push(onnx.TensorProto.decodeText(reader));
- break;
- case "sparse_initializer":
- message.sparse_initializer.push(onnx.SparseTensorProto.decodeText(reader));
- break;
- case "doc_string":
- message.doc_string = reader.string();
- break;
- case "input":
- message.input.push(onnx.ValueInfoProto.decodeText(reader));
- break;
- case "output":
- message.output.push(onnx.ValueInfoProto.decodeText(reader));
- break;
- case "value_info":
- message.value_info.push(onnx.ValueInfoProto.decodeText(reader));
- break;
- case "quantization_annotation":
- message.quantization_annotation.push(onnx.TensorAnnotation.decodeText(reader));
- break;
- case "metadata_props":
- message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.GraphProto();
- if ('node' in obj) {
- message.node = obj.node.map((obj) => onnx.NodeProto.decodeJson(obj));
- }
- if ('name' in obj) {
- message.name = obj.name;
- }
- if ('initializer' in obj) {
- message.initializer = obj.initializer.map((obj) => onnx.TensorProto.decodeJson(obj));
- }
- if ('sparseInitializer' in obj) {
- message.sparse_initializer = obj.sparseInitializer.map((obj) => onnx.SparseTensorProto.decodeJson(obj));
- }
- if ('docString' in obj) {
- message.doc_string = obj.docString;
- }
- if ('input' in obj) {
- message.input = obj.input.map((obj) => onnx.ValueInfoProto.decodeJson(obj));
- }
- if ('output' in obj) {
- message.output = obj.output.map((obj) => onnx.ValueInfoProto.decodeJson(obj));
- }
- if ('valueInfo' in obj) {
- message.value_info = obj.valueInfo.map((obj) => onnx.ValueInfoProto.decodeJson(obj));
- }
- if ('quantizationAnnotation' in obj) {
- message.quantization_annotation = obj.quantizationAnnotation.map((obj) => onnx.TensorAnnotation.decodeJson(obj));
- }
- if ('metadataProps' in obj) {
- message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.GraphProto.prototype.name = "";
- onnx.GraphProto.prototype.doc_string = "";
- onnx.TensorProto = class TensorProto {
- constructor() {
- this.dims = [];
- this.float_data = [];
- this.int32_data = [];
- this.string_data = [];
- this.int64_data = [];
- this.external_data = [];
- this.double_data = [];
- this.uint64_data = [];
- this.metadata_props = [];
- }
- static decode(reader, length) {
- const message = new onnx.TensorProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.dims = reader.array(message.dims, () => reader.int64(), tag);
- break;
- case 2:
- message.data_type = reader.int32();
- break;
- case 3:
- message.segment = onnx.TensorProto.Segment.decode(reader, reader.uint32());
- break;
- case 4:
- message.float_data = reader.floats(message.float_data, tag);
- break;
- case 5:
- message.int32_data = reader.array(message.int32_data, () => reader.int32(), tag);
- break;
- case 6:
- message.string_data.push(reader.bytes());
- break;
- case 7:
- message.int64_data = reader.array(message.int64_data, () => reader.int64(), tag);
- break;
- case 8:
- message.name = reader.string();
- break;
- case 12:
- message.doc_string = reader.string();
- break;
- case 9:
- message.raw_data = reader.bytes();
- break;
- case 13:
- message.external_data.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
- break;
- case 14:
- message.data_location = reader.int32();
- break;
- case 10:
- message.double_data = reader.doubles(message.double_data, tag);
- break;
- case 11:
- message.uint64_data = reader.array(message.uint64_data, () => reader.uint64(), tag);
- break;
- case 16:
- message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TensorProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "dims":
- reader.array(message.dims, () => reader.int64());
- break;
- case "data_type":
- message.data_type = reader.int32();
- break;
- case "segment":
- message.segment = onnx.TensorProto.Segment.decodeText(reader);
- break;
- case "float_data":
- reader.array(message.float_data, () => reader.float());
- break;
- case "int32_data":
- reader.array(message.int32_data, () => reader.int32());
- break;
- case "string_data":
- reader.array(message.string_data, () => reader.bytes());
- break;
- case "int64_data":
- reader.array(message.int64_data, () => reader.int64());
- break;
- case "name":
- message.name = reader.string();
- break;
- case "doc_string":
- message.doc_string = reader.string();
- break;
- case "raw_data":
- message.raw_data = reader.bytes();
- break;
- case "external_data":
- message.external_data.push(onnx.StringStringEntryProto.decodeText(reader));
- break;
- case "data_location":
- message.data_location = reader.enum(onnx.TensorProto.DataLocation);
- break;
- case "double_data":
- reader.array(message.double_data, () => reader.double());
- break;
- case "uint64_data":
- reader.array(message.uint64_data, () => reader.uint64());
- break;
- case "metadata_props":
- message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TensorProto();
- if ('dims' in obj) {
- message.dims = obj.dims.map((obj) => BigInt(obj));
- }
- if ('dataType' in obj) {
- message.data_type = Number(obj.dataType);
- }
- if ('segment' in obj) {
- message.segment = onnx.TensorProto.Segment.decodeJson(obj.segment);
- }
- if ('floatData' in obj) {
- message.float_data = obj.floatData.map((obj) => Number(obj));
- }
- if ('int32Data' in obj) {
- message.int32_data = obj.int32Data.map((obj) => Number(obj));
- }
- if ('stringData' in obj) {
- message.string_data = obj.stringData.map((obj) => typeof obj === 'string' ? Uint8Array.from(atob(obj), (c) => c.charCodeAt(0)) : Uint8Array.from(obj));
- }
- if ('int64Data' in obj) {
- message.int64_data = obj.int64Data.map((obj) => BigInt(obj));
- }
- if ('name' in obj) {
- message.name = obj.name;
- }
- if ('docString' in obj) {
- message.doc_string = obj.docString;
- }
- if ('rawData' in obj) {
- message.raw_data = typeof obj.rawData === 'string' ? Uint8Array.from(atob(obj.rawData), (c) => c.charCodeAt(0)) : Uint8Array.from(obj.rawData);
- }
- if ('externalData' in obj) {
- message.external_data = obj.externalData.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
- }
- if ('dataLocation' in obj) {
- message.data_location = typeof obj.dataLocation === 'string' ? onnx.TensorProto.DataLocation[obj.dataLocation] : obj.dataLocation;
- }
- if ('doubleData' in obj) {
- message.double_data = obj.doubleData.map((obj) => Number(obj));
- }
- if ('uint64Data' in obj) {
- message.uint64_data = obj.uint64Data.map((obj) => BigInt(obj));
- }
- if ('metadataProps' in obj) {
- message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.TensorProto.prototype.data_type = 0;
- onnx.TensorProto.prototype.segment = null;
- onnx.TensorProto.prototype.name = "";
- onnx.TensorProto.prototype.doc_string = "";
- onnx.TensorProto.prototype.raw_data = new Uint8Array([]);
- onnx.TensorProto.prototype.data_location = 0;
- onnx.TensorProto.DataType = {
- "UNDEFINED": 0,
- "FLOAT": 1,
- "UINT8": 2,
- "INT8": 3,
- "UINT16": 4,
- "INT16": 5,
- "INT32": 6,
- "INT64": 7,
- "STRING": 8,
- "BOOL": 9,
- "FLOAT16": 10,
- "DOUBLE": 11,
- "UINT32": 12,
- "UINT64": 13,
- "COMPLEX64": 14,
- "COMPLEX128": 15,
- "BFLOAT16": 16,
- "FLOAT8E4M3FN": 17,
- "FLOAT8E4M3FNUZ": 18,
- "FLOAT8E5M2": 19,
- "FLOAT8E5M2FNUZ": 20,
- "UINT4": 21,
- "INT4": 22,
- "FLOAT4E2M1": 23
- };
- onnx.TensorProto.Segment = class Segment {
- static decode(reader, length) {
- const message = new onnx.TensorProto.Segment();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.begin = reader.int64();
- break;
- case 2:
- message.end = reader.int64();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TensorProto.Segment();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "begin":
- message.begin = reader.int64();
- break;
- case "end":
- message.end = reader.int64();
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TensorProto.Segment();
- if ('begin' in obj) {
- message.begin = BigInt(obj.begin);
- }
- if ('end' in obj) {
- message.end = BigInt(obj.end);
- }
- return message;
- }
- };
- onnx.TensorProto.Segment.prototype.begin = 0n;
- onnx.TensorProto.Segment.prototype.end = 0n;
- onnx.TensorProto.DataLocation = {
- "DEFAULT": 0,
- "EXTERNAL": 1
- };
- onnx.SparseTensorProto = class SparseTensorProto {
- constructor() {
- this.dims = [];
- }
- static decode(reader, length) {
- const message = new onnx.SparseTensorProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.values = onnx.TensorProto.decode(reader, reader.uint32());
- break;
- case 2:
- message.indices = onnx.TensorProto.decode(reader, reader.uint32());
- break;
- case 3:
- message.dims = reader.array(message.dims, () => reader.int64(), tag);
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.SparseTensorProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "values":
- message.values = onnx.TensorProto.decodeText(reader);
- break;
- case "indices":
- message.indices = onnx.TensorProto.decodeText(reader);
- break;
- case "dims":
- reader.array(message.dims, () => reader.int64());
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.SparseTensorProto();
- if ('values' in obj) {
- message.values = onnx.TensorProto.decodeJson(obj.values);
- }
- if ('indices' in obj) {
- message.indices = onnx.TensorProto.decodeJson(obj.indices);
- }
- if ('dims' in obj) {
- message.dims = obj.dims.map((obj) => BigInt(obj));
- }
- return message;
- }
- };
- onnx.SparseTensorProto.prototype.values = null;
- onnx.SparseTensorProto.prototype.indices = null;
- onnx.TensorShapeProto = class TensorShapeProto {
- constructor() {
- this.dim = [];
- }
- static decode(reader, length) {
- const message = new onnx.TensorShapeProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.dim.push(onnx.TensorShapeProto.Dimension.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TensorShapeProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "dim":
- message.dim.push(onnx.TensorShapeProto.Dimension.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TensorShapeProto();
- if ('dim' in obj) {
- message.dim = obj.dim.map((obj) => onnx.TensorShapeProto.Dimension.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.TensorShapeProto.Dimension = class Dimension {
- get value() {
- onnx.TensorShapeProto.Dimension.valueSet = onnx.TensorShapeProto.Dimension.valueSet || new Set(["dim_value", "dim_param"]);
- return Object.keys(this).find((key) => onnx.TensorShapeProto.Dimension.valueSet.has(key) && this[key] !== null);
- }
- static decode(reader, length) {
- const message = new onnx.TensorShapeProto.Dimension();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.dim_value = reader.int64();
- break;
- case 2:
- message.dim_param = reader.string();
- break;
- case 3:
- message.denotation = reader.string();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TensorShapeProto.Dimension();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "dim_value":
- message.dim_value = reader.int64();
- break;
- case "dim_param":
- message.dim_param = reader.string();
- break;
- case "denotation":
- message.denotation = reader.string();
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TensorShapeProto.Dimension();
- if ('dimValue' in obj) {
- message.dim_value = BigInt(obj.dimValue);
- }
- if ('dimParam' in obj) {
- message.dim_param = obj.dimParam;
- }
- if ('denotation' in obj) {
- message.denotation = obj.denotation;
- }
- return message;
- }
- };
- onnx.TensorShapeProto.Dimension.prototype.denotation = "";
- onnx.TypeProto = class TypeProto {
- get value() {
- onnx.TypeProto.valueSet = onnx.TypeProto.valueSet || new Set(["tensor_type", "sequence_type", "map_type", "optional_type", "sparse_tensor_type", "opaque_type"]);
- return Object.keys(this).find((key) => onnx.TypeProto.valueSet.has(key) && this[key] !== null);
- }
- static decode(reader, length) {
- const message = new onnx.TypeProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.tensor_type = onnx.TypeProto.Tensor.decode(reader, reader.uint32());
- break;
- case 4:
- message.sequence_type = onnx.TypeProto.Sequence.decode(reader, reader.uint32());
- break;
- case 5:
- message.map_type = onnx.TypeProto.Map.decode(reader, reader.uint32());
- break;
- case 9:
- message.optional_type = onnx.TypeProto.Optional.decode(reader, reader.uint32());
- break;
- case 8:
- message.sparse_tensor_type = onnx.TypeProto.SparseTensor.decode(reader, reader.uint32());
- break;
- case 7:
- message.opaque_type = onnx.TypeProto.Opaque.decode(reader, reader.uint32());
- break;
- case 6:
- message.denotation = reader.string();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TypeProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "tensor_type":
- message.tensor_type = onnx.TypeProto.Tensor.decodeText(reader);
- break;
- case "sequence_type":
- message.sequence_type = onnx.TypeProto.Sequence.decodeText(reader);
- break;
- case "map_type":
- message.map_type = onnx.TypeProto.Map.decodeText(reader);
- break;
- case "optional_type":
- message.optional_type = onnx.TypeProto.Optional.decodeText(reader);
- break;
- case "sparse_tensor_type":
- message.sparse_tensor_type = onnx.TypeProto.SparseTensor.decodeText(reader);
- break;
- case "opaque_type":
- message.opaque_type = onnx.TypeProto.Opaque.decodeText(reader);
- break;
- case "denotation":
- message.denotation = reader.string();
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TypeProto();
- if ('tensorType' in obj) {
- message.tensor_type = onnx.TypeProto.Tensor.decodeJson(obj.tensorType);
- }
- if ('sequenceType' in obj) {
- message.sequence_type = onnx.TypeProto.Sequence.decodeJson(obj.sequenceType);
- }
- if ('mapType' in obj) {
- message.map_type = onnx.TypeProto.Map.decodeJson(obj.mapType);
- }
- if ('optionalType' in obj) {
- message.optional_type = onnx.TypeProto.Optional.decodeJson(obj.optionalType);
- }
- if ('sparseTensorType' in obj) {
- message.sparse_tensor_type = onnx.TypeProto.SparseTensor.decodeJson(obj.sparseTensorType);
- }
- if ('opaqueType' in obj) {
- message.opaque_type = onnx.TypeProto.Opaque.decodeJson(obj.opaqueType);
- }
- if ('denotation' in obj) {
- message.denotation = obj.denotation;
- }
- return message;
- }
- };
- onnx.TypeProto.prototype.denotation = "";
- onnx.TypeProto.Tensor = class Tensor {
- static decode(reader, length) {
- const message = new onnx.TypeProto.Tensor();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.elem_type = reader.int32();
- break;
- case 2:
- message.shape = onnx.TensorShapeProto.decode(reader, reader.uint32());
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TypeProto.Tensor();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "elem_type":
- message.elem_type = reader.int32();
- break;
- case "shape":
- message.shape = onnx.TensorShapeProto.decodeText(reader);
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TypeProto.Tensor();
- if ('elemType' in obj) {
- message.elem_type = Number(obj.elemType);
- }
- if ('shape' in obj) {
- message.shape = onnx.TensorShapeProto.decodeJson(obj.shape);
- }
- return message;
- }
- };
- onnx.TypeProto.Tensor.prototype.elem_type = 0;
- onnx.TypeProto.Tensor.prototype.shape = null;
- onnx.TypeProto.Sequence = class Sequence {
- static decode(reader, length) {
- const message = new onnx.TypeProto.Sequence();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.elem_type = onnx.TypeProto.decode(reader, reader.uint32());
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TypeProto.Sequence();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "elem_type":
- message.elem_type = onnx.TypeProto.decodeText(reader);
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TypeProto.Sequence();
- if ('elemType' in obj) {
- message.elem_type = onnx.TypeProto.decodeJson(obj.elemType);
- }
- return message;
- }
- };
- onnx.TypeProto.Sequence.prototype.elem_type = null;
- onnx.TypeProto.Map = class Map {
- static decode(reader, length) {
- const message = new onnx.TypeProto.Map();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.key_type = reader.int32();
- break;
- case 2:
- message.value_type = onnx.TypeProto.decode(reader, reader.uint32());
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TypeProto.Map();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "key_type":
- message.key_type = reader.int32();
- break;
- case "value_type":
- message.value_type = onnx.TypeProto.decodeText(reader);
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TypeProto.Map();
- if ('keyType' in obj) {
- message.key_type = Number(obj.keyType);
- }
- if ('valueType' in obj) {
- message.value_type = onnx.TypeProto.decodeJson(obj.valueType);
- }
- return message;
- }
- };
- onnx.TypeProto.Map.prototype.key_type = 0;
- onnx.TypeProto.Map.prototype.value_type = null;
- onnx.TypeProto.Optional = class Optional {
- static decode(reader, length) {
- const message = new onnx.TypeProto.Optional();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.elem_type = onnx.TypeProto.decode(reader, reader.uint32());
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TypeProto.Optional();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "elem_type":
- message.elem_type = onnx.TypeProto.decodeText(reader);
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TypeProto.Optional();
- if ('elemType' in obj) {
- message.elem_type = onnx.TypeProto.decodeJson(obj.elemType);
- }
- return message;
- }
- };
- onnx.TypeProto.Optional.prototype.elem_type = null;
- onnx.TypeProto.SparseTensor = class SparseTensor {
- static decode(reader, length) {
- const message = new onnx.TypeProto.SparseTensor();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.elem_type = reader.int32();
- break;
- case 2:
- message.shape = onnx.TensorShapeProto.decode(reader, reader.uint32());
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TypeProto.SparseTensor();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "elem_type":
- message.elem_type = reader.int32();
- break;
- case "shape":
- message.shape = onnx.TensorShapeProto.decodeText(reader);
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TypeProto.SparseTensor();
- if ('elemType' in obj) {
- message.elem_type = Number(obj.elemType);
- }
- if ('shape' in obj) {
- message.shape = onnx.TensorShapeProto.decodeJson(obj.shape);
- }
- return message;
- }
- };
- onnx.TypeProto.SparseTensor.prototype.elem_type = 0;
- onnx.TypeProto.SparseTensor.prototype.shape = null;
- onnx.TypeProto.Opaque = class Opaque {
- static decode(reader, length) {
- const message = new onnx.TypeProto.Opaque();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.domain = reader.string();
- break;
- case 2:
- message.name = reader.string();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.TypeProto.Opaque();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "domain":
- message.domain = reader.string();
- break;
- case "name":
- message.name = reader.string();
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.TypeProto.Opaque();
- if ('domain' in obj) {
- message.domain = obj.domain;
- }
- if ('name' in obj) {
- message.name = obj.name;
- }
- return message;
- }
- };
- onnx.TypeProto.Opaque.prototype.domain = "";
- onnx.TypeProto.Opaque.prototype.name = "";
- onnx.OperatorSetIdProto = class OperatorSetIdProto {
- static decode(reader, length) {
- const message = new onnx.OperatorSetIdProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.domain = reader.string();
- break;
- case 2:
- message.version = reader.int64();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.OperatorSetIdProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "domain":
- message.domain = reader.string();
- break;
- case "version":
- message.version = reader.int64();
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.OperatorSetIdProto();
- if ('domain' in obj) {
- message.domain = obj.domain;
- }
- if ('version' in obj) {
- message.version = BigInt(obj.version);
- }
- return message;
- }
- };
- onnx.OperatorSetIdProto.prototype.domain = "";
- onnx.OperatorSetIdProto.prototype.version = 0n;
- onnx.OperatorStatus = {
- "EXPERIMENTAL": 0,
- "STABLE": 1
- };
- onnx.FunctionProto = class FunctionProto {
- constructor() {
- this.input = [];
- this.output = [];
- this.attribute = [];
- this.attribute_proto = [];
- this.node = [];
- this.opset_import = [];
- this.value_info = [];
- this.metadata_props = [];
- }
- static decode(reader, length) {
- const message = new onnx.FunctionProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.name = reader.string();
- break;
- case 4:
- message.input.push(reader.string());
- break;
- case 5:
- message.output.push(reader.string());
- break;
- case 6:
- message.attribute.push(reader.string());
- break;
- case 11:
- message.attribute_proto.push(onnx.AttributeProto.decode(reader, reader.uint32()));
- break;
- case 7:
- message.node.push(onnx.NodeProto.decode(reader, reader.uint32()));
- break;
- case 8:
- message.doc_string = reader.string();
- break;
- case 9:
- message.opset_import.push(onnx.OperatorSetIdProto.decode(reader, reader.uint32()));
- break;
- case 10:
- message.domain = reader.string();
- break;
- case 13:
- message.overload = reader.string();
- break;
- case 12:
- message.value_info.push(onnx.ValueInfoProto.decode(reader, reader.uint32()));
- break;
- case 14:
- message.metadata_props.push(onnx.StringStringEntryProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.FunctionProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "name":
- message.name = reader.string();
- break;
- case "input":
- reader.array(message.input, () => reader.string());
- break;
- case "output":
- reader.array(message.output, () => reader.string());
- break;
- case "attribute":
- reader.array(message.attribute, () => reader.string());
- break;
- case "attribute_proto":
- message.attribute_proto.push(onnx.AttributeProto.decodeText(reader));
- break;
- case "node":
- message.node.push(onnx.NodeProto.decodeText(reader));
- break;
- case "doc_string":
- message.doc_string = reader.string();
- break;
- case "opset_import":
- message.opset_import.push(onnx.OperatorSetIdProto.decodeText(reader));
- break;
- case "domain":
- message.domain = reader.string();
- break;
- case "overload":
- message.overload = reader.string();
- break;
- case "value_info":
- message.value_info.push(onnx.ValueInfoProto.decodeText(reader));
- break;
- case "metadata_props":
- message.metadata_props.push(onnx.StringStringEntryProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.FunctionProto();
- if ('name' in obj) {
- message.name = obj.name;
- }
- if ('input' in obj) {
- message.input = obj.input;
- }
- if ('output' in obj) {
- message.output = obj.output;
- }
- if ('attribute' in obj) {
- message.attribute = obj.attribute;
- }
- if ('attributeProto' in obj) {
- message.attribute_proto = obj.attributeProto.map((obj) => onnx.AttributeProto.decodeJson(obj));
- }
- if ('node' in obj) {
- message.node = obj.node.map((obj) => onnx.NodeProto.decodeJson(obj));
- }
- if ('docString' in obj) {
- message.doc_string = obj.docString;
- }
- if ('opsetImport' in obj) {
- message.opset_import = obj.opsetImport.map((obj) => onnx.OperatorSetIdProto.decodeJson(obj));
- }
- if ('domain' in obj) {
- message.domain = obj.domain;
- }
- if ('overload' in obj) {
- message.overload = obj.overload;
- }
- if ('valueInfo' in obj) {
- message.value_info = obj.valueInfo.map((obj) => onnx.ValueInfoProto.decodeJson(obj));
- }
- if ('metadataProps' in obj) {
- message.metadata_props = obj.metadataProps.map((obj) => onnx.StringStringEntryProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.FunctionProto.prototype.name = "";
- onnx.FunctionProto.prototype.doc_string = "";
- onnx.FunctionProto.prototype.domain = "";
- onnx.FunctionProto.prototype.overload = "";
- onnx.OperatorProto = class OperatorProto {
- static decode(reader, length) {
- const message = new onnx.OperatorProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.op_type = reader.string();
- break;
- case 2:
- message.since_version = reader.int64();
- break;
- case 3:
- message.status = reader.int32();
- break;
- case 10:
- message.doc_string = reader.string();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.OperatorProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "op_type":
- message.op_type = reader.string();
- break;
- case "since_version":
- message.since_version = reader.int64();
- break;
- case "status":
- message.status = reader.enum(onnx.OperatorStatus);
- break;
- case "doc_string":
- message.doc_string = reader.string();
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.OperatorProto();
- if ('opType' in obj) {
- message.op_type = obj.opType;
- }
- if ('sinceVersion' in obj) {
- message.since_version = BigInt(obj.sinceVersion);
- }
- if ('status' in obj) {
- message.status = typeof obj.status === 'string' ? onnx.OperatorStatus[obj.status] : obj.status;
- }
- if ('docString' in obj) {
- message.doc_string = obj.docString;
- }
- return message;
- }
- };
- onnx.OperatorProto.prototype.op_type = "";
- onnx.OperatorProto.prototype.since_version = 0n;
- onnx.OperatorProto.prototype.status = 0;
- onnx.OperatorProto.prototype.doc_string = "";
- onnx.OperatorSetProto = class OperatorSetProto {
- constructor() {
- this.operator = [];
- this.functions = [];
- }
- static decode(reader, length) {
- const message = new onnx.OperatorSetProto();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.magic = reader.string();
- break;
- case 2:
- message.ir_version = reader.int64();
- break;
- case 3:
- message.ir_version_prerelease = reader.string();
- break;
- case 7:
- message.ir_build_metadata = reader.string();
- break;
- case 4:
- message.domain = reader.string();
- break;
- case 5:
- message.opset_version = reader.int64();
- break;
- case 6:
- message.doc_string = reader.string();
- break;
- case 8:
- message.operator.push(onnx.OperatorProto.decode(reader, reader.uint32()));
- break;
- case 9:
- message.functions.push(onnx.FunctionProto.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new onnx.OperatorSetProto();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "magic":
- message.magic = reader.string();
- break;
- case "ir_version":
- message.ir_version = reader.int64();
- break;
- case "ir_version_prerelease":
- message.ir_version_prerelease = reader.string();
- break;
- case "ir_build_metadata":
- message.ir_build_metadata = reader.string();
- break;
- case "domain":
- message.domain = reader.string();
- break;
- case "opset_version":
- message.opset_version = reader.int64();
- break;
- case "doc_string":
- message.doc_string = reader.string();
- break;
- case "operator":
- message.operator.push(onnx.OperatorProto.decodeText(reader));
- break;
- case "functions":
- message.functions.push(onnx.FunctionProto.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- static decodeJson(obj) {
- const message = new onnx.OperatorSetProto();
- if ('magic' in obj) {
- message.magic = obj.magic;
- }
- if ('irVersion' in obj) {
- message.ir_version = BigInt(obj.irVersion);
- }
- if ('irVersionPrerelease' in obj) {
- message.ir_version_prerelease = obj.irVersionPrerelease;
- }
- if ('irBuildMetadata' in obj) {
- message.ir_build_metadata = obj.irBuildMetadata;
- }
- if ('domain' in obj) {
- message.domain = obj.domain;
- }
- if ('opsetVersion' in obj) {
- message.opset_version = BigInt(obj.opsetVersion);
- }
- if ('docString' in obj) {
- message.doc_string = obj.docString;
- }
- if ('operator' in obj) {
- message.operator = obj.operator.map((obj) => onnx.OperatorProto.decodeJson(obj));
- }
- if ('functions' in obj) {
- message.functions = obj.functions.map((obj) => onnx.FunctionProto.decodeJson(obj));
- }
- return message;
- }
- };
- onnx.OperatorSetProto.prototype.magic = "";
- onnx.OperatorSetProto.prototype.ir_version = 0n;
- onnx.OperatorSetProto.prototype.ir_version_prerelease = "";
- onnx.OperatorSetProto.prototype.ir_build_metadata = "";
- onnx.OperatorSetProto.prototype.domain = "";
- onnx.OperatorSetProto.prototype.opset_version = 0n;
- onnx.OperatorSetProto.prototype.doc_string = "";
|