Fix PyTorch long storage (#133)

lutzroeder · Nov 28, 2018 · 0391658 · 0391658
1 parent ecef1e3
commit 0391658
Show file tree

Hide file tree

Showing 3 changed files with 185 additions and 41 deletions.
diff --git a/src/pytorch-metadata.json b/src/pytorch-metadata.json
@@ -162,6 +162,7 @@
       "attributes": [
         {
           "default": false,
+          "visible": false,
           "name": "inplace"
         },
         {
@@ -367,6 +368,29 @@
           "name": "track_running_stats"
         }
       ],
+      "inputs": [
+        {
+          "name": "input"
+        },
+        {
+          "name": "weight"
+        },
+        {
+          "name": "bias"
+        },
+        {
+          "name": "running_mean",
+          "visible": false
+        },
+        {
+          "name": "running_var",
+          "visible": false
+        },
+        {
+          "name": "num_batches_tracked",
+          "visible": false
+        }
+      ],
       "category": "Normalization",
       "package": "torch.nn.modules.batchnorm"
     }
@@ -448,6 +472,43 @@
           "visible": false
         }
       ],
+      "inputs": [
+        {
+          "name": "input"
+        },
+        {
+          "name": "weight_ih_l0",
+          "visible": false
+        },
+        {
+          "name": "weight_hh_l0",
+          "visible": false
+        },
+        {
+          "name": "bias_ih_l0",
+          "visible": false
+        },
+        {
+          "name": "bias_hh_l0",
+          "visible": false
+        },
+        {
+          "name": "weight_ih_l1",
+          "visible": false
+        },
+        {
+          "name": "weight_hh_l1",
+          "visible": false
+        },
+        {
+          "name": "bias_ih_l1",
+          "visible": false
+        },
+        {
+          "name": "bias_hh_l1",
+          "visible": false
+        }
+      ],
       "category": "Layer",
       "package": "torch.nn.modules.rnn"
     }