mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
4ebc9c3b5e
* add web * add script and test * fix lint * add test/data/ops * add test/data/node/ to gitignore * modify scripts * add onnxjs * fix tests * fix test-runner * fix sourcemap * fix onnxjs profiling * update test list * update README * resolve comments * set wasm as default backend * rename package * update copyright header * do not use class "Buffer" in browser context * revise readme
79 lines
1.6 KiB
Text
79 lines
1.6 KiB
Text
[
|
|
{
|
|
"name": "Split on Axis 0",
|
|
"operator": "Split",
|
|
"attributes": [
|
|
{ "name": "axis", "data": 0, "type": "int" },
|
|
{ "name": "split", "data": [2, 4], "type": "ints" }
|
|
],
|
|
"cases": [
|
|
{
|
|
"name": "T[6]",
|
|
"inputs": [
|
|
{
|
|
"data": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0],
|
|
"dims": [6],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [1.0, 2.0],
|
|
"dims": [2],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [3.0, 4.0, 5.0, 6.0],
|
|
"dims": [4],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "Split on Axis 1 - 2D",
|
|
"operator": "Split",
|
|
"attributes": [
|
|
{ "name": "axis", "data": 1, "type": "int" },
|
|
{ "name": "split", "data": [2, 4], "type": "ints" }
|
|
],
|
|
"cases": [
|
|
{
|
|
"name": "T[6]",
|
|
"inputs": [
|
|
{
|
|
"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": [2, 6],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [1.0, 2.0, 7.0, 8.0],
|
|
"dims": [2, 2],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [3.0, 4.0, 5.0, 6.0, 9.0, 10.0, 11.0, 12.0],
|
|
"dims": [2, 4],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
]
|