xref: /aosp_15_r20/external/tensorflow/tensorflow/lite/testdata/sparse_tensor.json (revision b6fb3261f9314811a0f4371741dbb8839866f948)
1{
2  "version": 3,
3  "operator_codes": [{"builtin_code": "CUSTOM", "custom_code": "FakeOp"}],
4  "subgraphs": [
5    {
6      "tensors": [
7        {
8          "shape": [
9            4,
10            4
11          ],
12          "name": "sparse_tensor",
13          "buffer": 1,
14          "type": "INT8",
15          "quantization": {
16          },
17          "is_variable": "false",
18          "sparsity": {
19            "traversal_order": [0, 1, 2, 3],
20            "block_map": [0, 1],
21            "dim_metadata": [
22              {
23                "format": "DENSE",
24                "dense_size": 2
25              },
26              {
27                "format": "SPARSE_CSR",
28                "array_segments_type": "Uint8Vector",
29                "array_segments": {"values": [0, 2, 3]},
30                "array_indices_type": "Uint8Vector",
31                "array_indices": {"values": [0, 1, 1]}
32              },
33              {
34                "format": "DENSE",
35                "dense_size": 2
36              },
37              {
38                "format": "DENSE",
39                "dense_size": 2
40              }
41            ]
42          }
43        },
44        {
45          "shape": [
46            4,
47            4
48          ],
49          "name": "output_tensor",
50          "type": "INT8"
51        }
52      ],
53      "inputs": [0],
54      "outputs": [0],
55      "operators": [{"inputs":[-1], "outputs":[1]}]
56    }
57  ],
58  "buffers": [
59    {
60    },
61    {
62      "data": [
63        1, 0, 0, 4,
64        2, 3, 0, 0,
65        5, 0, 0, 6
66      ]
67    }
68  ]
69}
70