Bladeren bron

Update tflite-schema.js

Lutz Roeder 7 jaren geleden
bovenliggende
commit
d0befa54fa
2 gewijzigde bestanden met toevoegingen van 55 en 4 verwijderingen
  1. 55 2
      src/tflite-schema.js
  2. 0 2
      tools/tflite

+ 55 - 2
src/tflite-schema.js

@@ -139,7 +139,8 @@ tflite_schema.BuiltinOperator = {
   UNIQUE: 103, 103: 'UNIQUE',
   CEIL: 104, 104: 'CEIL',
   REVERSE_V2: 105, 105: 'REVERSE_V2',
-  ADD_N: 106, 106: 'ADD_N'
+  ADD_N: 106, 106: 'ADD_N',
+  GATHER_ND: 107, 107: 'GATHER_ND'
 };
 
 /**
@@ -228,7 +229,8 @@ tflite_schema.BuiltinOptions = {
   SplitVOptions: 79, 79: 'SplitVOptions',
   UniqueOptions: 80, 80: 'UniqueOptions',
   ReverseV2Options: 81, 81: 'ReverseV2Options',
-  AddNOptions: 82, 82: 'AddNOptions'
+  AddNOptions: 82, 82: 'AddNOptions',
+  GatherNdOptions: 83, 83: 'GatherNdOptions'
 };
 
 /**
@@ -6802,6 +6804,57 @@ tflite_schema.AddNOptions.endAddNOptions = function(builder) {
   return offset;
 };
 
+/**
+ * @constructor
+ */
+tflite_schema.GatherNdOptions = function() {
+  /**
+   * @type {flatbuffers.ByteBuffer}
+   */
+  this.bb = null;
+
+  /**
+   * @type {number}
+   */
+  this.bb_pos = 0;
+};
+
+/**
+ * @param {number} i
+ * @param {flatbuffers.ByteBuffer} bb
+ * @returns {tflite_schema.GatherNdOptions}
+ */
+tflite_schema.GatherNdOptions.prototype.__init = function(i, bb) {
+  this.bb_pos = i;
+  this.bb = bb;
+  return this;
+};
+
+/**
+ * @param {flatbuffers.ByteBuffer} bb
+ * @param {tflite_schema.GatherNdOptions=} obj
+ * @returns {tflite_schema.GatherNdOptions}
+ */
+tflite_schema.GatherNdOptions.getRootAsGatherNdOptions = function(bb, obj) {
+  return (obj || new tflite_schema.GatherNdOptions).__init(bb.readInt32(bb.position()) + bb.position(), bb);
+};
+
+/**
+ * @param {flatbuffers.Builder} builder
+ */
+tflite_schema.GatherNdOptions.startGatherNdOptions = function(builder) {
+  builder.startObject(0);
+};
+
+/**
+ * @param {flatbuffers.Builder} builder
+ * @returns {flatbuffers.Offset}
+ */
+tflite_schema.GatherNdOptions.endGatherNdOptions = function(builder) {
+  var offset = builder.endObject();
+  return offset;
+};
+
 /**
  * @constructor
  */

+ 0 - 2
tools/tflite

@@ -42,7 +42,6 @@ sync() {
 
 install() {
     bold "flatbuffers install"
-    echo 1
     case "$(uname)" in
         "Linux")
             if [ -z "$(which cmake)" ]; then
@@ -55,7 +54,6 @@ install() {
             fi
             ;;
     esac
-    echo "Build flatbuffers..."
     pushd "${third_party}/flatbuffers" > /dev/null
     cmake -G "Unix Makefiles" .
     make