Lutz Roeder 7 лет назад
Родитель
Сommit
c46e1c482c
1 измененных файлов с 76 добавлено и 4 удалено
  1. 76 4
      src/tflite.js

+ 76 - 4
src/tflite.js

@@ -16,7 +16,8 @@ tflite.TensorType = {
   UINT8: 3,
   INT64: 4,
   STRING: 5,
-  BOOL: 6
+  BOOL: 6,
+  INT16: 7
 };
 
 /**
@@ -94,7 +95,8 @@ tflite.BuiltinOperator = {
   TILE: 69,
   EXPAND_DIMS: 70,
   EQUAL: 71,
-  NOT_EQUAL: 72
+  NOT_EQUAL: 72,
+  LOG: 73
 };
 
 /**
@@ -564,11 +566,19 @@ tflite.Tensor.prototype.quantization = function(obj) {
   return offset ? (obj || new tflite.QuantizationParameters).__init(this.bb.__indirect(this.bb_pos + offset), this.bb) : null;
 };
 
+/**
+ * @returns {boolean}
+ */
+tflite.Tensor.prototype.isVariable = function() {
+  var offset = this.bb.__offset(this.bb_pos, 14);
+  return offset ? !!this.bb.readInt8(this.bb_pos + offset) : false;
+};
+
 /**
  * @param {flatbuffers.Builder} builder
  */
 tflite.Tensor.startTensor = function(builder) {
-  builder.startObject(5);
+  builder.startObject(6);
 };
 
 /**
@@ -632,6 +642,14 @@ tflite.Tensor.addQuantization = function(builder, quantizationOffset) {
   builder.addFieldOffset(4, quantizationOffset, 0);
 };
 
+/**
+ * @param {flatbuffers.Builder} builder
+ * @param {boolean} isVariable
+ */
+tflite.Tensor.addIsVariable = function(builder, isVariable) {
+  builder.addFieldInt8(5, +isVariable, +false);
+};
+
 /**
  * @param {flatbuffers.Builder} builder
  * @returns {flatbuffers.Offset}
@@ -4945,11 +4963,36 @@ tflite.Operator.prototype.customOptionsFormat = function() {
   return offset ? /** @type {tflite.CustomOptionsFormat} */ (this.bb.readInt8(this.bb_pos + offset)) : tflite.CustomOptionsFormat.FLEXBUFFERS;
 };
 
+/**
+ * @param {number} index
+ * @returns {boolean}
+ */
+tflite.Operator.prototype.mutatingVariableInputs = function(index) {
+  var offset = this.bb.__offset(this.bb_pos, 18);
+  return offset ? !!this.bb.readInt8(this.bb.__vector(this.bb_pos + offset) + index) : false;
+};
+
+/**
+ * @returns {number}
+ */
+tflite.Operator.prototype.mutatingVariableInputsLength = function() {
+  var offset = this.bb.__offset(this.bb_pos, 18);
+  return offset ? this.bb.__vector_len(this.bb_pos + offset) : 0;
+};
+
+/**
+ * @returns {Int8Array}
+ */
+tflite.Operator.prototype.mutatingVariableInputsArray = function() {
+  var offset = this.bb.__offset(this.bb_pos, 18);
+  return offset ? new Int8Array(this.bb.bytes().buffer, this.bb.bytes().byteOffset + this.bb.__vector(this.bb_pos + offset), this.bb.__vector_len(this.bb_pos + offset)) : null;
+};
+
 /**
  * @param {flatbuffers.Builder} builder
  */
 tflite.Operator.startOperator = function(builder) {
-  builder.startObject(7);
+  builder.startObject(8);
 };
 
 /**
@@ -5071,6 +5114,35 @@ tflite.Operator.addCustomOptionsFormat = function(builder, customOptionsFormat)
   builder.addFieldInt8(6, customOptionsFormat, tflite.CustomOptionsFormat.FLEXBUFFERS);
 };
 
+/**
+ * @param {flatbuffers.Builder} builder
+ * @param {flatbuffers.Offset} mutatingVariableInputsOffset
+ */
+tflite.Operator.addMutatingVariableInputs = function(builder, mutatingVariableInputsOffset) {
+  builder.addFieldOffset(7, mutatingVariableInputsOffset, 0);
+};
+
+/**
+ * @param {flatbuffers.Builder} builder
+ * @param {Array.<boolean>} data
+ * @returns {flatbuffers.Offset}
+ */
+tflite.Operator.createMutatingVariableInputsVector = function(builder, data) {
+  builder.startVector(1, data.length, 1);
+  for (var i = data.length - 1; i >= 0; i--) {
+    builder.addInt8(+data[i]);
+  }
+  return builder.endVector();
+};
+
+/**
+ * @param {flatbuffers.Builder} builder
+ * @param {number} numElems
+ */
+tflite.Operator.startMutatingVariableInputsVector = function(builder, numElems) {
+  builder.startVector(1, numElems, 1);
+};
+
 /**
  * @param {flatbuffers.Builder} builder
  * @returns {flatbuffers.Offset}