onnxruntime/js/web/test/data/ops/conv.jsonc
Changming Sun 5d692b0136
Merge web machine pools (#23243)
### Description
The Web CI pipeline uses three different Windows machine pools:
1. onnxruntime-Win2022-webgpu-A10
2. onnxruntime-Win2022-VS2022-webgpu-A10
3. onnxruntime-Win-CPU-2022-web

This PR merges them together to reduce ongoing maintenance cost.
2025-01-03 13:53:17 -08:00

548 lines
16 KiB
Text

[
{
"name": "conv without bias addition A",
"operator": "Conv",
"attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [10, 20, 30, 40, 50, 60, 70, 80, 90],
"dims": [1, 1, 3, 3],
"type": "float32"
},
{
"data": [1, 2, 3, 4],
"dims": [1, 1, 2, 2],
"type": "float32"
}
],
"outputs": [
{
"data": [370, 470, 670, 770],
"dims": [1, 1, 2, 2],
"type": "float32"
}
]
}
]
},
{
"name": "Conv without bias addition B",
"operator": "Conv",
"attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [10, 20, 30, 40, 50, 60, 70, 80],
"dims": [1, 2, 2, 2],
"type": "float32"
},
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [1, 2, 2, 2],
"type": "float32"
}
],
"outputs": [
{
"data": [2040],
"dims": [1, 1, 1, 1],
"type": "float32"
}
]
}
]
},
{
"name": "Conv with bias addition A",
"operator": "Conv",
"attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [10, 20, 30, 40, 50, 60, 70, 80],
"dims": [1, 2, 2, 2],
"type": "float32"
},
{
"data": [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
"dims": [4, 2, 2, 2],
"type": "float32"
},
{
"data": [0.1, 0.2, 0.3, 0.4],
"dims": [4],
"type": "float32"
}
],
"outputs": [
{
"data": [360.1, 360.2, 360.3, 360.4],
"dims": [1, 4, 1, 1],
"type": "float32"
}
]
}
]
},
{
"name": "conv with bias addition B",
"operator": "Conv",
"attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [1, 2, 3, 4],
"dims": [1, 1, 2, 2],
"type": "float32"
},
{
"data": [1, 1, 1, 1],
"dims": [1, 1, 2, 2],
"type": "float32"
},
{
"data": [5],
"dims": [1],
"type": "float32"
}
],
"outputs": [
{
"data": [15],
"dims": [1, 1, 1, 1],
"type": "float32"
}
]
}
]
},
{
"name": "conv with bias addition C - NHWC",
"operator": "Conv",
"inputShapeDefinitions": "rankOnly",
"opset": { "domain": "", "version": 17 },
"attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12],
"dims": [3, 1, 2, 2],
"type": "float32"
},
{
"data": [1, 1, 1, 1, 2, 3, 4, 5],
"dims": [2, 1, 2, 2],
"type": "float32"
},
{
"data": [5, 6],
"dims": [2],
"type": "float32"
}
],
"outputs": [
{
"data": [15, 46, 31, 102, 47, 158],
"dims": [3, 2, 1, 1],
"type": "float32"
}
]
},
{
"name": "inChannel = 3, outChannel = 4",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 10],
"dims": [1, 3, 3, 3],
"type": "float32"
},
{
"data": [
1, 1, 1, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 1, 1, 1, 1, 2, 3, 4, 5, 6, 7, 8, 9,
10, 11, 12, 13, 14, 15, 16, 17, 1, 2, 3, 4, 5, 6, 7, 8
],
"dims": [4, 3, 2, 2],
"type": "float32"
},
{
"data": [5, 6, 7, 8],
"dims": [4],
"type": "float32"
}
],
"outputs": [
{
"data": [360, 334, 271, 323, 909, 963, 1024, 1028, 683, 655, 576, 650, 473, 508, 570, 677],
"dims": [1, 4, 2, 2],
"type": "float32"
}
]
}
]
},
{
"name": "conv - group - A",
"operator": "Conv",
"attributes": [
{ "name": "kernel_shape", "data": [1, 1], "type": "ints" },
{ "name": "group", "data": 2, "type": "int" }
],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0],
"dims": [1, 2, 3, 3],
"type": "float32"
},
{
"data": [1.0, 2.0],
"dims": [2, 1, 1, 1],
"type": "float32"
}
],
"outputs": [
{
"data": [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 18.0, 20.0, 22.0, 24.0, 26.0, 28.0, 30.0, 32.0, 34.0],
"dims": [1, 2, 3, 3],
"type": "float32"
}
]
}
]
},
{
"name": "conv - group - B",
"operator": "Conv",
"attributes": [
{ "name": "kernel_shape", "data": [2, 2], "type": "ints" },
{ "name": "group", "data": 3, "type": "int" }
],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [
0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0,
19.0, 20.0, 21.0, 22.0, 23.0, 0, 0, 0
],
"dims": [1, 3, 3, 3],
"type": "float32"
},
{
"data": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0],
"dims": [3, 1, 2, 2],
"type": "float32"
},
{
"data": [0.1, 0.2, 0.3],
"dims": [3],
"type": "float32"
}
],
"outputs": [
{
"data": [27.1, 37.1, 57.1, 67.1, 293.2, 319.2, 371.2, 397.2, 847.3, 889.3, 409.3, 428.3],
"dims": [1, 3, 2, 2],
"type": "float32"
}
]
}
]
},
{
"name": "conv - group - C",
"operator": "Conv",
"attributes": [
{ "name": "kernel_shape", "data": [2, 2], "type": "ints" },
{ "name": "group", "data": 3, "type": "int" }
],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [
0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0,
19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0
],
"dims": [1, 3, 3, 4],
"type": "float32"
},
{
"data": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0],
"dims": [3, 1, 2, 2],
"type": "float32"
}
],
"outputs": [
{
"data": [34, 44, 54, 74, 84, 94, 386, 412, 438, 490, 516, 542, 1122, 1164, 1206, 1290, 1332, 1374],
"dims": [1, 3, 2, 3],
"type": "float32"
}
]
}
]
},
{
"name": "conv - group - NHWC",
"operator": "Conv",
"inputShapeDefinitions": "rankOnly",
"opset": { "domain": "", "version": 17 },
"attributes": [
{ "name": "kernel_shape", "data": [2, 2], "type": "ints" },
{ "name": "group", "data": 2, "type": "int" }
],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [
0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0,
19.0, 20.0, 21.0, 22.0, 23.0, 0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0,
14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0,
10.0, 11.0, 12.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0
],
"dims": [1, 8, 3, 3],
"type": "float32"
},
{
"data": [
1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0,
9.0, 10.0, 11.0, 12.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0,
9.0, 10.0, 11.0, 12.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 1.0, 2.0, 3.0, 4.0,
5.0, 6.0, 7.0, 8.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 1.0, 2.0, 3.0, 4.0,
5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 1.0, 2.0, 3.0, 4.0,
5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0,
1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0
],
"dims": [8, 4, 2, 2],
"type": "float32"
},
{
"data": [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8],
"dims": [8],
"type": "float32"
}
],
"outputs": [
{
"data": [
1224.0999755859375, 1312.0999755859375, 936.0999755859375, 1024.0999755859375, 872.2000122070312,
976.2000122070312, 1016.2000122070312, 1120.199951171875, 1224.300048828125, 1312.300048828125,
936.2999877929688, 1024.300048828125, 872.4000244140625, 976.4000244140625, 1016.4000244140625,
1120.4000244140625, 859.5, 947.5, 594.5, 682.5, 1075.5999755859375, 1179.5999755859375, 866.5999755859375,
970.5999755859375, 859.7000122070312, 947.7000122070312, 594.7000122070312, 682.7000122070312,
1075.800048828125, 1179.800048828125, 866.7999877929688, 970.7999877929688
],
"dims": [1, 8, 2, 2],
"type": "float32"
}
]
}
]
},
{
"name": "conv - vectorize group - A",
"operator": "Conv",
"inputShapeDefinitions": "rankOnly",
"opset": { "domain": "", "version": 17 },
"attributes": [
{ "name": "kernel_shape", "data": [1, 1], "type": "ints" },
{ "name": "group", "data": 2, "type": "int" }
],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0],
"dims": [1, 2, 3, 3],
"type": "float32"
},
{
"data": [1.0, 2.0],
"dims": [2, 1, 1, 1],
"type": "float32"
}
],
"outputs": [
{
"data": [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 18.0, 20.0, 22.0, 24.0, 26.0, 28.0, 30.0, 32.0, 34.0],
"dims": [1, 2, 3, 3],
"type": "float32"
}
]
}
]
},
// {
// "name": "conv - vectorize group - B",
// "operator": "Conv",
// "inputShapeDefinitions": "rankOnly",
// "opset": { "domain": "", "version": 17 },
// "attributes": [
// { "name": "kernel_shape", "data": [2, 2], "type": "ints" },
// { "name": "group", "data": 3, "type": "int" }
// ],
// "cases": [
// {
// "name": "T[0]",
// "inputs": [
// {
// "data": [
// 0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0,
// 19.0, 20.0, 21.0, 22.0, 23.0, 0, 0, 0
// ],
// "dims": [1, 3, 3, 3],
// "type": "float32"
// },
// {
// "data": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0],
// "dims": [3, 1, 2, 2],
// "type": "float32"
// },
// {
// "data": [0.1, 0.2, 0.3],
// "dims": [3],
// "type": "float32"
// }
// ],
// "outputs": [
// {
// "data": [27.1, 37.1, 57.1, 67.1, 293.2, 319.2, 371.2, 397.2, 847.3, 889.3, 409.3, 428.3],
// "dims": [1, 3, 2, 2],
// "type": "float32"
// }
// ]
// }
// ]
// },
{
"name": "conv - vectorize group - C",
"operator": "Conv",
"inputShapeDefinitions": "rankOnly",
"opset": { "domain": "", "version": 17 },
"attributes": [
{ "name": "kernel_shape", "data": [2, 2], "type": "ints" },
{ "name": "group", "data": 3, "type": "int" }
],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [
0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0,
19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0
],
"dims": [1, 3, 3, 4],
"type": "float32"
},
{
"data": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0],
"dims": [3, 1, 2, 2],
"type": "float32"
}
],
"outputs": [
{
"data": [34, 44, 54, 74, 84, 94, 386, 412, 438, 490, 516, 542, 1122, 1164, 1206, 1290, 1332, 1374],
"dims": [1, 3, 2, 3],
"type": "float32"
}
]
}
]
},
// {
// "name": "conv - vectorize group - D",
// "operator": "Conv",
// "inputShapeDefinitions": "rankOnly",
// "opset": { "domain": "", "version": 17 },
// "attributes": [
// { "name": "kernel_shape", "data": [2, 2], "type": "ints" },
// { "name": "group", "data": 3, "type": "int" },
// { "name": "strides", "data": [2, 2], "type": "ints" }
// ],
// "cases": [
// {
// "name": "T[0] strides = [2, 2]",
// "inputs": [
// {
// "data": [
// 0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0,
// 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0
// ],
// "dims": [1, 3, 3, 4],
// "type": "float32"
// },
// {
// "data": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0],
// "dims": [3, 1, 2, 2],
// "type": "float32"
// }
// ],
// "outputs": [
// {
// "data": [34, 54, 386, 438, 1122, 1206],
// "dims": [1, 3, 1, 2],
// "type": "float32"
// }
// ]
// }
// ]
// },
{
"name": "conv - pointwise",
"operator": "Conv",
"attributes": [{ "name": "kernel_shape", "data": [1, 1], "type": "ints" }],
"cases": [
{
"name": "T[0]",
"inputs": [
{
"data": [
0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0,
19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0
],
"dims": [1, 8, 2, 2],
"type": "float32"
},
{
"data": [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0],
"dims": [2, 8, 1, 1],
"type": "float32"
},
{
"data": [0.5, 0.4],
"dims": [2],
"type": "float32"
}
],
"outputs": [
{
"data": [560.5, 588.5, 616.5, 644.5, 1456.4, 1548.4, 1640.4, 1732.4],
"dims": [1, 2, 2, 2],
"type": "float32"
}
]
}
]
}
]