[ { "name": "InputLayer", "schema": { "bindings": [ { "name": "layerBindingId", "type": "int", "src": "layerBindingId" } ] } }, { "name": "OutputLayer", "schema": { "category": "Tensor", "bindings": [ { "name": "layerBindingId", "type": "int", "src": "layerBindingId" } ] } }, { "name": "Pooling2dLayer", "schema": { "category": "Pool", "attributes": [ { "name": "type", "type": "string", "src": "poolType", "src_type": "PoolingAlgorithm"}, { "name": "padding", "type": "string", "src": ["padTop", "padRight", "padBottom", "padLeft"] }, { "name": "width", "type": "string", "src": "poolWidth" }, { "name": "height", "type": "string", "src": "poolHeight" }, { "name": "stride", "type": "string", "src": ["strideX", "strideY"] }, { "name": "outputShapeRounding", "type": "string", "src": "outputShapeRounding", "src_type": "OutputShapeRounding"}, { "name": "paddingMethod", "type": "string", "src": "paddingMethod", "src_type": "PaddingMethod"}, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "ReshapeLayer", "schema": { "category": "Shape", "attributes": [ { "name": "targetShape", "type": "string", "src": "targetShape" } ] } }, { "name": "SoftmaxLayer", "schema": { "category": "Activation", "attributes": [ { "name": "beta", "type": "float", "src": "beta" } ] } }, { "name": "Convolution2dLayer", "schema": { "category": "Layer", "inputs": [ { "name": "weight", "src": "weights" }, { "name": "bias", "src": "biases" } ], "attributes": [ { "name": "padding", "type": "string", "src": ["padTop", "padRight", "padBottom", "padLeft"] }, { "name": "stride", "type": "string", "src": ["strideX", "strideY"] }, { "name": "dilation", "type": "string", "src": ["dilationX", "dilationY"] }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "DepthwiseConvolution2dLayer", "schema": { "category": "Layer", "inputs": [ { "name": "weight", "src": "weights" }, { "name": "bias", "src": "biases" } ], "attributes": [ { "name": "padding", "type": "string", "src": ["padTop", "padRight", "padBottom", "padLeft"] }, { "name": "stride", "type": "string", "src": ["strideX", "strideY"] }, { "name": "dilation", "type": "string", "src": ["dilationX", "dilationY"] }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "ActivationLayer", "schema": { "category": "Activation", "attributes": [ { "name": "function", "type": "string", "src": "activationFunction", "src_type": "ActivationFunction" }, { "name": "a", "type": "float", "src": "a" }, { "name": "b", "type": "float", "src": "b" } ] } }, { "name": "PermuteLayer", "schema": { "category": "Shape", "attributes": [ { "name": "dimMappings", "type": "string", "src": "dimMappings" } ] } }, { "name": "FullyConnectedLayer", "schema": { "category": "Layer", "inputs": [ { "name": "weights", "src": "weights" }, { "name": "biases", "src": "biases" } ], "attributes": [ { "name": "transposeWeightsMatrix", "type": "bool", "src": "transposeWeightsMatrix" } ] } }, { "name": "ConstantLayer", "schema": { "category": "Constant", "inputs": [ { "name": "input", "src": "input" } ] } }, { "name": "SpaceToBatchNdLayer", "schema": { "category": "Layer", "attributes": [ { "name": "blockShape", "type": "string", "src": "blockShape" }, { "name": "padList", "type": "string", "src": "padList" }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "BatchToSpaceNdLayer", "schema": { "category": "Layer", "attributes": [ { "name": "blockShape", "type": "string", "src": "blockShape" }, { "name": "crops", "type": "string", "src": "crops" }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "DivisionLayer", "schema": { "category": "Layer" } }, { "name": "MinimumLayer", "schema": { "category": "Layer" } }, { "name": "EqualLayer", "schema": { "category": "Layer" } }, { "name": "MaximumLayer", "schema": { "category": "Layer" } }, { "name": "NormalizationLayer", "schema": { "category": "Normalization", "attributes": [ { "name": "normChannelType", "type": "string", "src": "normChannelType", "src_type": "NormalizationAlgorithmChannel" }, { "name": "normMethodType", "type": "string", "src": "normMethodType", "src_type": "NormalizationAlgorithmMethod" }, { "name": "normSize", "type": "uint", "src": "normSize" }, { "name": "alpha", "type": "float", "src": "alpha" }, { "name": "beta", "type": "float", "src": "beta" }, { "name": "k", "type": "float", "src": "k" }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "PadLayer", "schema": { "category": "Layer", "attributes": [ { "name": "padList", "type": "uint", "src": "padList" }, { "name": "padValue", "type": "float", "src": "padValue" } ] } }, { "name": "RsqrtLayer", "schema": { "category": "Layer" } }, { "name": "FloorLayer", "schema": { "category": "Layer" } }, { "name": "BatchNormalizationLayer", "schema": { "category": "Normalization", "inputs": [ { "name": "mean", "src": "mean" }, { "name": "variance", "src": "variance" }, { "name": "beta", "src": "beta" }, { "name": "gamma", "src": "gamma" } ], "attributes": [ { "name": "eps", "type": "float", "src": "eps" }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "GreaterLayer", "schema": { "category": "Layer", "attributes": [ ] } }, { "name": "ResizeBilinearLayer", "schema": { "category": "Layer", "attributes": [ { "name": "targetWidth", "type": "uint", "src": "targetWidth" }, { "name": "targetHeight", "type": "uint", "src": "targetHeight" }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "SubtractionLayer", "schema": { } }, { "name": "StridedSliceLayer", "schema": { "category": "Tensor", "attributes": [ { "name": "begin", "type": "int", "src": "begin" }, { "name": "end", "type": "int", "src": "end" }, { "name": "stride", "type": "int", "src": "stride" }, { "name": "beginMask", "type": "int", "src": "beginMask" }, { "name": "endMask", "type": "int", "src": "endMask" }, { "name": "shrinkAxisMask", "type": "int", "src": "shrinkAxisMask" }, { "name": "ellipsisMask", "type": "int", "src": "ellipsisMask" }, { "name": "newAxisMask", "type": "int", "src": "newAxisMask" }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "GatherLayer", "schema": { "category": "Tensor" } }, { "name": "MeanLayer", "schema": { "attributes": [ { "name": "axis", "type": "uint", "src": "axis" }, { "name": "keepDims", "type": "bool", "src": "keepDims" } ] } }, { "name": "MergerLayer", "schema": { "category": "Tensor" } }, { "name": "L2NormalizationLayer", "schema": { "category": "Normalization", "attributes": [ { "name": "eps", "type": "float", "src": "eps" }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "SplitterLayer", "schema": { "category": "Tensor", "attributes": [ { "name": "concatAxis", "type": "uint", "src": "concatAxis" }, { "name": "numViews", "type": "uint", "src": "numViewes" }, { "name": "numDimensions", "type": "uint", "src": "numDimensions" } ] } }, { "name": "DetectionPostProcessLayer", "schema": { "category": "Custom", "attributes": [ { "name": "maxDetections", "type": "uint", "src": "maxDetections" }, { "name": "maxClassesPerDetection", "type": "uint", "src": "maxClassesPerDetection" }, { "name": "detectionsPerClass", "type": "uint", "src": "detectionsPerClass" }, { "name": "nmsScoreThreshold", "type": "float", "src": "nmsScoreThreshold" }, { "name": "numIouThreshold", "type": "float", "src": "nmsIouThreshold" }, { "name": "numClasses", "type": "uint", "src": "numClasses" }, { "name": "useRegularNms", "type": "bool", "src": "useRegularNms" }, { "name": "scaleX", "type": "float", "src": "scaleX" }, { "name": "scaleY", "type": "float", "src": "scaleY" }, { "name": "scaleW", "type": "float", "src": "scaleW" }, { "name": "scaleH", "type": "float", "src": "scaleH" } ] } }, { "name": "LstmLayer", "schema": { "category": "Layer", "inputs": [ { "name": "inputToForgetWeights1", "src": "inputToForgetWeights1" }, { "name": "inputToCellWeights1", "src": "inputToCellWeights1" }, { "name": "inputToOutputWeights1", "src": "inputToOutputWeights1" }, { "name": "recurrentToForgetWeights1", "src": "recurrentToForgetWeights1" }, { "name": "recurrentToCellWeights1", "src": "recurrentToCellWeights1" }, { "name": "recurrentToOutputWeights1", "src": "recurrentToOutputWeights1" }, { "name": "forgetGateBias1", "src": "forgetGateBias1" }, { "name": "cellBias1", "src": "cellBias1" }, { "name": "outputGateBias1", "src": "outputGateBias1" }, { "name": "inputToInputWeights1", "src": "inputToInputWeights1" }, { "name": "recurrentToInputWeights1", "src": "recurrentToInputWeights1" }, { "name": "cellToInputWeights1", "src": "cellToInputWeights1" }, { "name": "inputGateBias1", "src": "inputGateBias1" }, { "name": "projectionWeights1", "src": "projectionWeights1" }, { "name": "projectionBias1", "src": "projectionBias1" }, { "name": "cellToForgetWeights1", "src": "cellToForgetWeights1" }, { "name": "cellToOutputWeights1", "src": "cellToOutputWeights1" }, { "name": "inputLayerNormWeights1", "src": "inputLayerNormWeights1" }, { "name": "forgetLayerNormWeights1", "src": "forgetLayerNormWeights1" }, { "name": "cellLayerNormWeights1", "src": "cellLayerNormWeights1" }, { "name": "outputLayerNormWeights1", "src": "outputLayerNormWeights1" } ], "attributes": [ { "name": "activationFunc", "type": "uint", "src": "activationFunc" }, { "name": "clippingThresCell", "type": "float", "src": "clippingThresCell" }, { "name": "clippingThresProj", "type": "float", "src": "clippingThresProj" }, { "name": "cifgEnabled", "type": "bool", "src": "cifgEnabled" }, { "name": "peepholeEnabled", "type": "bool", "src": "peepholeEnabled" }, { "name": "projectionEnabled", "type": "bool", "src": "projectionEnabled" }, { "name": "layerNormEnabled", "type": "bool", "src": "layerNormEnabled" } ] } }, { "name": "QuantizeLayer" }, { "name": "DequantizeLayer" }, { "name": "MergeLayer", "schema": { "category": "Layer" } }, { "name": "SwitchLayer", "schema": { "category": "Layer" } }, { "name": "ConcatLayer", "schema": { "category": "Tensor", "attributes": [ { "name": "concatAxis", "type": "uint", "src": "concatAxis" }, { "name": "numViews", "type": "uint", "src": "numViewes" }, { "name": "numDimensions", "type": "uint", "src": "numDimensions" } ] } }, { "name": "SpaceToDepthLayer", "schema": { "category": "Layer", "attributes": [ { "name": "blockSize", "type": "uint", "src": "blockSize" }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "PreluLayer", "schema": { "category": "Layer" } }, { "name": "TransposeConvolution2dLayer", "schema": { "category": "Layer", "inputs": [ { "name": "weight", "src": "weights" }, { "name": "bias", "src": "biases" } ], "attributes": [ { "name": "padding", "type": "string", "src": ["padTop", "padRight", "padBottom", "padLeft"] }, { "name": "stride", "type": "string", "src": ["strideX", "strideY"] }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "ResizeLayer", "schema": { "category": "Layer", "attributes": [ { "name": "targetWidth", "type": "uint", "src": "targetWidth" }, { "name": "targetHeight", "type": "uint", "src": "targetHeight" }, { "name": "method", "type": "string", "src": "method", "src_type": "ResizeMethod" }, { "name": "dataLayout", "type": "string", "src": "dataLayout", "src_type": "DataLayout" } ] } }, { "name": "StackLayer", "schema": { "category": "Layer", "attributes": [ { "name": "axis", "type": "uint", "src": "axis" }, { "name": "numInputs", "type": "uint", "src": "numInputs" }, { "name": "inputShape", "type": "uint", "src": "inputShape" } ] } }, { "name": "QuantizedLstmLayer", "schema": { "category": "Layer", "inputs": [ { "name": "inputToInputWeights1", "src": "inputToInputWeights1" }, { "name": "inputToForgetWeights1", "src": "inputToForgetWeights1" }, { "name": "inputToCellWeights1", "src": "inputToCellWeights1" }, { "name": "inputToOutputWeights1", "src": "inputToOutputWeights1" }, { "name": "recurrentToInputWeights1", "src": "recurrentToInputWeights1" }, { "name": "recurrentToForgetWeights1", "src": "recurrentToForgetWeights1" }, { "name": "recurrentToCellWeights1", "src": "recurrentToCellWeights1" }, { "name": "recurrentToOutputWeights1", "src": "recurrentToOutputWeights1" }, { "name": "inputGateBias1", "src": "inputGateBias1" }, { "name": "forgetGateBias1", "src": "forgetGateBias1" }, { "name": "cellBias1", "src": "cellBias1" }, { "name": "outputGateBias1", "src": "outputGateBias1" } ] } } ]