verif/conformance/tosa_main_profile_ops_info.json


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
96

{
    "avg_pool2d": {
        "group": "tensor",
        "generator_args": [
            [
                "--target-dtype",
                "fp32",
                "--target-dtype",
                "fp16",
                "--target-dtype",
                "bf16",
                "--fp-values-range",
                "-2.0,2.0",
                "--target-shape",
                "1,49,42,28",
                "--target-shape",
                "3,11,44,3",
                "--allow-pooling-and-conv-oversizes"
            ],
            [
                "--target-dtype",
                "fp32",
                "--fp-values-range",
                "-2.0,2.0",
                "--target-shape",
                "1,65535,5,1",
                "--target-shape",
                "1,3,65537,1",
                "--allow-pooling-and-conv-oversizes"
            ]
        ],
        "params": {
        },
        "permutes": [
            "shape",
            "type",
            "accum_type"
        ],
        "sparsity": {
            "pad": 17
        },
        "profile": [
            "tosa-mi"
        ]
    },
    "conv2d": {
        "group": "tensor",
        "generator_args": [
            [
                "--target-dtype",
                "fp32",
                "--target-dtype",
                "fp16",
                "--target-dtype",
                "bf16",
                "--fp-values-range",
                "-2.0,2.0",
                "--target-shape",
                "1,49,42,28",
                "--target-shape",
                "1,11,44,13",
                "--allow-pooling-and-conv-oversizes"
            ],
            [
                "--target-dtype",
                "fp32",
                "--fp-values-range",
                "-2.0,2.0",
                "--target-shape",
                "1,65535,4,1",
                "--target-shape",
                "1,5,65536,1",
                "--max-conv-dilation",
                "1",
                "--allow-pooling-and-conv-oversizes"
            ]
        ],
        "params": {
            "shape": [],
            "type": [],
            "kernel": [],
            "stride": [],
            "pad": [],
            "dilation": []
        },
        "permutes": [
            "kernel",
            "shape",
            "type",
            "pad"
        ],
        "profile": [
            "tosa-mi"
        ]
    }
}