| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497 |
- [
- {
- "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"
- }
- ]
|