|
|
@@ -22,6 +22,29 @@
|
|
|
]
|
|
|
}
|
|
|
},
|
|
|
+ {
|
|
|
+ "name": "Deconvolution",
|
|
|
+ "schema": {
|
|
|
+ "category": "Layer",
|
|
|
+ "inputs": [
|
|
|
+ { "name": "input" },
|
|
|
+ { "name": "filter" },
|
|
|
+ { "name": "bias" }
|
|
|
+ ],
|
|
|
+ "ouptuts": [
|
|
|
+ { "name": "output" }
|
|
|
+ ],
|
|
|
+ "attributes": [
|
|
|
+ { "name": "bias_term", "hidden": true },
|
|
|
+ { "name": "weight_filler", "hidden": true },
|
|
|
+ { "name": "bias_filler", "hidden": true },
|
|
|
+ { "name": "pad", "default": [] },
|
|
|
+ { "name": "kernel_size", "default": [] },
|
|
|
+ { "name": "stride", "default": [] },
|
|
|
+ { "name": "dilation", "default": [] }
|
|
|
+ ]
|
|
|
+ }
|
|
|
+ },
|
|
|
{
|
|
|
"name": "InnerProduct",
|
|
|
"schema": {
|
|
|
@@ -101,6 +124,16 @@
|
|
|
]
|
|
|
}
|
|
|
},
|
|
|
+ {
|
|
|
+ "name": "SoftmaxWithLoss",
|
|
|
+ "schema": {
|
|
|
+ "category": "Activation",
|
|
|
+ "inputs": [
|
|
|
+ { "name": "input" },
|
|
|
+ { "name": "labels" }
|
|
|
+ ]
|
|
|
+ }
|
|
|
+ },
|
|
|
{
|
|
|
"name": "ReLU",
|
|
|
"schema": {
|
|
|
@@ -139,6 +172,12 @@
|
|
|
"category": "Pool"
|
|
|
}
|
|
|
},
|
|
|
+ {
|
|
|
+ "name": "Crop",
|
|
|
+ "schema": {
|
|
|
+ "category": "Shape"
|
|
|
+ }
|
|
|
+ },
|
|
|
{
|
|
|
"name": "Data",
|
|
|
"schema": {
|
|
|
@@ -156,5 +195,13 @@
|
|
|
{ "name": "outputs", "option": "variadic" }
|
|
|
]
|
|
|
}
|
|
|
+ },
|
|
|
+ {
|
|
|
+ "name": "Python",
|
|
|
+ "schema": {
|
|
|
+ "outputs": [
|
|
|
+ { "name": "outputs", "option": "variadic" }
|
|
|
+ ]
|
|
|
+ }
|
|
|
}
|
|
|
]
|