om-proto.js 35 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055
  1. export const ge = {};
  2. ge.proto = {};
  3. ge.proto.DataType = {
  4. "DT_UNDEFINED": 0,
  5. "DT_FLOAT": 1,
  6. "DT_FLOAT16": 2,
  7. "DT_INT8": 3,
  8. "DT_UINT8": 4,
  9. "DT_INT16": 5,
  10. "DT_UINT16": 6,
  11. "DT_INT32": 7,
  12. "DT_INT64": 8,
  13. "DT_UINT32": 9,
  14. "DT_UINT64": 10,
  15. "DT_BOOL": 11,
  16. "DT_DOUBLE": 12,
  17. "DT_STRING": 13,
  18. "DT_DUAL_SUB_INT8": 14,
  19. "DT_DUAL_SUB_UINT8": 15,
  20. "DT_COMPLEX64": 16,
  21. "DT_COMPLEX128": 17,
  22. "DT_QINT8": 18,
  23. "DT_QINT16": 19,
  24. "DT_QINT32": 20,
  25. "DT_QUINT8": 21,
  26. "DT_QUINT16": 22,
  27. "DT_RESOURCE": 23,
  28. "DT_STRING_REF": 24,
  29. "DT_DUAL": 25,
  30. "DT_VARIANT": 26,
  31. "DT_BF16": 27,
  32. "DT_INT4": 28,
  33. "DT_UINT1": 29,
  34. "DT_INT2": 30,
  35. "DT_UINT2": 31
  36. };
  37. ge.proto.AttrDef = class AttrDef {
  38. get value() {
  39. 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"]);
  40. return Object.keys(this).find((key) => ge.proto.AttrDef.valueSet.has(key) && this[key] !== null);
  41. }
  42. static decode(reader, length) {
  43. const message = new ge.proto.AttrDef();
  44. const end = length === undefined ? reader.length : reader.position + length;
  45. while (reader.position < end) {
  46. const tag = reader.uint32();
  47. switch (tag >>> 3) {
  48. case 2:
  49. message.s = reader.bytes();
  50. break;
  51. case 3:
  52. message.i = reader.int64();
  53. break;
  54. case 4:
  55. message.f = reader.float();
  56. break;
  57. case 5:
  58. message.b = reader.bool();
  59. break;
  60. case 7:
  61. message.bt = reader.bytes();
  62. break;
  63. case 1:
  64. message.list = ge.proto.AttrDef.ListValue.decode(reader, reader.uint32());
  65. break;
  66. case 10:
  67. message.func = ge.proto.NamedAttrs.decode(reader, reader.uint32());
  68. break;
  69. case 11:
  70. message.td = ge.proto.TensorDescriptor.decode(reader, reader.uint32());
  71. break;
  72. case 12:
  73. message.t = ge.proto.TensorDef.decode(reader, reader.uint32());
  74. break;
  75. case 13:
  76. message.g = ge.proto.GraphDef.decode(reader, reader.uint32());
  77. break;
  78. case 14:
  79. message.list_list_int = ge.proto.AttrDef.ListListInt.decode(reader, reader.uint32());
  80. break;
  81. case 15:
  82. message.dt = reader.int64();
  83. break;
  84. case 16:
  85. message.list_list_float = ge.proto.AttrDef.ListListFloat.decode(reader, reader.uint32());
  86. break;
  87. default:
  88. reader.skipType(tag & 7);
  89. break;
  90. }
  91. }
  92. return message;
  93. }
  94. static decodeText(reader) {
  95. const message = new ge.proto.AttrDef();
  96. reader.start();
  97. while (!reader.end()) {
  98. const tag = reader.tag();
  99. switch (tag) {
  100. case "s":
  101. message.s = reader.bytes();
  102. break;
  103. case "i":
  104. message.i = reader.int64();
  105. break;
  106. case "f":
  107. message.f = reader.float();
  108. break;
  109. case "b":
  110. message.b = reader.bool();
  111. break;
  112. case "bt":
  113. message.bt = reader.bytes();
  114. break;
  115. case "list":
  116. message.list = ge.proto.AttrDef.ListValue.decodeText(reader);
  117. break;
  118. case "func":
  119. message.func = ge.proto.NamedAttrs.decodeText(reader);
  120. break;
  121. case "td":
  122. message.td = ge.proto.TensorDescriptor.decodeText(reader);
  123. break;
  124. case "t":
  125. message.t = ge.proto.TensorDef.decodeText(reader);
  126. break;
  127. case "g":
  128. message.g = ge.proto.GraphDef.decodeText(reader);
  129. break;
  130. case "list_list_int":
  131. message.list_list_int = ge.proto.AttrDef.ListListInt.decodeText(reader);
  132. break;
  133. case "dt":
  134. message.dt = reader.int64();
  135. break;
  136. case "list_list_float":
  137. message.list_list_float = ge.proto.AttrDef.ListListFloat.decodeText(reader);
  138. break;
  139. default:
  140. reader.field(tag, message);
  141. break;
  142. }
  143. }
  144. return message;
  145. }
  146. };
  147. ge.proto.AttrDef.ListValue = class ListValue {
  148. constructor() {
  149. this.s = [];
  150. this.i = [];
  151. this.f = [];
  152. this.b = [];
  153. this.bt = [];
  154. this.td = [];
  155. this.t = [];
  156. this.g = [];
  157. this.na = [];
  158. this.dt = [];
  159. }
  160. static decode(reader, length) {
  161. const message = new ge.proto.AttrDef.ListValue();
  162. const end = length === undefined ? reader.length : reader.position + length;
  163. while (reader.position < end) {
  164. const tag = reader.uint32();
  165. switch (tag >>> 3) {
  166. case 2:
  167. message.s.push(reader.bytes());
  168. break;
  169. case 3:
  170. message.i = reader.array(message.i, () => reader.int64(), tag);
  171. break;
  172. case 4:
  173. message.f = reader.floats(message.f, tag);
  174. break;
  175. case 5:
  176. message.b = reader.array(message.b, () => reader.bool(), tag);
  177. break;
  178. case 7:
  179. message.bt.push(reader.bytes());
  180. break;
  181. case 8:
  182. message.td.push(ge.proto.TensorDescriptor.decode(reader, reader.uint32()));
  183. break;
  184. case 9:
  185. message.t.push(ge.proto.TensorDef.decode(reader, reader.uint32()));
  186. break;
  187. case 10:
  188. message.g.push(ge.proto.GraphDef.decode(reader, reader.uint32()));
  189. break;
  190. case 11:
  191. message.na.push(ge.proto.NamedAttrs.decode(reader, reader.uint32()));
  192. break;
  193. case 12:
  194. message.dt = reader.array(message.dt, () => reader.int64(), tag);
  195. break;
  196. case 20:
  197. message.val_type = reader.int32();
  198. break;
  199. default:
  200. reader.skipType(tag & 7);
  201. break;
  202. }
  203. }
  204. return message;
  205. }
  206. static decodeText(reader) {
  207. const message = new ge.proto.AttrDef.ListValue();
  208. reader.start();
  209. while (!reader.end()) {
  210. const tag = reader.tag();
  211. switch (tag) {
  212. case "s":
  213. reader.array(message.s, () => reader.bytes());
  214. break;
  215. case "i":
  216. reader.array(message.i, () => reader.int64());
  217. break;
  218. case "f":
  219. reader.array(message.f, () => reader.float());
  220. break;
  221. case "b":
  222. reader.array(message.b, () => reader.bool());
  223. break;
  224. case "bt":
  225. reader.array(message.bt, () => reader.bytes());
  226. break;
  227. case "td":
  228. message.td.push(ge.proto.TensorDescriptor.decodeText(reader));
  229. break;
  230. case "t":
  231. message.t.push(ge.proto.TensorDef.decodeText(reader));
  232. break;
  233. case "g":
  234. message.g.push(ge.proto.GraphDef.decodeText(reader));
  235. break;
  236. case "na":
  237. message.na.push(ge.proto.NamedAttrs.decodeText(reader));
  238. break;
  239. case "dt":
  240. reader.array(message.dt, () => reader.int64());
  241. break;
  242. case "val_type":
  243. message.val_type = reader.enum(ge.proto.AttrDef.ListValue.ListValueType);
  244. break;
  245. default:
  246. reader.field(tag, message);
  247. break;
  248. }
  249. }
  250. return message;
  251. }
  252. };
  253. ge.proto.AttrDef.ListValue.prototype.val_type = 0;
  254. ge.proto.AttrDef.ListValue.ListValueType = {
  255. "VT_LIST_NONE": 0,
  256. "VT_LIST_STRING": 1,
  257. "VT_LIST_INT": 2,
  258. "VT_LIST_FLOAT": 3,
  259. "VT_LIST_BOOL": 4,
  260. "VT_LIST_BYTES": 5,
  261. "VT_LIST_TENSOR_DESC": 6,
  262. "VT_LIST_TENSOR": 7,
  263. "VT_LIST_GRAPH": 8,
  264. "VT_LIST_NAMED_ATTRS": 9,
  265. "VT_LIST_DATA_TYPE": 10
  266. };
  267. ge.proto.AttrDef.ListListInt = class ListListInt {
  268. constructor() {
  269. this.list_list_i = [];
  270. }
  271. static decode(reader, length) {
  272. const message = new ge.proto.AttrDef.ListListInt();
  273. const end = length === undefined ? reader.length : reader.position + length;
  274. while (reader.position < end) {
  275. const tag = reader.uint32();
  276. switch (tag >>> 3) {
  277. case 1:
  278. message.list_list_i.push(ge.proto.AttrDef.ListListInt.ListInt.decode(reader, reader.uint32()));
  279. break;
  280. default:
  281. reader.skipType(tag & 7);
  282. break;
  283. }
  284. }
  285. return message;
  286. }
  287. static decodeText(reader) {
  288. const message = new ge.proto.AttrDef.ListListInt();
  289. reader.start();
  290. while (!reader.end()) {
  291. const tag = reader.tag();
  292. switch (tag) {
  293. case "list_list_i":
  294. message.list_list_i.push(ge.proto.AttrDef.ListListInt.ListInt.decodeText(reader));
  295. break;
  296. default:
  297. reader.field(tag, message);
  298. break;
  299. }
  300. }
  301. return message;
  302. }
  303. };
  304. ge.proto.AttrDef.ListListInt.ListInt = class ListInt {
  305. constructor() {
  306. this.list_i = [];
  307. }
  308. static decode(reader, length) {
  309. const message = new ge.proto.AttrDef.ListListInt.ListInt();
  310. const end = length === undefined ? reader.length : reader.position + length;
  311. while (reader.position < end) {
  312. const tag = reader.uint32();
  313. switch (tag >>> 3) {
  314. case 1:
  315. message.list_i = reader.array(message.list_i, () => reader.int64(), tag);
  316. break;
  317. default:
  318. reader.skipType(tag & 7);
  319. break;
  320. }
  321. }
  322. return message;
  323. }
  324. static decodeText(reader) {
  325. const message = new ge.proto.AttrDef.ListListInt.ListInt();
  326. reader.start();
  327. while (!reader.end()) {
  328. const tag = reader.tag();
  329. switch (tag) {
  330. case "list_i":
  331. reader.array(message.list_i, () => reader.int64());
  332. break;
  333. default:
  334. reader.field(tag, message);
  335. break;
  336. }
  337. }
  338. return message;
  339. }
  340. };
  341. ge.proto.AttrDef.ListListFloat = class ListListFloat {
  342. constructor() {
  343. this.list_list_f = [];
  344. }
  345. static decode(reader, length) {
  346. const message = new ge.proto.AttrDef.ListListFloat();
  347. const end = length === undefined ? reader.length : reader.position + length;
  348. while (reader.position < end) {
  349. const tag = reader.uint32();
  350. switch (tag >>> 3) {
  351. case 1:
  352. message.list_list_f.push(ge.proto.AttrDef.ListListFloat.ListFloat.decode(reader, reader.uint32()));
  353. break;
  354. default:
  355. reader.skipType(tag & 7);
  356. break;
  357. }
  358. }
  359. return message;
  360. }
  361. static decodeText(reader) {
  362. const message = new ge.proto.AttrDef.ListListFloat();
  363. reader.start();
  364. while (!reader.end()) {
  365. const tag = reader.tag();
  366. switch (tag) {
  367. case "list_list_f":
  368. message.list_list_f.push(ge.proto.AttrDef.ListListFloat.ListFloat.decodeText(reader));
  369. break;
  370. default:
  371. reader.field(tag, message);
  372. break;
  373. }
  374. }
  375. return message;
  376. }
  377. };
  378. ge.proto.AttrDef.ListListFloat.ListFloat = class ListFloat {
  379. constructor() {
  380. this.list_f = [];
  381. }
  382. static decode(reader, length) {
  383. const message = new ge.proto.AttrDef.ListListFloat.ListFloat();
  384. const end = length === undefined ? reader.length : reader.position + length;
  385. while (reader.position < end) {
  386. const tag = reader.uint32();
  387. switch (tag >>> 3) {
  388. case 1:
  389. message.list_f = reader.floats(message.list_f, tag);
  390. break;
  391. default:
  392. reader.skipType(tag & 7);
  393. break;
  394. }
  395. }
  396. return message;
  397. }
  398. static decodeText(reader) {
  399. const message = new ge.proto.AttrDef.ListListFloat.ListFloat();
  400. reader.start();
  401. while (!reader.end()) {
  402. const tag = reader.tag();
  403. switch (tag) {
  404. case "list_f":
  405. reader.array(message.list_f, () => reader.float());
  406. break;
  407. default:
  408. reader.field(tag, message);
  409. break;
  410. }
  411. }
  412. return message;
  413. }
  414. };
  415. ge.proto.NamedAttrs = class NamedAttrs {
  416. constructor() {
  417. this.attr = {};
  418. }
  419. static decode(reader, length) {
  420. const message = new ge.proto.NamedAttrs();
  421. const end = length === undefined ? reader.length : reader.position + length;
  422. while (reader.position < end) {
  423. const tag = reader.uint32();
  424. switch (tag >>> 3) {
  425. case 1:
  426. message.name = reader.string();
  427. break;
  428. case 2:
  429. reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decode(reader, reader.uint32()));
  430. break;
  431. default:
  432. reader.skipType(tag & 7);
  433. break;
  434. }
  435. }
  436. return message;
  437. }
  438. static decodeText(reader) {
  439. const message = new ge.proto.NamedAttrs();
  440. reader.start();
  441. while (!reader.end()) {
  442. const tag = reader.tag();
  443. switch (tag) {
  444. case "name":
  445. message.name = reader.string();
  446. break;
  447. case "attr":
  448. reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decodeText(reader));
  449. break;
  450. default:
  451. reader.field(tag, message);
  452. break;
  453. }
  454. }
  455. return message;
  456. }
  457. };
  458. ge.proto.NamedAttrs.prototype.name = "";
  459. ge.proto.ShapeDef = class ShapeDef {
  460. constructor() {
  461. this.dim = [];
  462. }
  463. static decode(reader, length) {
  464. const message = new ge.proto.ShapeDef();
  465. const end = length === undefined ? reader.length : reader.position + length;
  466. while (reader.position < end) {
  467. const tag = reader.uint32();
  468. switch (tag >>> 3) {
  469. case 1:
  470. message.dim = reader.array(message.dim, () => reader.int64(), tag);
  471. break;
  472. default:
  473. reader.skipType(tag & 7);
  474. break;
  475. }
  476. }
  477. return message;
  478. }
  479. static decodeText(reader) {
  480. const message = new ge.proto.ShapeDef();
  481. reader.start();
  482. while (!reader.end()) {
  483. const tag = reader.tag();
  484. switch (tag) {
  485. case "dim":
  486. reader.array(message.dim, () => reader.int64());
  487. break;
  488. default:
  489. reader.field(tag, message);
  490. break;
  491. }
  492. }
  493. return message;
  494. }
  495. };
  496. ge.proto.TensorDescriptor = class TensorDescriptor {
  497. constructor() {
  498. this.attr = {};
  499. }
  500. static decode(reader, length) {
  501. const message = new ge.proto.TensorDescriptor();
  502. const end = length === undefined ? reader.length : reader.position + length;
  503. while (reader.position < end) {
  504. const tag = reader.uint32();
  505. switch (tag >>> 3) {
  506. case 1:
  507. message.name = reader.string();
  508. break;
  509. case 2:
  510. message.dtype = reader.int32();
  511. break;
  512. case 3:
  513. message.shape = ge.proto.ShapeDef.decode(reader, reader.uint32());
  514. break;
  515. case 4:
  516. message.layout = reader.string();
  517. break;
  518. case 9:
  519. message.has_out_attr = reader.bool();
  520. break;
  521. case 10:
  522. message.size = reader.int64();
  523. break;
  524. case 11:
  525. message.weight_size = reader.int64();
  526. break;
  527. case 12:
  528. message.reuse_input = reader.bool();
  529. break;
  530. case 13:
  531. message.output_tensor = reader.bool();
  532. break;
  533. case 14:
  534. message.device_type = reader.string();
  535. break;
  536. case 15:
  537. message.input_tensor = reader.bool();
  538. break;
  539. case 16:
  540. message.real_dim_cnt = reader.int64();
  541. break;
  542. case 17:
  543. message.reuse_input_index = reader.int64();
  544. break;
  545. case 18:
  546. message.data_offset = reader.int64();
  547. break;
  548. case 19:
  549. message.cmps_size = reader.int64();
  550. break;
  551. case 20:
  552. message.cmps_tab = reader.string();
  553. break;
  554. case 21:
  555. message.cmps_tab_offset = reader.int64();
  556. break;
  557. case 5:
  558. reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decode(reader, reader.uint32()));
  559. break;
  560. default:
  561. reader.skipType(tag & 7);
  562. break;
  563. }
  564. }
  565. return message;
  566. }
  567. static decodeText(reader) {
  568. const message = new ge.proto.TensorDescriptor();
  569. reader.start();
  570. while (!reader.end()) {
  571. const tag = reader.tag();
  572. switch (tag) {
  573. case "name":
  574. message.name = reader.string();
  575. break;
  576. case "dtype":
  577. message.dtype = reader.enum(ge.proto.DataType);
  578. break;
  579. case "shape":
  580. message.shape = ge.proto.ShapeDef.decodeText(reader);
  581. break;
  582. case "layout":
  583. message.layout = reader.string();
  584. break;
  585. case "has_out_attr":
  586. message.has_out_attr = reader.bool();
  587. break;
  588. case "size":
  589. message.size = reader.int64();
  590. break;
  591. case "weight_size":
  592. message.weight_size = reader.int64();
  593. break;
  594. case "reuse_input":
  595. message.reuse_input = reader.bool();
  596. break;
  597. case "output_tensor":
  598. message.output_tensor = reader.bool();
  599. break;
  600. case "device_type":
  601. message.device_type = reader.string();
  602. break;
  603. case "input_tensor":
  604. message.input_tensor = reader.bool();
  605. break;
  606. case "real_dim_cnt":
  607. message.real_dim_cnt = reader.int64();
  608. break;
  609. case "reuse_input_index":
  610. message.reuse_input_index = reader.int64();
  611. break;
  612. case "data_offset":
  613. message.data_offset = reader.int64();
  614. break;
  615. case "cmps_size":
  616. message.cmps_size = reader.int64();
  617. break;
  618. case "cmps_tab":
  619. message.cmps_tab = reader.string();
  620. break;
  621. case "cmps_tab_offset":
  622. message.cmps_tab_offset = reader.int64();
  623. break;
  624. case "attr":
  625. reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decodeText(reader));
  626. break;
  627. default:
  628. reader.field(tag, message);
  629. break;
  630. }
  631. }
  632. return message;
  633. }
  634. };
  635. ge.proto.TensorDescriptor.prototype.name = "";
  636. ge.proto.TensorDescriptor.prototype.dtype = 0;
  637. ge.proto.TensorDescriptor.prototype.shape = null;
  638. ge.proto.TensorDescriptor.prototype.layout = "";
  639. ge.proto.TensorDescriptor.prototype.has_out_attr = false;
  640. ge.proto.TensorDescriptor.prototype.size = 0n;
  641. ge.proto.TensorDescriptor.prototype.weight_size = 0n;
  642. ge.proto.TensorDescriptor.prototype.reuse_input = false;
  643. ge.proto.TensorDescriptor.prototype.output_tensor = false;
  644. ge.proto.TensorDescriptor.prototype.device_type = "";
  645. ge.proto.TensorDescriptor.prototype.input_tensor = false;
  646. ge.proto.TensorDescriptor.prototype.real_dim_cnt = 0n;
  647. ge.proto.TensorDescriptor.prototype.reuse_input_index = 0n;
  648. ge.proto.TensorDescriptor.prototype.data_offset = 0n;
  649. ge.proto.TensorDescriptor.prototype.cmps_size = 0n;
  650. ge.proto.TensorDescriptor.prototype.cmps_tab = "";
  651. ge.proto.TensorDescriptor.prototype.cmps_tab_offset = 0n;
  652. ge.proto.TensorDef = class TensorDef {
  653. static decode(reader, length) {
  654. const message = new ge.proto.TensorDef();
  655. const end = length === undefined ? reader.length : reader.position + length;
  656. while (reader.position < end) {
  657. const tag = reader.uint32();
  658. switch (tag >>> 3) {
  659. case 1:
  660. message.desc = ge.proto.TensorDescriptor.decode(reader, reader.uint32());
  661. break;
  662. case 2:
  663. message.data = reader.bytes();
  664. break;
  665. default:
  666. reader.skipType(tag & 7);
  667. break;
  668. }
  669. }
  670. return message;
  671. }
  672. static decodeText(reader) {
  673. const message = new ge.proto.TensorDef();
  674. reader.start();
  675. while (!reader.end()) {
  676. const tag = reader.tag();
  677. switch (tag) {
  678. case "desc":
  679. message.desc = ge.proto.TensorDescriptor.decodeText(reader);
  680. break;
  681. case "data":
  682. message.data = reader.bytes();
  683. break;
  684. default:
  685. reader.field(tag, message);
  686. break;
  687. }
  688. }
  689. return message;
  690. }
  691. };
  692. ge.proto.TensorDef.prototype.desc = null;
  693. ge.proto.TensorDef.prototype.data = new Uint8Array([]);
  694. ge.proto.OpDef = class OpDef {
  695. constructor() {
  696. this.input = [];
  697. this.attr = {};
  698. this.input_name = [];
  699. this.src_name = [];
  700. this.src_index = [];
  701. this.dst_name = [];
  702. this.dst_index = [];
  703. this.input_i = [];
  704. this.output_i = [];
  705. this.workspace = [];
  706. this.workspace_bytes = [];
  707. this.is_input_const = [];
  708. this.input_desc = [];
  709. this.output_desc = [];
  710. this.subgraph_name = [];
  711. }
  712. static decode(reader, length) {
  713. const message = new ge.proto.OpDef();
  714. const end = length === undefined ? reader.length : reader.position + length;
  715. while (reader.position < end) {
  716. const tag = reader.uint32();
  717. switch (tag >>> 3) {
  718. case 1:
  719. message.name = reader.string();
  720. break;
  721. case 2:
  722. message.type = reader.string();
  723. break;
  724. case 5:
  725. message.input.push(reader.string());
  726. break;
  727. case 10:
  728. reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decode(reader, reader.uint32()));
  729. break;
  730. case 20:
  731. message.has_out_attr = reader.bool();
  732. break;
  733. case 21:
  734. message.id = reader.int64();
  735. break;
  736. case 22:
  737. message.stream_id = reader.int64();
  738. break;
  739. case 23:
  740. message.input_name.push(reader.string());
  741. break;
  742. case 24:
  743. message.src_name.push(reader.string());
  744. break;
  745. case 25:
  746. message.src_index = reader.array(message.src_index, () => reader.int64(), tag);
  747. break;
  748. case 26:
  749. message.dst_name.push(reader.string());
  750. break;
  751. case 27:
  752. message.dst_index = reader.array(message.dst_index, () => reader.int64(), tag);
  753. break;
  754. case 28:
  755. message.input_i = reader.array(message.input_i, () => reader.int64(), tag);
  756. break;
  757. case 29:
  758. message.output_i = reader.array(message.output_i, () => reader.int64(), tag);
  759. break;
  760. case 30:
  761. message.workspace = reader.array(message.workspace, () => reader.int64(), tag);
  762. break;
  763. case 31:
  764. message.workspace_bytes = reader.array(message.workspace_bytes, () => reader.int64(), tag);
  765. break;
  766. case 32:
  767. message.is_input_const = reader.array(message.is_input_const, () => reader.bool(), tag);
  768. break;
  769. case 33:
  770. message.input_desc.push(ge.proto.TensorDescriptor.decode(reader, reader.uint32()));
  771. break;
  772. case 34:
  773. message.output_desc.push(ge.proto.TensorDescriptor.decode(reader, reader.uint32()));
  774. break;
  775. case 35:
  776. message.subgraph_name.push(reader.string());
  777. break;
  778. default:
  779. reader.skipType(tag & 7);
  780. break;
  781. }
  782. }
  783. return message;
  784. }
  785. static decodeText(reader) {
  786. const message = new ge.proto.OpDef();
  787. reader.start();
  788. while (!reader.end()) {
  789. const tag = reader.tag();
  790. switch (tag) {
  791. case "name":
  792. message.name = reader.string();
  793. break;
  794. case "type":
  795. message.type = reader.string();
  796. break;
  797. case "input":
  798. reader.array(message.input, () => reader.string());
  799. break;
  800. case "attr":
  801. reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decodeText(reader));
  802. break;
  803. case "has_out_attr":
  804. message.has_out_attr = reader.bool();
  805. break;
  806. case "id":
  807. message.id = reader.int64();
  808. break;
  809. case "stream_id":
  810. message.stream_id = reader.int64();
  811. break;
  812. case "input_name":
  813. reader.array(message.input_name, () => reader.string());
  814. break;
  815. case "src_name":
  816. reader.array(message.src_name, () => reader.string());
  817. break;
  818. case "src_index":
  819. reader.array(message.src_index, () => reader.int64());
  820. break;
  821. case "dst_name":
  822. reader.array(message.dst_name, () => reader.string());
  823. break;
  824. case "dst_index":
  825. reader.array(message.dst_index, () => reader.int64());
  826. break;
  827. case "input_i":
  828. reader.array(message.input_i, () => reader.int64());
  829. break;
  830. case "output_i":
  831. reader.array(message.output_i, () => reader.int64());
  832. break;
  833. case "workspace":
  834. reader.array(message.workspace, () => reader.int64());
  835. break;
  836. case "workspace_bytes":
  837. reader.array(message.workspace_bytes, () => reader.int64());
  838. break;
  839. case "is_input_const":
  840. reader.array(message.is_input_const, () => reader.bool());
  841. break;
  842. case "input_desc":
  843. message.input_desc.push(ge.proto.TensorDescriptor.decodeText(reader));
  844. break;
  845. case "output_desc":
  846. message.output_desc.push(ge.proto.TensorDescriptor.decodeText(reader));
  847. break;
  848. case "subgraph_name":
  849. reader.array(message.subgraph_name, () => reader.string());
  850. break;
  851. default:
  852. reader.field(tag, message);
  853. break;
  854. }
  855. }
  856. return message;
  857. }
  858. };
  859. ge.proto.OpDef.prototype.name = "";
  860. ge.proto.OpDef.prototype.type = "";
  861. ge.proto.OpDef.prototype.has_out_attr = false;
  862. ge.proto.OpDef.prototype.id = 0n;
  863. ge.proto.OpDef.prototype.stream_id = 0n;
  864. ge.proto.GraphDef = class GraphDef {
  865. constructor() {
  866. this.input = [];
  867. this.output = [];
  868. this.op = [];
  869. this.attr = {};
  870. }
  871. static decode(reader, length) {
  872. const message = new ge.proto.GraphDef();
  873. const end = length === undefined ? reader.length : reader.position + length;
  874. while (reader.position < end) {
  875. const tag = reader.uint32();
  876. switch (tag >>> 3) {
  877. case 1:
  878. message.name = reader.string();
  879. break;
  880. case 4:
  881. message.input.push(reader.string());
  882. break;
  883. case 5:
  884. message.output.push(reader.string());
  885. break;
  886. case 6:
  887. message.op.push(ge.proto.OpDef.decode(reader, reader.uint32()));
  888. break;
  889. case 11:
  890. reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decode(reader, reader.uint32()));
  891. break;
  892. default:
  893. reader.skipType(tag & 7);
  894. break;
  895. }
  896. }
  897. return message;
  898. }
  899. static decodeText(reader) {
  900. const message = new ge.proto.GraphDef();
  901. reader.start();
  902. while (!reader.end()) {
  903. const tag = reader.tag();
  904. switch (tag) {
  905. case "name":
  906. message.name = reader.string();
  907. break;
  908. case "input":
  909. reader.array(message.input, () => reader.string());
  910. break;
  911. case "output":
  912. reader.array(message.output, () => reader.string());
  913. break;
  914. case "op":
  915. message.op.push(ge.proto.OpDef.decodeText(reader));
  916. break;
  917. case "attr":
  918. reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decodeText(reader));
  919. break;
  920. default:
  921. reader.field(tag, message);
  922. break;
  923. }
  924. }
  925. return message;
  926. }
  927. };
  928. ge.proto.GraphDef.prototype.name = "";
  929. ge.proto.ModelDef = class ModelDef {
  930. constructor() {
  931. this.graph = [];
  932. this.attr = {};
  933. }
  934. static decode(reader, length) {
  935. const message = new ge.proto.ModelDef();
  936. const end = length === undefined ? reader.length : reader.position + length;
  937. while (reader.position < end) {
  938. const tag = reader.uint32();
  939. switch (tag >>> 3) {
  940. case 1:
  941. message.name = reader.string();
  942. break;
  943. case 2:
  944. message.version = reader.uint32();
  945. break;
  946. case 3:
  947. message.custom_version = reader.string();
  948. break;
  949. case 7:
  950. message.graph.push(ge.proto.GraphDef.decode(reader, reader.uint32()));
  951. break;
  952. case 11:
  953. reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decode(reader, reader.uint32()));
  954. break;
  955. default:
  956. reader.skipType(tag & 7);
  957. break;
  958. }
  959. }
  960. return message;
  961. }
  962. static decodeText(reader) {
  963. const message = new ge.proto.ModelDef();
  964. reader.start();
  965. while (!reader.end()) {
  966. const tag = reader.tag();
  967. switch (tag) {
  968. case "name":
  969. message.name = reader.string();
  970. break;
  971. case "version":
  972. message.version = reader.uint32();
  973. break;
  974. case "custom_version":
  975. message.custom_version = reader.string();
  976. break;
  977. case "graph":
  978. message.graph.push(ge.proto.GraphDef.decodeText(reader));
  979. break;
  980. case "attr":
  981. reader.entry(message.attr, () => reader.string(), () => ge.proto.AttrDef.decodeText(reader));
  982. break;
  983. default:
  984. reader.field(tag, message);
  985. break;
  986. }
  987. }
  988. return message;
  989. }
  990. };
  991. ge.proto.ModelDef.prototype.name = "";
  992. ge.proto.ModelDef.prototype.version = 0;
  993. ge.proto.ModelDef.prototype.custom_version = "";