Parcourir la source

Update tflite-schema.js

Lutz Roeder il y a 6 ans
Parent
commit
30c9aaead9
1 fichiers modifiés avec 78 ajouts et 4 suppressions
  1. 78 4
      src/tflite-schema.js

+ 78 - 4
src/tflite-schema.js

@@ -219,7 +219,8 @@ TFLITE.BuiltinOperator = {
   SCATTER_ND: 122,
   SELECT_V2: 123,
   DENSIFY: 124,
-  SEGMENT_SUM: 125
+  SEGMENT_SUM: 125,
+  BATCH_MATMUL: 126
 };
 
 /**
@@ -351,7 +352,8 @@ TFLITE.BuiltinOperatorName = {
   '122': 'SCATTER_ND',
   '123': 'SELECT_V2',
   '124': 'DENSIFY',
-  '125': 'SEGMENT_SUM'
+  '125': 'SEGMENT_SUM',
+  '126': 'BATCH_MATMUL'
 };
 
 /**
@@ -458,7 +460,8 @@ TFLITE.BuiltinOptions = {
   ScatterNdOptions: 97,
   SelectV2Options: 98,
   DensifyOptions: 99,
-  SegmentSumOptions: 100
+  SegmentSumOptions: 100,
+  BatchMatMulOptions: 101
 };
 
 /**
@@ -565,7 +568,8 @@ TFLITE.BuiltinOptionsName = {
   '97': 'ScatterNdOptions',
   '98': 'SelectV2Options',
   '99': 'DensifyOptions',
-  '100': 'SegmentSumOptions'
+  '100': 'SegmentSumOptions',
+  '101': 'BatchMatMulOptions'
 };
 
 /**
@@ -11333,6 +11337,76 @@ TFLITE.SegmentSumOptions.createSegmentSumOptions = function(builder) {
   return TFLITE.SegmentSumOptions.endSegmentSumOptions(builder);
 }
 
+/**
+ * @constructor
+ */
+TFLITE.BatchMatMulOptions = function() {
+  /**
+   * @type {flatbuffers.ByteBuffer}
+   */
+  this.bb = null;
+
+  /**
+   * @type {number}
+   */
+  this.bb_pos = 0;
+};
+
+/**
+ * @param {number} i
+ * @param {flatbuffers.ByteBuffer} bb
+ * @returns {TFLITE.BatchMatMulOptions}
+ */
+TFLITE.BatchMatMulOptions.prototype.__init = function(i, bb) {
+  this.bb_pos = i;
+  this.bb = bb;
+  return this;
+};
+
+/**
+ * @param {flatbuffers.ByteBuffer} bb
+ * @param {TFLITE.BatchMatMulOptions=} obj
+ * @returns {TFLITE.BatchMatMulOptions}
+ */
+TFLITE.BatchMatMulOptions.getRootAsBatchMatMulOptions = function(bb, obj) {
+  return (obj || new TFLITE.BatchMatMulOptions).__init(bb.readInt32(bb.position()) + bb.position(), bb);
+};
+
+/**
+ * @param {flatbuffers.ByteBuffer} bb
+ * @param {TFLITE.BatchMatMulOptions=} obj
+ * @returns {TFLITE.BatchMatMulOptions}
+ */
+TFLITE.BatchMatMulOptions.getSizePrefixedRootAsBatchMatMulOptions = function(bb, obj) {
+  bb.setPosition(bb.position() + flatbuffers.SIZE_PREFIX_LENGTH);
+  return (obj || new TFLITE.BatchMatMulOptions).__init(bb.readInt32(bb.position()) + bb.position(), bb);
+};
+
+/**
+ * @param {flatbuffers.Builder} builder
+ */
+TFLITE.BatchMatMulOptions.startBatchMatMulOptions = function(builder) {
+  builder.startObject(0);
+};
+
+/**
+ * @param {flatbuffers.Builder} builder
+ * @returns {flatbuffers.Offset}
+ */
+TFLITE.BatchMatMulOptions.endBatchMatMulOptions = function(builder) {
+  var offset = builder.endObject();
+  return offset;
+};
+
+/**
+ * @param {flatbuffers.Builder} builder
+ * @returns {flatbuffers.Offset}
+ */
+TFLITE.BatchMatMulOptions.createBatchMatMulOptions = function(builder) {
+  TFLITE.BatchMatMulOptions.startBatchMatMulOptions(builder);
+  return TFLITE.BatchMatMulOptions.endBatchMatMulOptions(builder);
+}
+
 /**
  * @constructor
  */