onnxruntime/js/web/test/data/ops/pad.jsonc
Yulong Wang d13f3153d7
[js/webgpu] enable op test for webgpu (#16542)
### Description
This change enables the JSON-format operator tests for webgpu.

Usage:

```
npm test -- op abs.jsonc -b=webgpu
```
2023-07-06 08:35:19 -07:00

1298 lines
24 KiB
Text

[
{
"name": "constant 2D",
"operator": "Pad",
"opsets": [{"domain": "", "version": 10}],
"attributes": [
{ "name": "mode", "data": "constant", "type": "string" },
{ "name": "value", "data": 1.2, "type": "float" },
{ "name": "pads", "data": [3, 2, 2, 3], "type": "ints" }
],
"cases": [
{
"name": "[2,2]->[7,7]",
"inputs": [
{
"data": [1.0, 2.0, 3.0, 4.0],
"dims": [2, 2],
"type": "float32"
}
],
"outputs": [
{
"data": [
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.0,
2.0,
1.2,
1.2,
1.2,
1.2,
1.2,
3.0,
4.0,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2,
1.2
],
"dims": [7, 7],
"type": "float32"
}
]
}
]
},
{
"name": "constant 3D",
"operator": "Pad",
"opsets": [{"domain": "", "version": 10}],
"attributes": [
{ "name": "mode", "data": "constant", "type": "string" },
{ "name": "value", "data": 2.3, "type": "float" },
{ "name": "pads", "data": [1, 2, 2, 2, 3, 1], "type": "ints" }
],
"cases": [
{
"name": "[2,3,4]->[5,8,7]",
"inputs": [
{
"data": [
1,
2,
3,
4,
5,
6,
7,
8,
1,
2,
3,
4,
5,
6,
7,
8,
1,
2,
3,
4,
5,
6,
7,
8
],
"dims": [2, 3, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
1.0,
2.0,
3.0,
4.0,
2.3,
2.3,
2.3,
5.0,
6.0,
7.0,
8.0,
2.3,
2.3,
2.3,
1.0,
2.0,
3.0,
4.0,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
5.0,
6.0,
7.0,
8.0,
2.3,
2.3,
2.3,
1.0,
2.0,
3.0,
4.0,
2.3,
2.3,
2.3,
5.0,
6.0,
7.0,
8.0,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3,
2.3
],
"dims": [5, 8, 7],
"type": "float32"
}
]
}
]
},
{
"name": "Reflect 1D",
"operator": "Pad",
"opsets": [{"domain": "", "version": 10}],
"attributes": [
{ "name": "mode", "data": "reflect", "type": "string" },
{ "name": "pads", "data": [5, 7], "type": "ints" }
],
"cases": [
{
"name": "[3]->[15]",
"inputs": [
{
"data": [1.0, 2.0, 3.0],
"dims": [3],
"type": "float32"
}
],
"outputs": [
{
"data": [
2.0,
1.0,
2.0,
3.0,
2.0,
1.0,
2.0,
3.0,
2.0,
1.0,
2.0,
3.0,
2.0,
1.0,
2.0
],
"dims": [15],
"type": "float32"
}
]
}
]
},
{
"name": "Reflect 2D",
"operator": "Pad",
"opsets": [{"domain": "", "version": 10}],
"attributes": [
{ "name": "mode", "data": "reflect", "type": "string" },
{ "name": "pads", "data": [3, 2, 2, 5], "type": "ints" }
],
"cases": [
{
"name": "[2,2]->[7,9]",
"inputs": [
{
"data": [1.0, 2.0, 3.0, 4.0],
"dims": [2, 2],
"type": "float32"
}
],
"outputs": [
{
"data": [
3.0,
4.0,
3.0,
4.0,
3.0,
4.0,
3.0,
4.0,
3.0,
1.0,
2.0,
1.0,
2.0,
1.0,
2.0,
1.0,
2.0,
1.0,
3.0,
4.0,
3.0,
4.0,
3.0,
4.0,
3.0,
4.0,
3.0,
1.0,
2.0,
1.0,
2.0,
1.0,
2.0,
1.0,
2.0,
1.0,
3.0,
4.0,
3.0,
4.0,
3.0,
4.0,
3.0,
4.0,
3.0,
1.0,
2.0,
1.0,
2.0,
1.0,
2.0,
1.0,
2.0,
1.0,
3.0,
4.0,
3.0,
4.0,
3.0,
4.0,
3.0,
4.0,
3.0
],
"dims": [7, 9],
"type": "float32"
}
]
}
]
},
{
"name": "Reflect 3D",
"operator": "Pad",
"opsets": [{"domain": "", "version": 10}],
"attributes": [
{ "name": "mode", "data": "reflect", "type": "string" },
{ "name": "pads", "data": [1, 2, 2, 2, 3, 1], "type": "ints" }
],
"cases": [
{
"name": "[2,3,4]->[5, 8, 7]",
"inputs": [
{
"data": [
1,
2,
3,
4,
5,
6,
7,
8,
1,
2,
3,
4,
5,
6,
7,
8,
1,
2,
3,
4,
5,
6,
7,
8
],
"dims": [2, 3, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0,
7.0,
6.0,
5.0,
6.0,
7.0,
8.0,
7.0,
3.0,
2.0,
1.0,
2.0,
3.0,
4.0,
3.0
],
"dims": [5, 8, 7],
"type": "float32"
}
]
}
]
},
{
"name": "Edge 2D",
"operator": "Pad",
"opsets": [{"domain": "", "version": 10}],
"attributes": [
{ "name": "mode", "data": "edge", "type": "string" },
{ "name": "pads", "data": [3, 2, 2, 3], "type": "ints" }
],
"cases": [
{
"name": "[2,2]->[7,7]",
"inputs": [
{
"data": [1.0, 2.0, 3.0, 4.0],
"dims": [2, 2],
"type": "float32"
}
],
"outputs": [
{
"data": [
1.0,
1.0,
1.0,
2.0,
2.0,
2.0,
2.0,
1.0,
1.0,
1.0,
2.0,
2.0,
2.0,
2.0,
1.0,
1.0,
1.0,
2.0,
2.0,
2.0,
2.0,
1.0,
1.0,
1.0,
2.0,
2.0,
2.0,
2.0,
3.0,
3.0,
3.0,
4.0,
4.0,
4.0,
4.0,
3.0,
3.0,
3.0,
4.0,
4.0,
4.0,
4.0,
3.0,
3.0,
3.0,
4.0,
4.0,
4.0,
4.0
],
"dims": [7, 7],
"type": "float32"
}
]
}
]
},
{
"name": "Edge 3D",
"operator": "Pad",
"opsets": [{"domain": "", "version": 10}],
"attributes": [
{ "name": "mode", "data": "edge", "type": "string" },
{ "name": "pads", "data": [1, 2, 2, 2, 3, 1], "type": "ints" }
],
"cases": [
{
"name": "[2,3,4]->[5, 8, 7]",
"inputs": [
{
"data": [
1,
2,
3,
4,
5,
6,
7,
8,
1,
2,
3,
4,
5,
6,
7,
8,
1,
2,
3,
4,
5,
6,
7,
8
],
"dims": [2, 3, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
1.0,
1.0,
1.0,
2.0,
3.0,
4.0,
4.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0,
5.0,
5.0,
5.0,
6.0,
7.0,
8.0,
8.0
],
"dims": [5, 8, 7],
"type": "float32"
}
]
}
]
}
]