mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
da180b20fa
### Description Fix JSEP ConvTranspose shader code errors. ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
289 lines
8.1 KiB
Text
289 lines
8.1 KiB
Text
[
|
|
{
|
|
"name": "ConvTranspose without bias addition A",
|
|
"operator": "ConvTranspose",
|
|
"attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
|
|
"cases": [
|
|
{
|
|
"name": "T[0]",
|
|
"inputs": [
|
|
{
|
|
"data": [10, 20, 30, 40],
|
|
"dims": [1, 1, 2, 2],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [1, 2, 3, 4],
|
|
"dims": [1, 1, 2, 2],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [10, 40, 40, 60, 200, 160, 90, 240, 160],
|
|
"dims": [1, 1, 3, 3],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "ConvTranspose without bias addition B",
|
|
"operator": "ConvTranspose",
|
|
"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": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15],
|
|
"dims": [2, 2, 2, 2],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [
|
|
400, 940, 560, 1080, 2520, 1480, 760, 1740, 1000, 640, 1500, 880, 1720, 3960, 2280, 1160, 2620, 1480
|
|
],
|
|
"dims": [1, 2, 3, 3],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "ConvTranspose with bias addition A",
|
|
"operator": "ConvTranspose",
|
|
"attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
|
|
"cases": [
|
|
{
|
|
"name": "T[0]",
|
|
"inputs": [
|
|
{
|
|
"data": [10, 20, 30, 40],
|
|
"dims": [1, 4, 1, 1],
|
|
"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, 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, 4, 2, 2],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [0.1, 0.2, 0.3, 0.4],
|
|
"dims": [4],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [
|
|
100.0999984741211, 100.0999984741211, 100.0999984741211, 100.0999984741211, 100.19999694824219,
|
|
100.19999694824219, 100.19999694824219, 100.19999694824219, 100.30000305175781, 100.30000305175781,
|
|
100.30000305175781, 100.30000305175781, 100.4000015258789, 100.4000015258789, 100.4000015258789,
|
|
100.4000015258789
|
|
],
|
|
"dims": [1, 4, 2, 2],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "ConvTranspose with bias addition B",
|
|
"operator": "ConvTranspose",
|
|
"attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
|
|
"cases": [
|
|
{
|
|
"name": "T[0]",
|
|
"inputs": [
|
|
{
|
|
"data": [6, 8, 7, 9, 15, 11, 8, 12, 9],
|
|
"dims": [1, 1, 3, 3],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [1, 1, 1, 1],
|
|
"dims": [1, 1, 2, 2],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [5],
|
|
"dims": [1],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [11, 19, 20, 12, 20, 43, 46, 23, 22, 49, 52, 25, 13, 25, 26, 14],
|
|
"dims": [1, 1, 4, 4],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "ConvTranspose- group - A",
|
|
"operator": "ConvTranspose",
|
|
"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, 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"
|
|
},
|
|
{
|
|
"data": [1.0, 2.0],
|
|
"dims": [2, 1, 1, 1],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [0, 1, 2, 3, 4, 5, 6, 7, 8, 36, 40, 44, 48, 52, 56, 60, 64, 68],
|
|
"dims": [1, 2, 3, 3],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "ConvTranspose- group - B",
|
|
"operator": "ConvTranspose",
|
|
"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.125, 0.25, 0.375],
|
|
"dims": [3],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [
|
|
0.125, 1.125, 4.125, 4.125, 3.125, 13.125, 23.125, 18.125, 15.125, 43.125, 53.125, 36.125, 18.125, 45.125,
|
|
52.125, 32.125, 45.25, 104.25, 115.25, 66.25, 123.25, 279.25, 305.25, 172.25, 159.25, 357.25, 383.25,
|
|
214.25, 105.25, 232.25, 247.25, 136.25, 162.375, 351.375, 370.375, 200.375, 387.375, 833.375, 875.375,
|
|
470.375, 231.375, 494.375, 517.375, 276.375, 0.375, 0.375, 0.375, 0.375
|
|
],
|
|
"dims": [1, 3, 4, 4],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "ConvTranspose- group - C",
|
|
"operator": "ConvTranspose",
|
|
"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": [
|
|
0, 1, 4, 7, 6, 4, 16, 26, 36, 26, 20, 56, 66, 76, 50, 24, 59, 66, 73, 44, 60, 137, 148, 159, 90, 164, 368,
|
|
394, 420, 234, 212, 472, 498, 524, 290, 140, 307, 322, 337, 184, 216, 465, 484, 503, 270, 516, 1104, 1146,
|
|
1188, 634, 596, 1272, 1314, 1356, 722, 352, 747, 770, 793, 420
|
|
],
|
|
"dims": [1, 3, 4, 5],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
|
|
{
|
|
"name": "ConvTranspose- pointwise",
|
|
"operator": "ConvTranspose",
|
|
"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],
|
|
"dims": [1, 2, 2, 2],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [0.0, 1.0, 2.0, 3.0],
|
|
"dims": [2, 2, 1, 1],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [1, 2],
|
|
"dims": [2],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [9, 11, 13, 15, 14, 18, 22, 26],
|
|
"dims": [1, 2, 2, 2],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
]
|