Lutz Roeder 7 éve
szülő
commit
c9a6aeb495
1 módosított fájl, 25 hozzáadás és 1 törlés
  1. 25 1
      src/tflite.js

+ 25 - 1
src/tflite.js

@@ -183,6 +183,14 @@ tflite.LSHProjectionType = {
   DENSE: 2
 };
 
+/**
+ * @enum
+ */
+tflite.LSTMKernelType = {
+  FULL: 0,
+  BASIC: 1
+};
+
 /**
  * @enum
  */
@@ -2204,11 +2212,19 @@ tflite.LSTMOptions.prototype.projClip = function() {
   return offset ? this.bb.readFloat32(this.bb_pos + offset) : 0.0;
 };
 
+/**
+ * @returns {tflite.LSTMKernelType}
+ */
+tflite.LSTMOptions.prototype.kernelType = function() {
+  var offset = this.bb.__offset(this.bb_pos, 10);
+  return offset ? /** @type {tflite.LSTMKernelType} */ (this.bb.readInt8(this.bb_pos + offset)) : tflite.LSTMKernelType.FULL;
+};
+
 /**
  * @param {flatbuffers.Builder} builder
  */
 tflite.LSTMOptions.startLSTMOptions = function(builder) {
-  builder.startObject(3);
+  builder.startObject(4);
 };
 
 /**
@@ -2235,6 +2251,14 @@ tflite.LSTMOptions.addProjClip = function(builder, projClip) {
   builder.addFieldFloat32(2, projClip, 0.0);
 };
 
+/**
+ * @param {flatbuffers.Builder} builder
+ * @param {tflite.LSTMKernelType} kernelType
+ */
+tflite.LSTMOptions.addKernelType = function(builder, kernelType) {
+  builder.addFieldInt8(3, kernelType, tflite.LSTMKernelType.FULL);
+};
+
 /**
  * @param {flatbuffers.Builder} builder
  * @returns {flatbuffers.Offset}