Parcourir la source

Update keras-operator.json

Lutz Roeder il y a 8 ans
Parent
commit
1cd9ed45d1
1 fichiers modifiés avec 85 ajouts et 1 suppressions
  1. 85 1
      src/keras-operator.json

+ 85 - 1
src/keras-operator.json

@@ -2,6 +2,7 @@
   {
     "name": "Bidirectional",
     "schema": {
+      "package": "keras.layers",
       "category": "Wrapper",
       "inputs": [
         { "name": "input" }
@@ -17,6 +18,7 @@
   {
     "name": "TimeDistributed",
     "schema": {
+      "package": "keras.layers",
       "category": "Wrapper",
       "inputs": [
         { "name": "input" }
@@ -31,6 +33,7 @@
   {
     "name": "Activation",
     "schema": {
+      "package": "keras.layers",
       "category": "Activation",
       "inputs": [
         { "name": "input" }
@@ -45,6 +48,7 @@
   {
     "name": "LeakyReLU",
     "schema": {
+      "package": "keras.layers",
       "category": "Activation",
       "inputs": [
         { "name": "input" }
@@ -59,6 +63,7 @@
   {
     "name": "ThresholdedReLU",
     "schema": {
+      "package": "keras.layers",
       "category": "Activation",
       "inputs": [
         { "name": "input" }
@@ -73,6 +78,7 @@
   {
     "name": "MaxPooling1D",
     "schema": {
+      "package": "keras.layers",
       "category": "Pool",
       "inputs": [
         { "name": "input" }
@@ -91,6 +97,7 @@
   {
     "name": "MaxPooling2D",
     "schema": {
+      "package": "keras.layers",
       "category": "Pool",
       "inputs": [
         { "name": "input" }
@@ -106,9 +113,42 @@
       ]
     }
   },
+  {
+    "name": "UpSampling1D",
+    "schema": {
+      "package": "keras.layers",
+      "category": "Layer",
+      "inputs": [
+        { "name": "input" }
+      ],
+      "outputs": [
+        { "name": "output" }
+      ],
+      "attributes": [
+        { "name": "data_format", "default": "channels_last" }
+      ]
+    }
+  },
   {
     "name": "UpSampling2D",
     "schema": {
+      "package": "keras.layers",
+      "category": "Layer",
+      "inputs": [
+        { "name": "input" }
+      ],
+      "outputs": [
+        { "name": "output" }
+      ],
+      "attributes": [
+        { "name": "data_format", "default": "channels_last" }
+      ]
+    }
+  },
+  {
+    "name": "UpSampling3D",
+    "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "inputs": [
         { "name": "input" }
@@ -124,6 +164,7 @@
   {
     "name": "GlobalMaxPooling1D",
     "schema": {
+      "package": "keras.layers",
       "category": "Pool",
       "inputs": [
         { "name": "input" }
@@ -136,6 +177,7 @@
   {
     "name": "GlobalMaxPooling2D",
     "schema": {
+      "package": "keras.layers",
       "category": "Pool",
       "inputs": [
         { "name": "input" }
@@ -151,6 +193,7 @@
   {
     "name": "GlobalAveragePooling1D",
     "schema": {
+      "package": "keras.layers",
       "category": "Pool",
       "inputs": [
         { "name": "input" }
@@ -165,6 +208,7 @@
   {
     "name": "GlobalAveragePooling2D",
     "schema": {
+      "package": "keras.layers",
       "category": "Pool",
       "inputs": [
         { "name": "input" }
@@ -180,6 +224,7 @@
   {
     "name": "AveragePooling2D",
     "schema": {
+      "package": "keras.layers",
       "category": "Pool",
       "inputs": [
         { "name": "input" }
@@ -195,6 +240,7 @@
   {
     "name": "BatchNormalization",
     "schema": {
+      "package": "keras.layers",
       "category": "Normalization",
       "inputs": [
         { "name": "input" },
@@ -221,8 +267,8 @@
   },
   {
     "name": "Dense",
-    "class": "keras.layers.Dense",
     "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "description": "Just your regular densely-connected NN layer.\n\nDense implements the operation: `output = activation(dot(input, kernel) + bias)` where activation is the element-wise activation function passed as the activation argument, kernel is a weights matrix created by the layer, and bias is a bias vector created by the layer (only applicable if use_bias is True).\n\nNote: if the input to the layer has a rank greater than 2, then it is flattened prior to the initial dot product with kernel.",
       "inputs": [
@@ -250,6 +296,7 @@
   {
     "name": "LSTM",
     "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "inputs": [
         { "name": "input" },
@@ -282,6 +329,7 @@
   {
     "name": "GRU",
     "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "inputs": [
         { "name": "input" },
@@ -312,6 +360,7 @@
   {
     "name": "SimpleRNN",
     "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "inputs": [
         { "name": "input" },
@@ -341,6 +390,7 @@
   {
     "name": "RNN",
     "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "outputs": [
         { "name": "output" }
@@ -357,6 +407,7 @@
   {
     "name": "Conv1D",
     "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "inputs": [
         { "name": "input" },
@@ -381,6 +432,7 @@
   {
     "name": "Conv2D",
     "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "inputs": [
         { "name": "input" },
@@ -405,6 +457,7 @@
   {
     "name": "SeparableConv2D",
     "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "inputs": [
         { "name": "input" },
@@ -432,6 +485,7 @@
   {
     "name": "Convolution2D",
     "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "inputs": [
         { "name": "input" },
@@ -457,6 +511,7 @@
   {
     "name": "DepthwiseConv2D",
     "schema": {
+      "package": "keras.layers",
       "category": "Layer",
       "inputs": [
         { "name": "input" },
@@ -481,6 +536,7 @@
   {
     "name": "Concatenate",
     "schema": {
+      "package": "keras.layers",
       "category": "Tensor",
       "inputs": [
         { "name": "inputs", "option": "variadic" }
@@ -493,6 +549,7 @@
   {
     "name": "Merge",
     "schema": {
+      "package": "keras.layers",
       "category": "Tensor",
       "inputs": [
         { "name": "inputs", "option": "variadic" }
@@ -511,6 +568,7 @@
   {
     "name": "Average",
     "schema": {
+      "package": "keras.layers",
       "category": "Tensor",
       "inputs": [
         { "name": "inputs", "option": "variadic" }
@@ -523,6 +581,7 @@
   {
     "name": "Flatten",
     "schema": {
+      "package": "keras.layers",
       "category": "Shape",
       "inputs": [
         { "name": "input" }
@@ -535,6 +594,20 @@
   {
     "name": "Reshape",
     "schema": {
+      "package": "keras.layers",
+      "category": "Shape",
+      "inputs": [
+        { "name": "input" }
+      ],
+      "outputs": [
+        { "name": "output" }
+      ]
+    }
+  },
+  {
+    "name": "Permute",
+    "schema": {
+      "package": "keras.layers",
       "category": "Shape",
       "inputs": [
         { "name": "input" }
@@ -547,6 +620,7 @@
   {
     "name": "RepeatVector",
     "schema": {
+      "package": "keras.layers",
       "category": "Shape",
       "inputs": [
         { "name": "input" }
@@ -559,6 +633,7 @@
   {
     "name": "Dropout",
     "schema": {
+      "package": "keras.layers",
       "category": "Dropout",
       "inputs": [
         { "name": "input" }
@@ -571,6 +646,7 @@
   {
     "name": "Embedding",
     "schema": {
+      "package": "keras.layers",
       "category": "Transform",
       "inputs": [
         { "name": "input" },
@@ -588,6 +664,7 @@
   {
     "name": "Add",
     "schema": {
+      "package": "keras.layers",
       "inputs": [
         { "name": "x" },
         { "name": "y" }
@@ -600,6 +677,7 @@
   {
     "name": "Sub",
     "schema": {
+      "package": "keras.layers",
       "inputs": [
         { "name": "x" },
         { "name": "y" }
@@ -608,5 +686,11 @@
         { "name": "output" }
       ]
     }
+  },
+  {
+    "name": "Lambda",
+    "schema": {
+      "package": "keras.layers"
+    }
   }
 ]