Przeglądaj źródła

Update tflite-schema.js

Lutz Roeder 7 lat temu
rodzic
commit
af300b39f6
4 zmienionych plików z 74 dodań i 5 usunięć
  1. 71 2
      src/tflite-schema.js
  2. 1 1
      tools/caffe
  3. 1 1
      tools/cntk
  4. 1 1
      tools/coreml

+ 71 - 2
src/tflite-schema.js

@@ -134,7 +134,8 @@ tflite_schema.BuiltinOperator = {
   LEAKY_RELU: 98, 98: 'LEAKY_RELU',
   SQUARED_DIFFERENCE: 99, 99: 'SQUARED_DIFFERENCE',
   MIRROR_PAD: 100, 100: 'MIRROR_PAD',
-  ABS: 101, 101: 'ABS'
+  ABS: 101, 101: 'ABS',
+  SPLIT_V: 102, 102: 'SPLIT_V'
 };
 
 /**
@@ -219,7 +220,8 @@ tflite_schema.BuiltinOptions = {
   LeakyReluOptions: 75, 75: 'LeakyReluOptions',
   SquaredDifferenceOptions: 76, 76: 'SquaredDifferenceOptions',
   MirrorPadOptions: 77, 77: 'MirrorPadOptions',
-  AbsOptions: 78, 78: 'AbsOptions'
+  AbsOptions: 78, 78: 'AbsOptions',
+  SplitVOptions: 79, 79: 'SplitVOptions'
 };
 
 /**
@@ -4136,6 +4138,73 @@ tflite_schema.SplitOptions.endSplitOptions = function(builder) {
   return offset;
 };
 
+/**
+ * @constructor
+ */
+tflite_schema.SplitVOptions = function() {
+  /**
+   * @type {flatbuffers.ByteBuffer}
+   */
+  this.bb = null;
+
+  /**
+   * @type {number}
+   */
+  this.bb_pos = 0;
+};
+
+/**
+ * @param {number} i
+ * @param {flatbuffers.ByteBuffer} bb
+ * @returns {tflite_schema.SplitVOptions}
+ */
+tflite_schema.SplitVOptions.prototype.__init = function(i, bb) {
+  this.bb_pos = i;
+  this.bb = bb;
+  return this;
+};
+
+/**
+ * @param {flatbuffers.ByteBuffer} bb
+ * @param {tflite_schema.SplitVOptions=} obj
+ * @returns {tflite_schema.SplitVOptions}
+ */
+tflite_schema.SplitVOptions.getRootAsSplitVOptions = function(bb, obj) {
+  return (obj || new tflite_schema.SplitVOptions).__init(bb.readInt32(bb.position()) + bb.position(), bb);
+};
+
+/**
+ * @returns {number}
+ */
+tflite_schema.SplitVOptions.prototype.numSplits = function() {
+  var offset = this.bb.__offset(this.bb_pos, 4);
+  return offset ? this.bb.readInt32(this.bb_pos + offset) : 0;
+};
+
+/**
+ * @param {flatbuffers.Builder} builder
+ */
+tflite_schema.SplitVOptions.startSplitVOptions = function(builder) {
+  builder.startObject(1);
+};
+
+/**
+ * @param {flatbuffers.Builder} builder
+ * @param {number} numSplits
+ */
+tflite_schema.SplitVOptions.addNumSplits = function(builder, numSplits) {
+  builder.addFieldInt32(0, numSplits, 0);
+};
+
+/**
+ * @param {flatbuffers.Builder} builder
+ * @returns {flatbuffers.Offset}
+ */
+tflite_schema.SplitVOptions.endSplitVOptions = function(builder) {
+  var offset = builder.endObject();
+  return offset;
+};
+
 /**
  * @constructor
  */

+ 1 - 1
tools/caffe

@@ -28,7 +28,7 @@ git_sync() {
 
 clean() {
     bold "caffe clean"
-    rm -tf ${third_party}/caffe
+    rm -rf ${third_party}/caffe
 }
 
 sync() {

+ 1 - 1
tools/cntk

@@ -28,7 +28,7 @@ git_sync() {
 
 clean() {
     bold "cntk clean"
-    rm -tf ${third_party}/cntk
+    rm -rf ${third_party}/cntk
 }
 
 sync() {

+ 1 - 1
tools/coreml

@@ -36,7 +36,7 @@ git_sync () {
 
 clean() {
     bold "coreml clean"
-    rm -tf ${third_party}/coremltools
+    rm -rf ${third_party}/coremltools
 }
 
 sync() {