Browse Source

Update caffe2-proto.js

Lutz Roeder 5 years ago
parent
commit
58d414a8ab
1 changed files with 2 additions and 1 deletions
  1. 2 1
      source/caffe2-proto.js

+ 2 - 1
source/caffe2-proto.js

@@ -224,7 +224,8 @@ $root.caffe2.TensorProto.DataType = {
     "INT64": 10,
     "FLOAT16": 12,
     "DOUBLE": 13,
-    "ZERO_COLLISION_HASH": 14
+    "ZERO_COLLISION_HASH": 14,
+    "REBATCHING_BUFFER": 15
 };
 
 $root.caffe2.TensorProto.StorageType = {