onnxruntime/js/web/test/data/ops/where_broadcast.jsonc
Xu Xing 992f3e4609
[js/webgpu] Support where (#17544)
Supported type: float. int32_t, uint32_t, bool.
Case where_broadcast.jsonc is not enabled due to
https://github.com/microsoft/onnxruntime/issues/17405.

### Description
<!-- Describe your changes. -->



### 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. -->

---------

Co-authored-by: Yulong Wang <7679871+fs-eire@users.noreply.github.com>
2023-10-03 14:28:21 -07:00

84 lines
2 KiB
Text

[
{
"name": "Where with no attributes",
"operator": "Where",
"attributes": [],
"cases": [
{
// This failed due to: https://github.com/microsoft/onnxruntime/issues/17405.
"name": "T[3 6] T[3 6] T[1] float32 broadcast",
"inputs": [
{
"data": [
true,
true,
true,
true,
true,
false,
false,
false,
false,
false,
false,
true,
true,
true,
true,
true,
true,
true
],
"dims": [3, 6],
"type": "bool"
},
{
"data": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17],
"dims": [3, 6],
"type": "float32"
},
{
"data": [-1.0],
"dims": [1],
"type": "float32"
}
],
"outputs": [
{
"data": [0, 1, 2, 3, 4, -1, -1, -1, -1, -1, -1, 11, 12, 13, 14, 15, 16, 17],
"dims": [3, 6],
"type": "float32"
}
]
},
{
// This failed due to: https://github.com/microsoft/onnxruntime/issues/17405.
"name": "T[3 1] T[3 6] T[1] float32 broadcast",
"inputs": [
{
"data": [true, false, true],
"dims": [3, 1],
"type": "bool"
},
{
"data": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17],
"dims": [3, 6],
"type": "float32"
},
{
"data": [-1.0],
"dims": [1],
"type": "float32"
}
],
"outputs": [
{
"data": [0, 1, 2, 3, 4, 5, -1, -1, -1, -1, -1, -1, 12, 13, 14, 15, 16, 17],
"dims": [3, 6],
"type": "float32"
}
]
}
]
}
]