mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-29 20:14:01 +00:00
[js/webgpu] Fix reshape int32 test case (#17113)
Co-authored-by: Xing Xu <xing.xu@intel.com> Co-authored-by: Xing Xu <xing.xu@intel.com>
This commit is contained in:
parent
8289e8b6ef
commit
33ecde9af1
1 changed files with 96 additions and 1 deletions
|
|
@ -1,6 +1,6 @@
|
|||
[
|
||||
{
|
||||
"name": "Reshape with '0' and '-1' in the shape tensor input",
|
||||
"name": "Reshape float32 with '0' and '-1' in the shape tensor input",
|
||||
"operator": "Reshape",
|
||||
"attributes": [],
|
||||
"cases": [
|
||||
|
|
@ -93,5 +93,100 @@
|
|||
]
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "Reshape int32 with '0' and '-1' in the shape tensor input",
|
||||
"operator": "Reshape",
|
||||
"attributes": [],
|
||||
"cases": [
|
||||
{
|
||||
"name": "T[0]",
|
||||
"inputs": [
|
||||
{
|
||||
"data": [1, 1, 1, 1],
|
||||
"dims": [2, 2],
|
||||
"type": "int32"
|
||||
},
|
||||
{
|
||||
"data": [-1, 0],
|
||||
"dims": [2],
|
||||
"type": "int64"
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"data": [1, 1, 1, 1],
|
||||
"dims": [2, 2],
|
||||
"type": "int32"
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "Scalar to 1D",
|
||||
"inputs": [
|
||||
{
|
||||
"data": [1],
|
||||
"dims": [],
|
||||
"type": "int32"
|
||||
},
|
||||
{
|
||||
"data": [1],
|
||||
"dims": [1],
|
||||
"type": "int64"
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"data": [1],
|
||||
"dims": [1],
|
||||
"type": "int32"
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "Scalar to 2D",
|
||||
"inputs": [
|
||||
{
|
||||
"data": [1],
|
||||
"dims": [],
|
||||
"type": "int32"
|
||||
},
|
||||
{
|
||||
"data": [1, 1],
|
||||
"dims": [2],
|
||||
"type": "int64"
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"data": [1],
|
||||
"dims": [1, 1],
|
||||
"type": "int32"
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "Scalar to 2D with -1 in shape hints",
|
||||
"inputs": [
|
||||
{
|
||||
"data": [1],
|
||||
"dims": [],
|
||||
"type": "int32"
|
||||
},
|
||||
{
|
||||
"data": [-1, 1],
|
||||
"dims": [2],
|
||||
"type": "int64"
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"data": [1],
|
||||
"dims": [1, 1],
|
||||
"type": "int32"
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
|
|
|
|||
Loading…
Reference in a new issue