Update caffe.js

lutzroeder · Aug 18, 2018 · 4bb36e0 · 4bb36e0
1 parent baff93a
commit 4bb36e0
Showing 1 changed file with 49 additions and 0 deletions.
diff --git a/src/caffe.js b/src/caffe.js
@@ -14697,6 +14697,7 @@
              * @property {number|null} [strideW] PoolingParameter strideW
              * @property {caffe.PoolingParameter.Engine|null} [engine] PoolingParameter engine
              * @property {boolean|null} [globalPooling] PoolingParameter globalPooling
+             * @property {caffe.PoolingParameter.RoundMode|null} [roundMode] PoolingParameter roundMode
              */
 
             /**
@@ -14810,6 +14811,14 @@
              */
             PoolingParameter.prototype.globalPooling = false;
 
+            /**
+             * PoolingParameter roundMode.
+             * @member {caffe.PoolingParameter.RoundMode} roundMode
+             * @memberof caffe.PoolingParameter
+             * @instance
+             */
+            PoolingParameter.prototype.roundMode = 0;
+
             /**
              * Creates a new PoolingParameter instance using the specified properties.
              * @function create
@@ -14858,6 +14867,8 @@
                     writer.uint32(/* id 11, wireType 0 =*/88).int32(message.engine);
                 if (message.globalPooling != null && message.hasOwnProperty("globalPooling"))
                     writer.uint32(/* id 12, wireType 0 =*/96).bool(message.globalPooling);
+                if (message.roundMode != null && message.hasOwnProperty("roundMode"))
+                    writer.uint32(/* id 13, wireType 0 =*/104).int32(message.roundMode);
                 return writer;
             };
 
@@ -14928,6 +14939,9 @@
                     case 12:
                         message.globalPooling = reader.bool();
                         break;
+                    case 13:
+                        message.roundMode = reader.int32();
+                        break;
                     default:
                         reader.skipType(tag & 7);
                         break;
@@ -15011,6 +15025,14 @@
                 if (message.globalPooling != null && message.hasOwnProperty("globalPooling"))
                     if (typeof message.globalPooling !== "boolean")
                         return "globalPooling: boolean expected";
+                if (message.roundMode != null && message.hasOwnProperty("roundMode"))
+                    switch (message.roundMode) {
+                    default:
+                        return "roundMode: enum value expected";
+                    case 0:
+                    case 1:
+                        break;
+                    }
                 return null;
             };
 
@@ -15074,6 +15096,16 @@
                 }
                 if (object.globalPooling != null)
                     message.globalPooling = Boolean(object.globalPooling);
+                switch (object.roundMode) {
+                case "CEIL":
+                case 0:
+                    message.roundMode = 0;
+                    break;
+                case "FLOOR":
+                case 1:
+                    message.roundMode = 1;
+                    break;
+                }
                 return message;
             };
 
@@ -15103,6 +15135,7 @@
                     object.padW = 0;
                     object.engine = options.enums === String ? "DEFAULT" : 0;
                     object.globalPooling = false;
+                    object.roundMode = options.enums === String ? "CEIL" : 0;
                 }
                 if (message.pool != null && message.hasOwnProperty("pool"))
                     object.pool = options.enums === String ? $root.caffe.PoolingParameter.PoolMethod[message.pool] : message.pool;
@@ -15128,6 +15161,8 @@
                     object.engine = options.enums === String ? $root.caffe.PoolingParameter.Engine[message.engine] : message.engine;
                 if (message.globalPooling != null && message.hasOwnProperty("globalPooling"))
                     object.globalPooling = message.globalPooling;
+                if (message.roundMode != null && message.hasOwnProperty("roundMode"))
+                    object.roundMode = options.enums === String ? $root.caffe.PoolingParameter.RoundMode[message.roundMode] : message.roundMode;
                 return object;
             };
 
@@ -15174,6 +15209,20 @@
                 return values;
             })();
 
+            /**
+             * RoundMode enum.
+             * @name caffe.PoolingParameter.RoundMode
+             * @enum {string}
+             * @property {number} CEIL=0 CEIL value
+             * @property {number} FLOOR=1 FLOOR value
+             */
+            PoolingParameter.RoundMode = (function() {
+                var valuesById = {}, values = Object.create(valuesById);
+                values[valuesById[0] = "CEIL"] = 0;
+                values[valuesById[1] = "FLOOR"] = 1;
+                return values;
+            })();
+
             return PoolingParameter;
         })();