bigdl-proto.js 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630
  1. export const com = {};
  2. export const google = {};
  3. com.intel = {};
  4. com.intel.analytics = {};
  5. com.intel.analytics.bigdl = {};
  6. com.intel.analytics.bigdl.serialization = {};
  7. com.intel.analytics.bigdl.serialization.BigDLModule = class BigDLModule {
  8. constructor() {
  9. this.subModules = [];
  10. this.preModules = [];
  11. this.nextModules = [];
  12. this.attr = {};
  13. this.parameters = [];
  14. this.inputScales = [];
  15. this.outputScales = [];
  16. this.weightScales = [];
  17. }
  18. static decode(reader, length) {
  19. const message = new com.intel.analytics.bigdl.serialization.BigDLModule();
  20. const end = length === undefined ? reader.length : reader.position + length;
  21. while (reader.position < end) {
  22. const tag = reader.uint32();
  23. switch (tag >>> 3) {
  24. case 1:
  25. message.name = reader.string();
  26. break;
  27. case 2:
  28. message.subModules.push(com.intel.analytics.bigdl.serialization.BigDLModule.decode(reader, reader.uint32()));
  29. break;
  30. case 3:
  31. message.weight = com.intel.analytics.bigdl.serialization.BigDLTensor.decode(reader, reader.uint32());
  32. break;
  33. case 4:
  34. message.bias = com.intel.analytics.bigdl.serialization.BigDLTensor.decode(reader, reader.uint32());
  35. break;
  36. case 5:
  37. message.preModules.push(reader.string());
  38. break;
  39. case 6:
  40. message.nextModules.push(reader.string());
  41. break;
  42. case 7:
  43. message.moduleType = reader.string();
  44. break;
  45. case 8:
  46. reader.entry(message.attr, () => reader.string(), () => com.intel.analytics.bigdl.serialization.AttrValue.decode(reader, reader.uint32()));
  47. break;
  48. case 9:
  49. message.version = reader.string();
  50. break;
  51. case 10:
  52. message.train = reader.bool();
  53. break;
  54. case 11:
  55. message.namePostfix = reader.string();
  56. break;
  57. case 12:
  58. message.id = reader.int32();
  59. break;
  60. case 13:
  61. message.inputShape = com.intel.analytics.bigdl.serialization.Shape.decode(reader, reader.uint32());
  62. break;
  63. case 14:
  64. message.outputShape = com.intel.analytics.bigdl.serialization.Shape.decode(reader, reader.uint32());
  65. break;
  66. case 15:
  67. message.hasParameters = reader.bool();
  68. break;
  69. case 16:
  70. message.parameters.push(com.intel.analytics.bigdl.serialization.BigDLTensor.decode(reader, reader.uint32()));
  71. break;
  72. case 17:
  73. message.isMklInt8Enabled = reader.bool();
  74. break;
  75. case 18:
  76. message.inputDimMasks = reader.int32();
  77. break;
  78. case 19:
  79. message.inputScales.push(com.intel.analytics.bigdl.serialization.AttrValue.decode(reader, reader.uint32()));
  80. break;
  81. case 20:
  82. message.outputDimMasks = reader.int32();
  83. break;
  84. case 21:
  85. message.outputScales.push(com.intel.analytics.bigdl.serialization.AttrValue.decode(reader, reader.uint32()));
  86. break;
  87. case 22:
  88. message.weightDimMasks = reader.int32();
  89. break;
  90. case 23:
  91. message.weightScales.push(com.intel.analytics.bigdl.serialization.AttrValue.decode(reader, reader.uint32()));
  92. break;
  93. default:
  94. reader.skipType(tag & 7);
  95. break;
  96. }
  97. }
  98. return message;
  99. }
  100. };
  101. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.name = "";
  102. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.weight = null;
  103. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.bias = null;
  104. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.moduleType = "";
  105. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.version = "";
  106. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.train = false;
  107. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.namePostfix = "";
  108. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.id = 0;
  109. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.inputShape = null;
  110. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.outputShape = null;
  111. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.hasParameters = false;
  112. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.isMklInt8Enabled = false;
  113. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.inputDimMasks = 0;
  114. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.outputDimMasks = 0;
  115. com.intel.analytics.bigdl.serialization.BigDLModule.prototype.weightDimMasks = 0;
  116. com.intel.analytics.bigdl.serialization.VarFormat = {
  117. "EMPTY_FORMAT": 0,
  118. "DEFAULT": 1,
  119. "ONE_D": 2,
  120. "IN_OUT": 3,
  121. "OUT_IN": 4,
  122. "IN_OUT_KW_KH": 5,
  123. "OUT_IN_KW_KH": 6,
  124. "GP_OUT_IN_KW_KH": 7,
  125. "GP_IN_OUT_KW_KH": 8,
  126. "OUT_IN_KT_KH_KW": 9
  127. };
  128. com.intel.analytics.bigdl.serialization.InitMethodType = {
  129. "EMPTY_INITIALIZATION": 0,
  130. "RANDOM_UNIFORM": 1,
  131. "RANDOM_UNIFORM_PARAM": 2,
  132. "RANDOM_NORMAL": 3,
  133. "ZEROS": 4,
  134. "ONES": 5,
  135. "CONST": 6,
  136. "XAVIER": 7,
  137. "BILINEARFILLER": 8
  138. };
  139. com.intel.analytics.bigdl.serialization.RegularizerType = {
  140. "L1L2Regularizer": 0,
  141. "L1Regularizer": 1,
  142. "L2Regularizer": 2
  143. };
  144. com.intel.analytics.bigdl.serialization.InputDataFormat = {
  145. "NCHW": 0,
  146. "NHWC": 1
  147. };
  148. com.intel.analytics.bigdl.serialization.TensorType = {
  149. "DENSE": 0,
  150. "QUANT": 1
  151. };
  152. com.intel.analytics.bigdl.serialization.InitMethod = class InitMethod {
  153. constructor() {
  154. this.data = [];
  155. }
  156. static decode(reader, length) {
  157. const message = new com.intel.analytics.bigdl.serialization.InitMethod();
  158. const end = length === undefined ? reader.length : reader.position + length;
  159. while (reader.position < end) {
  160. const tag = reader.uint32();
  161. switch (tag >>> 3) {
  162. case 1:
  163. message.methodType = reader.int32();
  164. break;
  165. case 2:
  166. message.data = reader.doubles(message.data, tag);
  167. break;
  168. default:
  169. reader.skipType(tag & 7);
  170. break;
  171. }
  172. }
  173. return message;
  174. }
  175. };
  176. com.intel.analytics.bigdl.serialization.InitMethod.prototype.methodType = 0;
  177. com.intel.analytics.bigdl.serialization.BigDLTensor = class BigDLTensor {
  178. constructor() {
  179. this.size = [];
  180. this.stride = [];
  181. }
  182. static decode(reader, length) {
  183. const message = new com.intel.analytics.bigdl.serialization.BigDLTensor();
  184. const end = length === undefined ? reader.length : reader.position + length;
  185. while (reader.position < end) {
  186. const tag = reader.uint32();
  187. switch (tag >>> 3) {
  188. case 1:
  189. message.datatype = reader.int32();
  190. break;
  191. case 2:
  192. message.size = reader.array(message.size, () => reader.int32(), tag);
  193. break;
  194. case 3:
  195. message.stride = reader.array(message.stride, () => reader.int32(), tag);
  196. break;
  197. case 4:
  198. message.offset = reader.int32();
  199. break;
  200. case 5:
  201. message.dimension = reader.int32();
  202. break;
  203. case 6:
  204. message.nElements = reader.int32();
  205. break;
  206. case 7:
  207. message.isScalar = reader.bool();
  208. break;
  209. case 8:
  210. message.storage = com.intel.analytics.bigdl.serialization.TensorStorage.decode(reader, reader.uint32());
  211. break;
  212. case 9:
  213. message.id = reader.int32();
  214. break;
  215. case 10:
  216. message.tensorType = reader.int32();
  217. break;
  218. default:
  219. reader.skipType(tag & 7);
  220. break;
  221. }
  222. }
  223. return message;
  224. }
  225. };
  226. com.intel.analytics.bigdl.serialization.BigDLTensor.prototype.datatype = 0;
  227. com.intel.analytics.bigdl.serialization.BigDLTensor.prototype.offset = 0;
  228. com.intel.analytics.bigdl.serialization.BigDLTensor.prototype.dimension = 0;
  229. com.intel.analytics.bigdl.serialization.BigDLTensor.prototype.nElements = 0;
  230. com.intel.analytics.bigdl.serialization.BigDLTensor.prototype.isScalar = false;
  231. com.intel.analytics.bigdl.serialization.BigDLTensor.prototype.storage = null;
  232. com.intel.analytics.bigdl.serialization.BigDLTensor.prototype.id = 0;
  233. com.intel.analytics.bigdl.serialization.BigDLTensor.prototype.tensorType = 0;
  234. com.intel.analytics.bigdl.serialization.TensorStorage = class TensorStorage {
  235. constructor() {
  236. this.float_data = [];
  237. this.double_data = [];
  238. this.bool_data = [];
  239. this.string_data = [];
  240. this.int_data = [];
  241. this.long_data = [];
  242. this.bytes_data = [];
  243. }
  244. static decode(reader, length) {
  245. const message = new com.intel.analytics.bigdl.serialization.TensorStorage();
  246. const end = length === undefined ? reader.length : reader.position + length;
  247. while (reader.position < end) {
  248. const tag = reader.uint32();
  249. switch (tag >>> 3) {
  250. case 1:
  251. message.datatype = reader.int32();
  252. break;
  253. case 2:
  254. message.float_data = reader.floats(message.float_data, tag);
  255. break;
  256. case 3:
  257. message.double_data = reader.doubles(message.double_data, tag);
  258. break;
  259. case 4:
  260. message.bool_data = reader.array(message.bool_data, () => reader.bool(), tag);
  261. break;
  262. case 5:
  263. message.string_data.push(reader.string());
  264. break;
  265. case 6:
  266. message.int_data = reader.array(message.int_data, () => reader.int32(), tag);
  267. break;
  268. case 7:
  269. message.long_data = reader.array(message.long_data, () => reader.int64(), tag);
  270. break;
  271. case 8:
  272. message.bytes_data.push(reader.bytes());
  273. break;
  274. case 9:
  275. message.id = reader.int32();
  276. break;
  277. default:
  278. reader.skipType(tag & 7);
  279. break;
  280. }
  281. }
  282. return message;
  283. }
  284. };
  285. com.intel.analytics.bigdl.serialization.TensorStorage.prototype.datatype = 0;
  286. com.intel.analytics.bigdl.serialization.TensorStorage.prototype.id = 0;
  287. com.intel.analytics.bigdl.serialization.Regularizer = class Regularizer {
  288. constructor() {
  289. this.regularData = [];
  290. }
  291. static decode(reader, length) {
  292. const message = new com.intel.analytics.bigdl.serialization.Regularizer();
  293. const end = length === undefined ? reader.length : reader.position + length;
  294. while (reader.position < end) {
  295. const tag = reader.uint32();
  296. switch (tag >>> 3) {
  297. case 1:
  298. message.regularizerType = reader.int32();
  299. break;
  300. case 2:
  301. message.regularData = reader.doubles(message.regularData, tag);
  302. break;
  303. default:
  304. reader.skipType(tag & 7);
  305. break;
  306. }
  307. }
  308. return message;
  309. }
  310. };
  311. com.intel.analytics.bigdl.serialization.Regularizer.prototype.regularizerType = 0;
  312. com.intel.analytics.bigdl.serialization.DataType = {
  313. "INT32": 0,
  314. "INT64": 1,
  315. "FLOAT": 2,
  316. "DOUBLE": 3,
  317. "STRING": 4,
  318. "BOOL": 5,
  319. "CHAR": 6,
  320. "SHORT": 7,
  321. "BYTES": 8,
  322. "REGULARIZER": 9,
  323. "TENSOR": 10,
  324. "VARIABLE_FORMAT": 11,
  325. "INITMETHOD": 12,
  326. "MODULE": 13,
  327. "NAME_ATTR_LIST": 14,
  328. "ARRAY_VALUE": 15,
  329. "DATA_FORMAT": 16,
  330. "CUSTOM": 17,
  331. "SHAPE": 18
  332. };
  333. com.intel.analytics.bigdl.serialization.AttrValue = class AttrValue {
  334. get value() {
  335. com.intel.analytics.bigdl.serialization.AttrValue.valueSet = com.intel.analytics.bigdl.serialization.AttrValue.valueSet || new Set(["int32Value", "int64Value", "floatValue", "doubleValue", "stringValue", "boolValue", "regularizerValue", "tensorValue", "variableFormatValue", "initMethodValue", "bigDLModuleValue", "nameAttrListValue", "arrayValue", "dataFormatValue", "customValue", "shape"]);
  336. return Object.keys(this).find((key) => com.intel.analytics.bigdl.serialization.AttrValue.valueSet.has(key) && this[key] !== null);
  337. }
  338. static decode(reader, length) {
  339. const message = new com.intel.analytics.bigdl.serialization.AttrValue();
  340. const end = length === undefined ? reader.length : reader.position + length;
  341. while (reader.position < end) {
  342. const tag = reader.uint32();
  343. switch (tag >>> 3) {
  344. case 1:
  345. message.dataType = reader.int32();
  346. break;
  347. case 2:
  348. message.subType = reader.string();
  349. break;
  350. case 3:
  351. message.int32Value = reader.int32();
  352. break;
  353. case 4:
  354. message.int64Value = reader.int64();
  355. break;
  356. case 5:
  357. message.floatValue = reader.float();
  358. break;
  359. case 6:
  360. message.doubleValue = reader.double();
  361. break;
  362. case 7:
  363. message.stringValue = reader.string();
  364. break;
  365. case 8:
  366. message.boolValue = reader.bool();
  367. break;
  368. case 9:
  369. message.regularizerValue = com.intel.analytics.bigdl.serialization.Regularizer.decode(reader, reader.uint32());
  370. break;
  371. case 10:
  372. message.tensorValue = com.intel.analytics.bigdl.serialization.BigDLTensor.decode(reader, reader.uint32());
  373. break;
  374. case 11:
  375. message.variableFormatValue = reader.int32();
  376. break;
  377. case 12:
  378. message.initMethodValue = com.intel.analytics.bigdl.serialization.InitMethod.decode(reader, reader.uint32());
  379. break;
  380. case 13:
  381. message.bigDLModuleValue = com.intel.analytics.bigdl.serialization.BigDLModule.decode(reader, reader.uint32());
  382. break;
  383. case 14:
  384. message.nameAttrListValue = com.intel.analytics.bigdl.serialization.NameAttrList.decode(reader, reader.uint32());
  385. break;
  386. case 15:
  387. message.arrayValue = com.intel.analytics.bigdl.serialization.AttrValue.ArrayValue.decode(reader, reader.uint32());
  388. break;
  389. case 16:
  390. message.dataFormatValue = reader.int32();
  391. break;
  392. case 17:
  393. message.customValue = google.protobuf.Any.decode(reader, reader.uint32());
  394. break;
  395. case 18:
  396. message.shape = com.intel.analytics.bigdl.serialization.Shape.decode(reader, reader.uint32());
  397. break;
  398. default:
  399. reader.skipType(tag & 7);
  400. break;
  401. }
  402. }
  403. return message;
  404. }
  405. };
  406. com.intel.analytics.bigdl.serialization.AttrValue.prototype.dataType = 0;
  407. com.intel.analytics.bigdl.serialization.AttrValue.prototype.subType = "";
  408. com.intel.analytics.bigdl.serialization.AttrValue.ArrayValue = class ArrayValue {
  409. constructor() {
  410. this.i32 = [];
  411. this.i64 = [];
  412. this.flt = [];
  413. this.dbl = [];
  414. this.str = [];
  415. this.boolean = [];
  416. this.Regularizer = [];
  417. this.tensor = [];
  418. this.variableFormat = [];
  419. this.initMethod = [];
  420. this.bigDLModule = [];
  421. this.nameAttrList = [];
  422. this.dataFormat = [];
  423. this.custom = [];
  424. this.shape = [];
  425. }
  426. static decode(reader, length) {
  427. const message = new com.intel.analytics.bigdl.serialization.AttrValue.ArrayValue();
  428. const end = length === undefined ? reader.length : reader.position + length;
  429. while (reader.position < end) {
  430. const tag = reader.uint32();
  431. switch (tag >>> 3) {
  432. case 1:
  433. message.size = reader.int32();
  434. break;
  435. case 2:
  436. message.datatype = reader.int32();
  437. break;
  438. case 3:
  439. message.i32 = reader.array(message.i32, () => reader.int32(), tag);
  440. break;
  441. case 4:
  442. message.i64 = reader.array(message.i64, () => reader.int64(), tag);
  443. break;
  444. case 5:
  445. message.flt = reader.floats(message.flt, tag);
  446. break;
  447. case 6:
  448. message.dbl = reader.doubles(message.dbl, tag);
  449. break;
  450. case 7:
  451. message.str.push(reader.string());
  452. break;
  453. case 8:
  454. message.boolean = reader.array(message.boolean, () => reader.bool(), tag);
  455. break;
  456. case 9:
  457. message.Regularizer.push(com.intel.analytics.bigdl.serialization.Regularizer.decode(reader, reader.uint32()));
  458. break;
  459. case 10:
  460. message.tensor.push(com.intel.analytics.bigdl.serialization.BigDLTensor.decode(reader, reader.uint32()));
  461. break;
  462. case 11:
  463. message.variableFormat = reader.array(message.variableFormat, () => reader.int32(), tag);
  464. break;
  465. case 12:
  466. message.initMethod.push(com.intel.analytics.bigdl.serialization.InitMethod.decode(reader, reader.uint32()));
  467. break;
  468. case 13:
  469. message.bigDLModule.push(com.intel.analytics.bigdl.serialization.BigDLModule.decode(reader, reader.uint32()));
  470. break;
  471. case 14:
  472. message.nameAttrList.push(com.intel.analytics.bigdl.serialization.NameAttrList.decode(reader, reader.uint32()));
  473. break;
  474. case 15:
  475. message.dataFormat = reader.array(message.dataFormat, () => reader.int32(), tag);
  476. break;
  477. case 16:
  478. message.custom.push(google.protobuf.Any.decode(reader, reader.uint32()));
  479. break;
  480. case 17:
  481. message.shape.push(com.intel.analytics.bigdl.serialization.Shape.decode(reader, reader.uint32()));
  482. break;
  483. default:
  484. reader.skipType(tag & 7);
  485. break;
  486. }
  487. }
  488. return message;
  489. }
  490. };
  491. com.intel.analytics.bigdl.serialization.AttrValue.ArrayValue.prototype.size = 0;
  492. com.intel.analytics.bigdl.serialization.AttrValue.ArrayValue.prototype.datatype = 0;
  493. com.intel.analytics.bigdl.serialization.NameAttrList = class NameAttrList {
  494. constructor() {
  495. this.attr = {};
  496. }
  497. static decode(reader, length) {
  498. const message = new com.intel.analytics.bigdl.serialization.NameAttrList();
  499. const end = length === undefined ? reader.length : reader.position + length;
  500. while (reader.position < end) {
  501. const tag = reader.uint32();
  502. switch (tag >>> 3) {
  503. case 1:
  504. message.name = reader.string();
  505. break;
  506. case 2:
  507. reader.entry(message.attr, () => reader.string(), () => com.intel.analytics.bigdl.serialization.AttrValue.decode(reader, reader.uint32()));
  508. break;
  509. default:
  510. reader.skipType(tag & 7);
  511. break;
  512. }
  513. }
  514. return message;
  515. }
  516. };
  517. com.intel.analytics.bigdl.serialization.NameAttrList.prototype.name = "";
  518. com.intel.analytics.bigdl.serialization.Shape = class Shape {
  519. constructor() {
  520. this.shapeValue = [];
  521. this.shape = [];
  522. }
  523. static decode(reader, length) {
  524. const message = new com.intel.analytics.bigdl.serialization.Shape();
  525. const end = length === undefined ? reader.length : reader.position + length;
  526. while (reader.position < end) {
  527. const tag = reader.uint32();
  528. switch (tag >>> 3) {
  529. case 1:
  530. message.shapeType = reader.int32();
  531. break;
  532. case 2:
  533. message.ssize = reader.int32();
  534. break;
  535. case 3:
  536. message.shapeValue = reader.array(message.shapeValue, () => reader.int32(), tag);
  537. break;
  538. case 4:
  539. message.shape.push(com.intel.analytics.bigdl.serialization.Shape.decode(reader, reader.uint32()));
  540. break;
  541. default:
  542. reader.skipType(tag & 7);
  543. break;
  544. }
  545. }
  546. return message;
  547. }
  548. };
  549. com.intel.analytics.bigdl.serialization.Shape.prototype.shapeType = 0;
  550. com.intel.analytics.bigdl.serialization.Shape.prototype.ssize = 0;
  551. com.intel.analytics.bigdl.serialization.Shape.ShapeType = {
  552. "SINGLE": 0,
  553. "MULTI": 1
  554. };
  555. google.protobuf = {};
  556. google.protobuf.Any = class Any {
  557. static decode(reader, length) {
  558. const message = new google.protobuf.Any();
  559. const end = length === undefined ? reader.length : reader.position + length;
  560. while (reader.position < end) {
  561. const tag = reader.uint32();
  562. switch (tag >>> 3) {
  563. case 1:
  564. message.type_url = reader.string();
  565. break;
  566. case 2:
  567. message.value = reader.bytes();
  568. break;
  569. default:
  570. reader.skipType(tag & 7);
  571. break;
  572. }
  573. }
  574. return message;
  575. }
  576. };
  577. google.protobuf.Any.prototype.type_url = "";
  578. google.protobuf.Any.prototype.value = new Uint8Array([]);