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