CMSIS_5/CMSIS/NN/Tests/UnitTest/TestCases/Common/svdf_template.json

{
  "version": 3,
  "operator_codes": [
    {
      "deprecated_builtin_code": 27,
      "version": 1
    }
  ],
  "subgraphs": [
    {
      "tensors": [
        {
          "shape": [
            batches,
            input_size
          ],
          "type": "INT8",
          "buffer": 0,
          "name": "tensor_input",
          "quantization": {
            "scale": [
              input_scale
            ],
            "zero_point": [
              input_zp
            ],
            "quantized_dimension": 0
          },
          "is_variable": false
        },
        {
          "shape": [
            number_filters,
            input_size
          ],
          "type": "INT8",
          "buffer": 1,
          "name": "tensor_weight_1",
          "quantization": {
            "scale": [
              w_1_scale
            ],
            "zero_point": [
              w_1_zp
            ],
            "quantized_dimension": 0
          },
          "is_variable": false
        },
        {
          "shape": [
            number_filters,
            memory_size
          ],
          "type": "INT16",
          "buffer": 2,
          "name": "tensor_weight_2",
          "quantization": {
            "scale": [
              w_2_scale
            ],
            "zero_point": [
              w_2_zp
            ],
            "quantized_dimension": 0
          },
          "is_variable": false
        },
        {
          "shape": [
              number_units
          ],
          "type": "INT32",
          "buffer": 3,
          "name": "tensor_bias",
          "quantization": {
            "scale": [
              bias_scale
            ],
            "zero_point": [
              bias_zp
            ],
            "quantized_dimension": 0
          },
          "is_variable": false
        },
        {
          "shape": [
            batches,
            memory_sizeXnumber_filters
          ],
          "type": "INT16",
          "buffer": 4,
          "name": "tensor_state",
          "quantization": {
            "scale": [
              state_scale
            ],
            "zero_point": [
              state_zp
            ],
            "quantized_dimension": 0
          },
          "is_variable": true
        },
        {
          "shape": [
            batches,
            number_units
          ],
          "type": "INT8",
          "buffer": 5,
          "name": "tensor_output",
          "quantization": {
            "scale": [
              output_scale
            ],
            "zero_point": [
              output_zp
            ],
            "quantized_dimension": 0
          },
          "is_variable": false
        }
      ],
      "inputs": [
        0
      ],
      "outputs": [
        5
      ],
      "operators": [
        {
          "opcode_index": 0,
          "inputs": [
            0,
            1,
            2,
            3,
            4
          ],
          "outputs": [
            5
          ],
          "builtin_options_type": "SVDFOptions",
          "builtin_options": {
            "rank": rank_value,
            "fused_activation_function": "RELU"
          },
          "custom_options_format": "FLEXBUFFERS"
        }
      ]
    }
  ],
  "description": "CMSIS-NN unit test model",
  "buffers": [
    {},
    {
      "data": []
    },
    {
      "data": []
    },
    {
      "data": []
    },
    {
      "data": []
    },
    {
      "data": []
    }
  ]
}