Browse Source

Update torch-metadata.json

Lutz Roeder 5 năm trước cách đây
mục cha
commit
744196a5af
1 tập tin đã thay đổi với 24 bổ sung0 xóa
  1. 24 0
      source/torch-metadata.json

+ 24 - 0
source/torch-metadata.json

@@ -559,5 +559,29 @@
     "name": "nn.MM",
     "schema": {
     }
+  },
+  {
+    "name": "cudnn.BLSTM",
+    "schema": {
+      "category": "Layer"
+    }
+  },
+  {
+    "name": "cudnn.RNN",
+    "schema": {
+      "category": "Layer"
+    }
+  },
+  {
+    "name": "cudnn.BatchBRNNReLU",
+    "schema": {
+      "category": "Normalization"
+    }
+  },
+  {
+    "name": "nn.Transpose",
+    "schema": {
+      "category": "Transform"
+    }
   }
 ]