| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055 |
- export const ge = {};
- ge.proto = {};
- ge.proto.DataType = {
- "DT_UNDEFINED": 0,
- "DT_FLOAT": 1,
- "DT_FLOAT16": 2,
- "DT_INT8": 3,
- "DT_UINT8": 4,
- "DT_INT16": 5,
- "DT_UINT16": 6,
- "DT_INT32": 7,
- "DT_INT64": 8,
- "DT_UINT32": 9,
- "DT_UINT64": 10,
- "DT_BOOL": 11,
- "DT_DOUBLE": 12,
- "DT_STRING": 13,
- "DT_DUAL_SUB_INT8": 14,
- "DT_DUAL_SUB_UINT8": 15,
- "DT_COMPLEX64": 16,
- "DT_COMPLEX128": 17,
- "DT_QINT8": 18,
- "DT_QINT16": 19,
- "DT_QINT32": 20,
- "DT_QUINT8": 21,
- "DT_QUINT16": 22,
- "DT_RESOURCE": 23,
- "DT_STRING_REF": 24,
- "DT_DUAL": 25,
- "DT_VARIANT": 26,
- "DT_BF16": 27,
- "DT_INT4": 28,
- "DT_UINT1": 29,
- "DT_INT2": 30,
- "DT_UINT2": 31
- };
- ge.proto.AttrDef = class AttrDef {
- get value() {
- ge.proto.AttrDef.valueSet = ge.proto.AttrDef.valueSet || new Set(["s", "i", "f", "b", "bt", "list", "func", "td", "t", "g", "list_list_int", "dt", "list_list_float"]);
- return Object.keys(this).find((key) => ge.proto.AttrDef.valueSet.has(key) && this[key] !== null);
- }
- static decode(reader, length) {
- const message = new ge.proto.AttrDef();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 2:
- message.s = reader.bytes();
- break;
- case 3:
- message.i = reader.int64();
- break;
- case 4:
- message.f = reader.float();
- break;
- case 5:
- message.b = reader.bool();
- break;
- case 7:
- message.bt = reader.bytes();
- break;
- case 1:
- message.list = ge.proto.AttrDef.ListValue.decode(reader, reader.uint32());
- break;
- case 10:
- message.func = ge.proto.NamedAttrs.decode(reader, reader.uint32());
- break;
- case 11:
- message.td = ge.proto.TensorDescriptor.decode(reader, reader.uint32());
- break;
- case 12:
- message.t = ge.proto.TensorDef.decode(reader, reader.uint32());
- break;
- case 13:
- message.g = ge.proto.GraphDef.decode(reader, reader.uint32());
- break;
- case 14:
- message.list_list_int = ge.proto.AttrDef.ListListInt.decode(reader, reader.uint32());
- break;
- case 15:
- message.dt = reader.int64();
- break;
- case 16:
- message.list_list_float = ge.proto.AttrDef.ListListFloat.decode(reader, reader.uint32());
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.AttrDef();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "s":
- message.s = reader.bytes();
- break;
- case "i":
- message.i = reader.int64();
- break;
- case "f":
- message.f = reader.float();
- break;
- case "b":
- message.b = reader.bool();
- break;
- case "bt":
- message.bt = reader.bytes();
- break;
- case "list":
- message.list = ge.proto.AttrDef.ListValue.decodeText(reader);
- break;
- case "func":
- message.func = ge.proto.NamedAttrs.decodeText(reader);
- break;
- case "td":
- message.td = ge.proto.TensorDescriptor.decodeText(reader);
- break;
- case "t":
- message.t = ge.proto.TensorDef.decodeText(reader);
- break;
- case "g":
- message.g = ge.proto.GraphDef.decodeText(reader);
- break;
- case "list_list_int":
- message.list_list_int = ge.proto.AttrDef.ListListInt.decodeText(reader);
- break;
- case "dt":
- message.dt = reader.int64();
- break;
- case "list_list_float":
- message.list_list_float = ge.proto.AttrDef.ListListFloat.decodeText(reader);
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.AttrDef.ListValue = class ListValue {
- constructor() {
- this.s = [];
- this.i = [];
- this.f = [];
- this.b = [];
- this.bt = [];
- this.td = [];
- this.t = [];
- this.g = [];
- this.na = [];
- this.dt = [];
- }
- static decode(reader, length) {
- const message = new ge.proto.AttrDef.ListValue();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 2:
- message.s.push(reader.bytes());
- break;
- case 3:
- message.i = reader.array(message.i, () => reader.int64(), tag);
- break;
- case 4:
- message.f = reader.floats(message.f, tag);
- break;
- case 5:
- message.b = reader.array(message.b, () => reader.bool(), tag);
- break;
- case 7:
- message.bt.push(reader.bytes());
- break;
- case 8:
- message.td.push(ge.proto.TensorDescriptor.decode(reader, reader.uint32()));
- break;
- case 9:
- message.t.push(ge.proto.TensorDef.decode(reader, reader.uint32()));
- break;
- case 10:
- message.g.push(ge.proto.GraphDef.decode(reader, reader.uint32()));
- break;
- case 11:
- message.na.push(ge.proto.NamedAttrs.decode(reader, reader.uint32()));
- break;
- case 12:
- message.dt = reader.array(message.dt, () => reader.int64(), tag);
- break;
- case 20:
- message.val_type = reader.int32();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.AttrDef.ListValue();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "s":
- reader.array(message.s, () => reader.bytes());
- break;
- case "i":
- reader.array(message.i, () => reader.int64());
- break;
- case "f":
- reader.array(message.f, () => reader.float());
- break;
- case "b":
- reader.array(message.b, () => reader.bool());
- break;
- case "bt":
- reader.array(message.bt, () => reader.bytes());
- break;
- case "td":
- message.td.push(ge.proto.TensorDescriptor.decodeText(reader));
- break;
- case "t":
- message.t.push(ge.proto.TensorDef.decodeText(reader));
- break;
- case "g":
- message.g.push(ge.proto.GraphDef.decodeText(reader));
- break;
- case "na":
- message.na.push(ge.proto.NamedAttrs.decodeText(reader));
- break;
- case "dt":
- reader.array(message.dt, () => reader.int64());
- break;
- case "val_type":
- message.val_type = reader.enum(ge.proto.AttrDef.ListValue.ListValueType);
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.AttrDef.ListValue.prototype.val_type = 0;
- ge.proto.AttrDef.ListValue.ListValueType = {
- "VT_LIST_NONE": 0,
- "VT_LIST_STRING": 1,
- "VT_LIST_INT": 2,
- "VT_LIST_FLOAT": 3,
- "VT_LIST_BOOL": 4,
- "VT_LIST_BYTES": 5,
- "VT_LIST_TENSOR_DESC": 6,
- "VT_LIST_TENSOR": 7,
- "VT_LIST_GRAPH": 8,
- "VT_LIST_NAMED_ATTRS": 9,
- "VT_LIST_DATA_TYPE": 10
- };
- ge.proto.AttrDef.ListListInt = class ListListInt {
- constructor() {
- this.list_list_i = [];
- }
- static decode(reader, length) {
- const message = new ge.proto.AttrDef.ListListInt();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.list_list_i.push(ge.proto.AttrDef.ListListInt.ListInt.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.AttrDef.ListListInt();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "list_list_i":
- message.list_list_i.push(ge.proto.AttrDef.ListListInt.ListInt.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.AttrDef.ListListInt.ListInt = class ListInt {
- constructor() {
- this.list_i = [];
- }
- static decode(reader, length) {
- const message = new ge.proto.AttrDef.ListListInt.ListInt();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.list_i = reader.array(message.list_i, () => reader.int64(), tag);
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.AttrDef.ListListInt.ListInt();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "list_i":
- reader.array(message.list_i, () => reader.int64());
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.AttrDef.ListListFloat = class ListListFloat {
- constructor() {
- this.list_list_f = [];
- }
- static decode(reader, length) {
- const message = new ge.proto.AttrDef.ListListFloat();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.list_list_f.push(ge.proto.AttrDef.ListListFloat.ListFloat.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.AttrDef.ListListFloat();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "list_list_f":
- message.list_list_f.push(ge.proto.AttrDef.ListListFloat.ListFloat.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.AttrDef.ListListFloat.ListFloat = class ListFloat {
- constructor() {
- this.list_f = [];
- }
- static decode(reader, length) {
- const message = new ge.proto.AttrDef.ListListFloat.ListFloat();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.list_f = reader.floats(message.list_f, tag);
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.AttrDef.ListListFloat.ListFloat();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "list_f":
- reader.array(message.list_f, () => reader.float());
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.NamedAttrs = class NamedAttrs {
- constructor() {
- this.attr = {};
- }
- static decode(reader, length) {
- const message = new ge.proto.NamedAttrs();
- 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:
- reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.NamedAttrs();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "name":
- message.name = reader.string();
- break;
- case "attr":
- reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.NamedAttrs.prototype.name = "";
- ge.proto.ShapeDef = class ShapeDef {
- constructor() {
- this.dim = [];
- }
- static decode(reader, length) {
- const message = new ge.proto.ShapeDef();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.dim = reader.array(message.dim, () => reader.int64(), tag);
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.ShapeDef();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "dim":
- reader.array(message.dim, () => reader.int64());
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.TensorDescriptor = class TensorDescriptor {
- constructor() {
- this.attr = {};
- }
- static decode(reader, length) {
- const message = new ge.proto.TensorDescriptor();
- 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.dtype = reader.int32();
- break;
- case 3:
- message.shape = ge.proto.ShapeDef.decode(reader, reader.uint32());
- break;
- case 4:
- message.layout = reader.string();
- break;
- case 9:
- message.has_out_attr = reader.bool();
- break;
- case 10:
- message.size = reader.int64();
- break;
- case 11:
- message.weight_size = reader.int64();
- break;
- case 12:
- message.reuse_input = reader.bool();
- break;
- case 13:
- message.output_tensor = reader.bool();
- break;
- case 14:
- message.device_type = reader.string();
- break;
- case 15:
- message.input_tensor = reader.bool();
- break;
- case 16:
- message.real_dim_cnt = reader.int64();
- break;
- case 17:
- message.reuse_input_index = reader.int64();
- break;
- case 18:
- message.data_offset = reader.int64();
- break;
- case 19:
- message.cmps_size = reader.int64();
- break;
- case 20:
- message.cmps_tab = reader.string();
- break;
- case 21:
- message.cmps_tab_offset = reader.int64();
- break;
- case 5:
- reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.TensorDescriptor();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "name":
- message.name = reader.string();
- break;
- case "dtype":
- message.dtype = reader.enum(ge.proto.DataType);
- break;
- case "shape":
- message.shape = ge.proto.ShapeDef.decodeText(reader);
- break;
- case "layout":
- message.layout = reader.string();
- break;
- case "has_out_attr":
- message.has_out_attr = reader.bool();
- break;
- case "size":
- message.size = reader.int64();
- break;
- case "weight_size":
- message.weight_size = reader.int64();
- break;
- case "reuse_input":
- message.reuse_input = reader.bool();
- break;
- case "output_tensor":
- message.output_tensor = reader.bool();
- break;
- case "device_type":
- message.device_type = reader.string();
- break;
- case "input_tensor":
- message.input_tensor = reader.bool();
- break;
- case "real_dim_cnt":
- message.real_dim_cnt = reader.int64();
- break;
- case "reuse_input_index":
- message.reuse_input_index = reader.int64();
- break;
- case "data_offset":
- message.data_offset = reader.int64();
- break;
- case "cmps_size":
- message.cmps_size = reader.int64();
- break;
- case "cmps_tab":
- message.cmps_tab = reader.string();
- break;
- case "cmps_tab_offset":
- message.cmps_tab_offset = reader.int64();
- break;
- case "attr":
- reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.TensorDescriptor.prototype.name = "";
- ge.proto.TensorDescriptor.prototype.dtype = 0;
- ge.proto.TensorDescriptor.prototype.shape = null;
- ge.proto.TensorDescriptor.prototype.layout = "";
- ge.proto.TensorDescriptor.prototype.has_out_attr = false;
- ge.proto.TensorDescriptor.prototype.size = 0n;
- ge.proto.TensorDescriptor.prototype.weight_size = 0n;
- ge.proto.TensorDescriptor.prototype.reuse_input = false;
- ge.proto.TensorDescriptor.prototype.output_tensor = false;
- ge.proto.TensorDescriptor.prototype.device_type = "";
- ge.proto.TensorDescriptor.prototype.input_tensor = false;
- ge.proto.TensorDescriptor.prototype.real_dim_cnt = 0n;
- ge.proto.TensorDescriptor.prototype.reuse_input_index = 0n;
- ge.proto.TensorDescriptor.prototype.data_offset = 0n;
- ge.proto.TensorDescriptor.prototype.cmps_size = 0n;
- ge.proto.TensorDescriptor.prototype.cmps_tab = "";
- ge.proto.TensorDescriptor.prototype.cmps_tab_offset = 0n;
- ge.proto.TensorDef = class TensorDef {
- static decode(reader, length) {
- const message = new ge.proto.TensorDef();
- const end = length === undefined ? reader.length : reader.position + length;
- while (reader.position < end) {
- const tag = reader.uint32();
- switch (tag >>> 3) {
- case 1:
- message.desc = ge.proto.TensorDescriptor.decode(reader, reader.uint32());
- break;
- case 2:
- message.data = reader.bytes();
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.TensorDef();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "desc":
- message.desc = ge.proto.TensorDescriptor.decodeText(reader);
- break;
- case "data":
- message.data = reader.bytes();
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.TensorDef.prototype.desc = null;
- ge.proto.TensorDef.prototype.data = new Uint8Array([]);
- ge.proto.OpDef = class OpDef {
- constructor() {
- this.input = [];
- this.attr = {};
- this.input_name = [];
- this.src_name = [];
- this.src_index = [];
- this.dst_name = [];
- this.dst_index = [];
- this.input_i = [];
- this.output_i = [];
- this.workspace = [];
- this.workspace_bytes = [];
- this.is_input_const = [];
- this.input_desc = [];
- this.output_desc = [];
- this.subgraph_name = [];
- }
- static decode(reader, length) {
- const message = new ge.proto.OpDef();
- 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 = reader.string();
- break;
- case 5:
- message.input.push(reader.string());
- break;
- case 10:
- reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decode(reader, reader.uint32()));
- break;
- case 20:
- message.has_out_attr = reader.bool();
- break;
- case 21:
- message.id = reader.int64();
- break;
- case 22:
- message.stream_id = reader.int64();
- break;
- case 23:
- message.input_name.push(reader.string());
- break;
- case 24:
- message.src_name.push(reader.string());
- break;
- case 25:
- message.src_index = reader.array(message.src_index, () => reader.int64(), tag);
- break;
- case 26:
- message.dst_name.push(reader.string());
- break;
- case 27:
- message.dst_index = reader.array(message.dst_index, () => reader.int64(), tag);
- break;
- case 28:
- message.input_i = reader.array(message.input_i, () => reader.int64(), tag);
- break;
- case 29:
- message.output_i = reader.array(message.output_i, () => reader.int64(), tag);
- break;
- case 30:
- message.workspace = reader.array(message.workspace, () => reader.int64(), tag);
- break;
- case 31:
- message.workspace_bytes = reader.array(message.workspace_bytes, () => reader.int64(), tag);
- break;
- case 32:
- message.is_input_const = reader.array(message.is_input_const, () => reader.bool(), tag);
- break;
- case 33:
- message.input_desc.push(ge.proto.TensorDescriptor.decode(reader, reader.uint32()));
- break;
- case 34:
- message.output_desc.push(ge.proto.TensorDescriptor.decode(reader, reader.uint32()));
- break;
- case 35:
- message.subgraph_name.push(reader.string());
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.OpDef();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "name":
- message.name = reader.string();
- break;
- case "type":
- message.type = reader.string();
- break;
- case "input":
- reader.array(message.input, () => reader.string());
- break;
- case "attr":
- reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decodeText(reader));
- break;
- case "has_out_attr":
- message.has_out_attr = reader.bool();
- break;
- case "id":
- message.id = reader.int64();
- break;
- case "stream_id":
- message.stream_id = reader.int64();
- break;
- case "input_name":
- reader.array(message.input_name, () => reader.string());
- break;
- case "src_name":
- reader.array(message.src_name, () => reader.string());
- break;
- case "src_index":
- reader.array(message.src_index, () => reader.int64());
- break;
- case "dst_name":
- reader.array(message.dst_name, () => reader.string());
- break;
- case "dst_index":
- reader.array(message.dst_index, () => reader.int64());
- break;
- case "input_i":
- reader.array(message.input_i, () => reader.int64());
- break;
- case "output_i":
- reader.array(message.output_i, () => reader.int64());
- break;
- case "workspace":
- reader.array(message.workspace, () => reader.int64());
- break;
- case "workspace_bytes":
- reader.array(message.workspace_bytes, () => reader.int64());
- break;
- case "is_input_const":
- reader.array(message.is_input_const, () => reader.bool());
- break;
- case "input_desc":
- message.input_desc.push(ge.proto.TensorDescriptor.decodeText(reader));
- break;
- case "output_desc":
- message.output_desc.push(ge.proto.TensorDescriptor.decodeText(reader));
- break;
- case "subgraph_name":
- reader.array(message.subgraph_name, () => reader.string());
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.OpDef.prototype.name = "";
- ge.proto.OpDef.prototype.type = "";
- ge.proto.OpDef.prototype.has_out_attr = false;
- ge.proto.OpDef.prototype.id = 0n;
- ge.proto.OpDef.prototype.stream_id = 0n;
- ge.proto.GraphDef = class GraphDef {
- constructor() {
- this.input = [];
- this.output = [];
- this.op = [];
- this.attr = {};
- }
- static decode(reader, length) {
- const message = new ge.proto.GraphDef();
- 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.op.push(ge.proto.OpDef.decode(reader, reader.uint32()));
- break;
- case 11:
- reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.GraphDef();
- 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 "op":
- message.op.push(ge.proto.OpDef.decodeText(reader));
- break;
- case "attr":
- reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.GraphDef.prototype.name = "";
- ge.proto.ModelDef = class ModelDef {
- constructor() {
- this.graph = [];
- this.attr = {};
- }
- static decode(reader, length) {
- const message = new ge.proto.ModelDef();
- 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.version = reader.uint32();
- break;
- case 3:
- message.custom_version = reader.string();
- break;
- case 7:
- message.graph.push(ge.proto.GraphDef.decode(reader, reader.uint32()));
- break;
- case 11:
- reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decode(reader, reader.uint32()));
- break;
- default:
- reader.skipType(tag & 7);
- break;
- }
- }
- return message;
- }
- static decodeText(reader) {
- const message = new ge.proto.ModelDef();
- reader.start();
- while (!reader.end()) {
- const tag = reader.tag();
- switch (tag) {
- case "name":
- message.name = reader.string();
- break;
- case "version":
- message.version = reader.uint32();
- break;
- case "custom_version":
- message.custom_version = reader.string();
- break;
- case "graph":
- message.graph.push(ge.proto.GraphDef.decodeText(reader));
- break;
- case "attr":
- reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decodeText(reader));
- break;
- default:
- reader.field(tag, message);
- break;
- }
- }
- return message;
- }
- };
- ge.proto.ModelDef.prototype.name = "";
- ge.proto.ModelDef.prototype.version = 0;
- ge.proto.ModelDef.prototype.custom_version = "";
|