7254f7b4d1
Build / Build (macos-latest) (push) Has been cancelled
Build / Build (ubuntu-latest) (push) Has been cancelled
Build / Build (windows-latest) (push) Has been cancelled
Build / Analyze (javascript) (push) Has been cancelled
Build / Analyze (python) (push) Has been cancelled
497 lines
16 KiB
JSON
497 lines
16 KiB
JSON
[
|
|
{
|
|
"name": "activation",
|
|
"category": "Activation",
|
|
"description": "Applies specified type of activation function to input."
|
|
},
|
|
{
|
|
"name": "add",
|
|
"description": "A layer that performs elementwise addition.",
|
|
"inputs": [
|
|
{ "name": "x" },
|
|
{ "name": "y" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "z" }
|
|
]
|
|
},
|
|
{
|
|
"name": "average",
|
|
"description": "A layer that computes the elementwise average of the inputs."
|
|
},
|
|
{
|
|
"name": "batchnorm",
|
|
"category": "Normalization",
|
|
"description": "A layer that performs batch normalization, which is performed along the channel axis, and repeated along the other axes, if present.",
|
|
"attributes": [
|
|
{ "name": "epsilon", "default": 0.000009999999747378752 },
|
|
{ "name": "computeMeanVar", "visible": false },
|
|
{ "name": "instanceNormalization", "visible": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "bias",
|
|
"category": "Layer",
|
|
"description": "A layer that performs elementwise addition of a bias, which is broadcasted to match the input shape."
|
|
},
|
|
{
|
|
"name": "biDirectionalLSTM",
|
|
"category": "Layer",
|
|
"description": "Bidirectional long short-term memory (LSTM) layer. The first LSTM operates on the input sequence in the forward direction. The second LSTM operates on the input sequence in the reverse direction.",
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "h" },
|
|
{ "name": "c" },
|
|
{ "name": "h_rev" },
|
|
{ "name": "c_rev" },
|
|
{ "name": "inputGateWeightMatrix", "visible": false },
|
|
{ "name": "forgetGateWeightMatrix", "visible": false },
|
|
{ "name": "blockInputWeightMatrix", "visible": false },
|
|
{ "name": "outputGateWeightMatrix", "visible": false },
|
|
{ "name": "inputGateRecursionMatrix", "visible": false },
|
|
{ "name": "forgetGateRecursionMatrix", "visible": false },
|
|
{ "name": "blockInputRecursionMatrix", "visible": false },
|
|
{ "name": "outputGateRecursionMatrix", "visible": false },
|
|
{ "name": "inputGateBiasVector", "visible": false },
|
|
{ "name": "forgetGateBiasVector", "visible": false },
|
|
{ "name": "blockInputBiasVector", "visible": false },
|
|
{ "name": "outputGateBiasVector", "visible": false },
|
|
{ "name": "inputGateWeightMatrix_rev", "visible": false },
|
|
{ "name": "forgetGateWeightMatrix_rev", "visible": false },
|
|
{ "name": "blockInputWeightMatrix_rev", "visible": false },
|
|
{ "name": "outputGateWeightMatrix_rev", "visible": false },
|
|
{ "name": "inputGateRecursionMatrix_rev", "visible": false },
|
|
{ "name": "forgetGateRecursionMatrix_rev", "visible": false },
|
|
{ "name": "blockInputRecursionMatrix_rev", "visible": false },
|
|
{ "name": "outputGateRecursionMatrix_rev", "visible": false },
|
|
{ "name": "inputGateBiasVector_rev", "visible": false },
|
|
{ "name": "forgetGateBiasVector_rev", "visible": false },
|
|
{ "name": "blockInputBiasVector_rev", "visible": false },
|
|
{ "name": "outputGateBiasVector_rev", "visible": false }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" },
|
|
{ "name": "h" },
|
|
{ "name": "c" },
|
|
{ "name": "h_rev" },
|
|
{ "name": "c_rev" }
|
|
]
|
|
},
|
|
{
|
|
"name": "concat",
|
|
"category": "Tensor",
|
|
"description": "A layer that concatenates along the channel axis (default) or sequence axis.",
|
|
"inputs": [
|
|
{ "name": "inputs", "type": "Tensor[]" }
|
|
]
|
|
},
|
|
{
|
|
"name": "convolution",
|
|
"category": "Layer",
|
|
"description": "A layer that performs spatial convolution or deconvolution.",
|
|
"attributes": [
|
|
{ "name": "outputShape", "type": "uint64[]", "description": "Either None or a 2-tuple, specifying the output shape (output_height, output_width). Used only when is_deconv == True. When is_deconv == False, this parameter is ignored. If it is None, the output shape is calculated automatically using the border_mode. Kindly refer to NeuralNetwork.proto for details.", "visible": false },
|
|
{ "name": "outputChannels", "type": "uint64", "description": "The number of kernels. Same as ``C_out`` used in the layer description.", "visible": false },
|
|
{ "name": "kernelChannels", "type": "uint64", "description": "Channel dimension of the kernels. Must be equal to ``inputChannels / nGroups``, if isDeconvolution == False. Must be equal to ``inputChannels``, if isDeconvolution == True.", "visible": false },
|
|
{ "name": "nGroups", "type": "uint64", "description": "Group convolution, i.e. weight reuse along channel axis. Input and kernels are divided into g groups and convolution / deconvolution is applied within the groups independently. If not set or 0, it is set to the default value 1.", "default": 1 },
|
|
{ "name": "isDeconvolution", "type": "boolean", "description": "Flag to specify whether it is a deconvolution layer." },
|
|
{ "name": "valid", "type": "ValidPadding", "visible": false },
|
|
{ "name": "same", "type": "SamePadding", "visible": false },
|
|
{ "name": "dilationFactor", "type": "uint64[]", "default": [ 1, 1 ] },
|
|
{ "name": "stride", "type": "uint64[]", "default": [ 1, 1 ] },
|
|
{ "name": "kernelSize", "type": "uint64[]", "default": [ 3, 3 ] },
|
|
{ "name": "hasBias", "type": "boolean", "description": "Flag to specify whether a bias is to be added or not.", "visible": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "crop",
|
|
"category": "Data",
|
|
"description": "A layer that crops the spatial dimensions of an input. If two inputs are provided, the shape of the second input is used as the reference shape.",
|
|
"inputs": [
|
|
{ "name": "x1" },
|
|
{ "name": "x2" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "y" }
|
|
]
|
|
},
|
|
{
|
|
"name": "dot",
|
|
"description": "If true, inputs are normalized first, thereby computing the cosine similarity."
|
|
},
|
|
{
|
|
"name": "embedding",
|
|
"category": "Transform",
|
|
"description": "A layer that performs a matrix lookup and optionally adds a bias."
|
|
},
|
|
{
|
|
"name": "featureVectorizer",
|
|
"inputs": [
|
|
{ "name": "inputs", "type": "Tensor[]" }
|
|
]
|
|
},
|
|
{
|
|
"name": "flatten",
|
|
"category": "Shape",
|
|
"description": "A layer that flattens the input.",
|
|
"attributes": [
|
|
{ "name": "mode", "type": "FlattenLayerParams.FlattenOrder" }
|
|
]
|
|
},
|
|
{
|
|
"name": "gather",
|
|
"category": "Transform",
|
|
"description": "Gather layer that gathers elements from the first input, along a specified axis, at indices specified in the second input.",
|
|
"inputs": [
|
|
{ "name": "input", "type": "Tensor" },
|
|
{ "name": "indices", "type": "Tensor" }
|
|
]
|
|
},
|
|
{
|
|
"name": "gelu",
|
|
"category": "Activation",
|
|
"description": "Gaussian error linear unit activation.",
|
|
"attributes": [
|
|
{ "name": "mode", "type": "GeluLayerParams.GeluMode" }
|
|
]
|
|
},
|
|
{
|
|
"name": "gru",
|
|
"category": "Layer",
|
|
"description": "Gated-Recurrent Unit (GRU) Layer",
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "h" },
|
|
{ "name": "updateGateWeightMatrix", "visible": false },
|
|
{ "name": "resetGateWeightMatrix", "visible": false },
|
|
{ "name": "outputGateWeightMatrix", "visible": false },
|
|
{ "name": "updateGateRecursionMatrix", "visible": false },
|
|
{ "name": "resetGateRecursionMatrix", "visible": false },
|
|
{ "name": "outputGateRecursionMatrix", "visible": false },
|
|
{ "name": "updateGateBiasVector", "visible": false },
|
|
{ "name": "resetGateBiasVector", "visible": false },
|
|
{ "name": "outputGateBiasVector", "visible": false }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" },
|
|
{ "name": "h" }
|
|
]
|
|
},
|
|
{
|
|
"name": "innerProduct",
|
|
"category": "Layer",
|
|
"description": "A layer that performs a matrix vector product. This is equivalent to a fully-connected, or dense layer.",
|
|
"attributes": [
|
|
{ "name": "inputChannels", "type": "uint64", "visible": false },
|
|
{ "name": "outputChannels", "type": "uint64", "visible": false },
|
|
{ "name": "hasBias", "type": "boolean", "visible": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "int64ClassLabels",
|
|
"category": "Data",
|
|
"outputs": [
|
|
{ "name": "probabilities" },
|
|
{ "name": "feature" }
|
|
]
|
|
},
|
|
{
|
|
"name": "itemSimilarityRecommender",
|
|
"inputs": [
|
|
{ "name": "item" },
|
|
{ "name": "numRecommendations" },
|
|
{ "name": "itemRestriction" },
|
|
{ "name": "itemExclusion" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "recommendedItemList" },
|
|
{ "name": "recommendedItemScore" }
|
|
]
|
|
},
|
|
{
|
|
"name": "l2normalize",
|
|
"category": "Normalization",
|
|
"description": "A layer that performs L2 normalization, i.e. divides by the the square root of the sum of squares of all elements of input."
|
|
},
|
|
{
|
|
"name": "loadConstant",
|
|
"category": "Data"
|
|
},
|
|
{
|
|
"name": "lrn",
|
|
"category": "Normalization",
|
|
"description": "A layer that performs local response normalization (LRN).",
|
|
"attributes": [
|
|
{ "name": "k", "default": 1 }
|
|
]
|
|
},
|
|
{
|
|
"name": "max",
|
|
"description": "A layer that computes the elementwise maximum over the inputs."
|
|
},
|
|
{
|
|
"name": "min",
|
|
"description": "A layer that computes the elementwise minimum over the inputs."
|
|
},
|
|
{
|
|
"name": "multiply",
|
|
"description": "A layer that performs elementwise multiplication.",
|
|
"inputs": [
|
|
{ "name": "x" },
|
|
{ "name": "y" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "z" }
|
|
]
|
|
},
|
|
{
|
|
"name": "mvn",
|
|
"category": "Normalization",
|
|
"description": "A layer that performs mean variance normalization, along axis = -3."
|
|
},
|
|
{
|
|
"name": "nonMaximumSuppression",
|
|
"attributes": [
|
|
{ "name": "iouThreshold" },
|
|
{ "name": "confidenceThreshold" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "confidence" },
|
|
{ "name": "coordinates" },
|
|
{ "name": "iouThreshold" },
|
|
{ "name": "confidenceThreshold" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "confidence" },
|
|
{ "name": "coordinates" }
|
|
]
|
|
},
|
|
{
|
|
"name": "padding",
|
|
"category": "Shape",
|
|
"description": "Fill a constant value in the padded region.",
|
|
"attributes": [
|
|
{ "name": "paddingAmounts", "visible": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "permute",
|
|
"category": "Shape",
|
|
"description": "A layer that rearranges the dimensions and data of an input."
|
|
},
|
|
{
|
|
"name": "pooling",
|
|
"category": "Pool",
|
|
"description": "Spatial Pooling layer to reduce dimensions of input using the specified kernel size and type.",
|
|
"attributes": [
|
|
{ "name": "includeLastPixel", "type": "ValidCompletePadding", "visible": false },
|
|
{ "name": "same", "type": "SamePadding", "visible": false },
|
|
{ "name": "valid", "type": "ValidCompletePadding", "visible": false },
|
|
{ "name": "type", "type": "PoolingLayerParams.PoolingType" },
|
|
{ "name": "globalPooling", "type": "boolean", "default": false },
|
|
{ "name": "stride", "type": "uint64", "default": [ 1, 1 ] },
|
|
{ "name": "kernelSize", "type": "uint64[]", "default": [ 3, 3 ] },
|
|
{ "name": "avgPoolExcludePadding", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "reduce",
|
|
"description": "A layer that reduces the input using a specified operation."
|
|
},
|
|
{
|
|
"name": "reorganizeData",
|
|
"category": "Shape",
|
|
"description": "A layer that reorganizes data in the input in: 1. SPACE_TO_DEPTH, 2. DEPTH_TO_SPACE."
|
|
},
|
|
{
|
|
"name": "reshape",
|
|
"category": "Shape",
|
|
"description": "A layer that recasts the input into a new shape."
|
|
},
|
|
{
|
|
"name": "scale",
|
|
"category": "Layer",
|
|
"description": "A layer that performs elmentwise multiplication by a scale factor and optionally adds a bias.",
|
|
"attributes": [
|
|
{ "name": "hasBias", "type": "boolean", "visible": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "scaler",
|
|
"category": "Data"
|
|
},
|
|
{
|
|
"name": "sequenceRepeat",
|
|
"category": "Shape",
|
|
"description": "A layer that repeats a sequence."
|
|
},
|
|
{
|
|
"name": "slice",
|
|
"description": "A layer that uniformly splits across the channel dimension to produce a specified number of outputs."
|
|
},
|
|
{
|
|
"name": "softmax",
|
|
"category": "Activation",
|
|
"description": "A layer that performs softmax normalization. Normalization is done along the channel axis."
|
|
},
|
|
{
|
|
"name": "softmaxND",
|
|
"category": "Activation",
|
|
"description": "A layer that performs softmax normalization along a specified axis."
|
|
},
|
|
{
|
|
"name": "squeeze",
|
|
"category": "Transform"
|
|
},
|
|
{
|
|
"name": "stringClassLabels",
|
|
"category": "Data",
|
|
"outputs": [
|
|
{ "name": "probabilities" },
|
|
{ "name": "feature" }
|
|
]
|
|
},
|
|
{
|
|
"name": "textClassifier",
|
|
"attributes": [
|
|
{ "name": "revision", "visible": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "unary",
|
|
"description": "A layer that applies a unary function.",
|
|
"attributes": [
|
|
{ "name": "type", "type": "UnaryFunctionLayerParams.Operation" },
|
|
{ "name": "alpha", "default": 1 },
|
|
{ "name": "scale", "default": 1 },
|
|
{ "name": "epsilon", "default": 9.999999974752427e-7 }
|
|
],
|
|
"inputs": [
|
|
{ "name": "x" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "z" }
|
|
]
|
|
},
|
|
{
|
|
"name": "uniDirectionalLSTM",
|
|
"category": "Layer",
|
|
"description": "A unidirectional long short-term memory (LSTM) layer.",
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "h" },
|
|
{ "name": "c" },
|
|
{ "name": "inputGateWeightMatrix", "visible": false },
|
|
{ "name": "forgetGateWeightMatrix", "visible": false },
|
|
{ "name": "blockInputWeightMatrix", "visible": false },
|
|
{ "name": "outputGateWeightMatrix", "visible": false },
|
|
{ "name": "inputGateRecursionMatrix", "visible": false },
|
|
{ "name": "forgetGateRecursionMatrix", "visible": false },
|
|
{ "name": "blockInputRecursionMatrix", "visible": false },
|
|
{ "name": "outputGateRecursionMatrix", "visible": false },
|
|
{ "name": "inputGateBiasVector", "visible": false },
|
|
{ "name": "forgetGateBiasVector", "visible": false },
|
|
{ "name": "blockInputBiasVector", "visible": false },
|
|
{ "name": "outputGateBiasVector", "visible": false }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" },
|
|
{ "name": "h" },
|
|
{ "name": "c" }
|
|
]
|
|
},
|
|
{
|
|
"name": "upsample",
|
|
"category": "Data",
|
|
"description": "A layer that scales up spatial dimensions. It supports two modes: nearest neighbour (default) and bilinear."
|
|
},
|
|
{
|
|
"name": "transpose",
|
|
"category": "Transform"
|
|
},
|
|
{
|
|
"name": "wordTagger",
|
|
"attributes": [
|
|
{ "name": "revision", "visible": false }
|
|
],
|
|
"outputs": [
|
|
{ "name": "tokens" },
|
|
{ "name": "tags" },
|
|
{ "name": "locations" },
|
|
{ "name": "lengths" }
|
|
]
|
|
},
|
|
{
|
|
"name": "program:conv",
|
|
"category": "Layer",
|
|
"inputs": [
|
|
{ "name": "x" },
|
|
{ "name": "weight" },
|
|
{ "name": "bias" }
|
|
]
|
|
},
|
|
{
|
|
"name": "program:batch_norm",
|
|
"category": "Normalization",
|
|
"inputs": [
|
|
{ "name": "x" },
|
|
{ "name": "mean" },
|
|
{ "name": "variance" },
|
|
{ "name": "gamma" },
|
|
{ "name": "beta" }
|
|
]
|
|
},
|
|
{
|
|
"name": "program:linear",
|
|
"category": "Layer",
|
|
"inputs": [
|
|
{ "name": "x" },
|
|
{ "name": "weight" },
|
|
{ "name": "bias" }
|
|
]
|
|
},
|
|
{
|
|
"name": "program:pad",
|
|
"category": "Tensor"
|
|
},
|
|
{
|
|
"name": "program:transpose",
|
|
"category": "Transform"
|
|
},
|
|
{
|
|
"name": "program:sigmoid",
|
|
"category": "Activation"
|
|
},
|
|
{
|
|
"name": "program:softmax",
|
|
"category": "Activation"
|
|
},
|
|
{
|
|
"name": "program:relu",
|
|
"category": "Activation"
|
|
},
|
|
{
|
|
"name": "program:relu6",
|
|
"category": "Activation"
|
|
},
|
|
{
|
|
"name": "program:reshape",
|
|
"category": "Shape"
|
|
},
|
|
{
|
|
"name": "program:concat",
|
|
"category": "Tensor"
|
|
},
|
|
{
|
|
"name": "program:layer_norm",
|
|
"category": "Normalization"
|
|
},
|
|
{
|
|
"name": "program:max_pool",
|
|
"category": "Pool"
|
|
},
|
|
{
|
|
"name": "program:gather",
|
|
"category": "Transform"
|
|
}
|
|
] |