TorchScript 1.3 prototype (#281)

lutzroeder · Sep 2, 2019 · 041ed91 · 041ed91
1 parent 3c8ea76
commit 041ed91
Show file tree

Hide file tree

Showing 4 changed files with 379 additions and 174 deletions.
diff --git a/src/mlnet.js b/src/mlnet.js
@@ -272,7 +272,7 @@ mlnet.TensorType = class {
 
     constructor(codec) {
 
-        mlnet.TensorType._map = mlnet.TensorType.map || new Map([ 
+        mlnet.TensorType._map = mlnet.TensorType._map || new Map([ 
             [ 'Boolean', 'boolean' ],
             [ 'Single', 'float32' ],
             [ 'Double', 'float64' ],

diff --git a/src/torchscript-metadata.json b/src/torchscript-metadata.json
@@ -689,5 +689,21 @@
         { "name": "output" }
       ]
     }
+  },
+  {
+    "name": "flatten",
+    "schema": {
+      "category": "Shape",
+      "attributes": [
+        { "name": "start_dim", "type": "int64" },
+        { "name": "end_dim", "type": "int64" }
+      ],
+      "inputs": [
+        { "name": "inputs" }
+      ],
+      "outputs": [
+        { "name": "output" }
+      ]
+    }
   }
 ]