|
|
@@ -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}
|