om-proto.js 36 KB

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