tensor-quantization-metadata-study/conformance/cyclonedx-2.0/valid-ai-ml-quantization-2.0.json at main · JunHwan-Kwon/tensor-quantization-metadata-study · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
{
  "$schema": "https://cyclonedx.org/schema/2.0/cyclonedx-2.0.schema.json",
  "specFormat": "CycloneDX",
  "specVersion": "2.0",
  "serialNumber": "urn:uuid:7405c6d8-e9cb-4ed2-adff-8978f1ecf101",
  "version": 1,
  "components": [
    {
      "type": "machine-learning-model",
      "bom-ref": "model-per-tensor-external-parameter",
      "name": "per-tensor-external-parameter-fixture",
      "version": "1",
      "modelProperties": {
        "inputs": [
          {
            "name": "image",
            "modality": "image",
            "dataType": "uint8",
            "quantization": {
              "bits": 8,
              "scheme": "affine",
              "granularity": "per-tensor"
            },
            "shape": [
              1,
              224,
              224,
              3
            ],
            "required": true
          }
        ]
      }
    },
    {
      "type": "machine-learning-model",
      "bom-ref": "model-per-channel-parameter",
      "name": "per-channel-parameter-fixture",
      "version": "1",
      "modelProperties": {
        "inputs": [
          {
            "name": "channel-indexed-parameter",
            "dataType": "int8",
            "quantization": {
              "bits": 8,
              "scheme": "symmetric",
              "granularity": "per-channel",
              "axis": 0
            },
            "shape": [
              16,
              3,
              3,
              3
            ]
          }
        ]
      }
    },
    {
      "type": "machine-learning-model",
      "bom-ref": "model-onnx-blocked-quantization",
      "name": "onnx-blocked-quantization-fixture",
      "version": "1",
      "modelProperties": {
        "quantization": {
          "method": "onnx-QuantizeLinear",
          "bits": 8,
          "scheme": "affine",
          "granularity": "per-group",
          "groupSize": 2,
          "axis": 1
        }
      }
    },
    {
      "type": "machine-learning-model",
      "bom-ref": "model-custom-quantization",
      "name": "custom-quantization-fixture",
      "version": "1",
      "modelProperties": {
        "quantization": {
          "method": "ternary-weight-encoding",
          "bits": 1.58,
          "scheme": {
            "name": "balanced-ternary",
            "description": "A fixture for the draft custom-scheme extension point."
          },
          "granularity": "per-tensor"
        }
      }
    }
  ]
}