saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-28 22:56:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/js/web/test/data/ops/cumsum.jsonc

1363 lines
30 KiB
Text
Raw Normal View History

[JS/Web] Added CumSum operator to JSEP (#18637) ### Description Added CumSum operator ### Motivation and Context Reduce CPU <->GPU data movement.
2023-12-05 15:51:53 +00:00
[
{
"name": "CumSum",
"operator": "CumSum",
"attributes": [
{ "name": "exclusive", "data": 0, "type": "int" },
{ "name": "reverse", "data": 0, "type": "int" }
],
"opset": {
"domain": "",
"version": 11
},
"cases": [
{
"name": "CumSum 1-D; axis = 0; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5],
"dims": [5],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 3, 6, 10, 15],
"dims": [5],
"type": "float32"
}
]
},
{
"name": "CumSum 1-D; axis = -1; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5],
"dims": [5],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 3, 6, 10, 15],
"dims": [5],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = 0; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 2, 3, 5, 7, 9],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = -1; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 3, 6, 4, 9, 15],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = 1; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 3, 6, 4, 9, 15],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = -2; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [-2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 2, 3, 5, 7, 9],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (3x3); axis = 0; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9],
"dims": [3, 3],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 2, 3, 5, 7, 9, 12, 15, 18],
"dims": [3, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (3x3); axis = 1; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9],
"dims": [3, 3],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 3, 6, 4, 9, 15, 7, 15, 24],
"dims": [3, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 0; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 2, 3, 4, 6, 8, 10, 12],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 1; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 2, 4, 6, 5, 6, 12, 14],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -1; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 3, 3, 7, 5, 11, 7, 15],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 2; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 3, 3, 7, 5, 11, 7, 15],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -2; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 2, 4, 6, 5, 6, 12, 14],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -3; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-3],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 2, 3, 4, 6, 8, 10, 12],
"dims": [2, 2, 2],
"type": "float32"
}
]
}
]
},
{
"name": "CumSum",
"operator": "CumSum",
"attributes": [
{ "name": "exclusive", "data": 1, "type": "int" },
{ "name": "reverse", "data": 0, "type": "int" }
],
"opset": {
"domain": "",
"version": 11
},
"cases": [
{
"name": "CumSum 1-D; axis = 0; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5],
"dims": [5],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 1, 3, 6, 10],
"dims": [5],
"type": "float32"
}
]
},
{
"name": "CumSum 1-D; axis = -1; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5],
"dims": [5],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 1, 3, 6, 10],
"dims": [5],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = 0; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 0, 0, 1, 2, 3],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = -1; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 1, 3, 0, 4, 9],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = 1; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 1, 3, 0, 4, 9],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = -2",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [-2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 0, 0, 1, 2, 3],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (3x3); axis = 0; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9],
"dims": [3, 3],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 0, 0, 1, 2, 3, 5, 7, 9],
"dims": [3, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (3x3); axis = 1; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9],
"dims": [3, 3],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 1, 3, 0, 4, 9, 0, 7, 15],
"dims": [3, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 0; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 0, 0, 0, 1, 2, 3, 4],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 1; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 0, 1, 2, 0, 0, 5, 6],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -1; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 1, 0, 3, 0, 5, 0, 7],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 2; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 1, 0, 3, 0, 5, 0, 7],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -2; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 0, 1, 2, 0, 0, 5, 6],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -3; exclusive = 1, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-3],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [0, 0, 0, 0, 1, 2, 3, 4],
"dims": [2, 2, 2],
"type": "float32"
}
]
}
]
},
{
"name": "CumSum",
"operator": "CumSum",
"attributes": [
{ "name": "exclusive", "data": 0, "type": "int" },
{ "name": "reverse", "data": 1, "type": "int" }
],
"opset": {
"domain": "",
"version": 11
},
"cases": [
{
"name": "CumSum 1-D; axis = 0; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5],
"dims": [5],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [15, 14, 12, 9, 5],
"dims": [5],
"type": "float32"
}
]
},
{
"name": "CumSum 1-D; axis = -1; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5],
"dims": [5],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [15, 14, 12, 9, 5],
"dims": [5],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = 0; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [5, 7, 9, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = -1; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [6, 5, 3, 15, 11, 6],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = 1; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [6, 5, 3, 15, 11, 6],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = -2; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [-2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [5, 7, 9, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (3x3); axis = 0; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9],
"dims": [3, 3],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [12, 15, 18, 11, 13, 15, 7, 8, 9],
"dims": [3, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (3x3); axis = 1; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9],
"dims": [3, 3],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [6, 5, 3, 15, 11, 6, 24, 17, 9],
"dims": [3, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 0; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [6, 8, 10, 12, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 1; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [4, 6, 3, 4, 12, 14, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -1; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [3, 2, 7, 4, 11, 6, 15, 8],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 2; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [3, 2, 7, 4, 11, 6, 15, 8],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -2; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [4, 6, 3, 4, 12, 14, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -3; exclusive = 0, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-3],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [6, 8, 10, 12, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
}
]
}
]
},
{
"name": "CumSum",
"operator": "CumSum",
"attributes": [
{ "name": "exclusive", "data": 1, "type": "int" },
{ "name": "reverse", "data": 1, "type": "int" }
],
"opset": {
"domain": "",
"version": 11
},
"cases": [
{
"name": "CumSum 1-D; axis = 0; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5],
"dims": [5],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [14, 12, 9, 5, 0],
"dims": [5],
"type": "float32"
}
]
},
{
"name": "CumSum 1-D; axis = -1; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5],
"dims": [5],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [14, 12, 9, 5, 0],
"dims": [5],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = 0; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [4, 5, 6, 0, 0, 0],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = -1; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [5, 3, 0, 11, 6, 0],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = 1; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [5, 3, 0, 11, 6, 0],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (2x3); axis = -2; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [-2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [4, 5, 6, 0, 0, 0],
"dims": [2, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (3x3); axis = 0; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9],
"dims": [3, 3],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [11, 13, 15, 7, 8, 9, 0, 0, 0],
"dims": [3, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 2-D (3x3); axis = 1; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9],
"dims": [3, 3],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [5, 3, 0, 11, 6, 0, 17, 9, 0],
"dims": [3, 3],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 0; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [0],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [5, 6, 7, 8, 0, 0, 0, 0],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 1; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [3, 4, 0, 0, 7, 8, 0, 0],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -1; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-1],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [2, 0, 4, 0, 6, 0, 8, 0],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = 2; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [2, 0, 4, 0, 6, 0, 8, 0],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -2; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-2],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [3, 4, 0, 0, 7, 8, 0, 0],
"dims": [2, 2, 2],
"type": "float32"
}
]
},
{
"name": "CumSum 3-D; axis = -3; exclusive = 1, reverse = 1",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 2, 2],
"type": "float32"
},
{
"data": [-3],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [5, 6, 7, 8, 0, 0, 0, 0],
"dims": [2, 2, 2],
"type": "float32"
}
]
}
]
},
{
"name": "CumSum",
"operator": "CumSum",
"attributes": [
{ "name": "exclusive", "data": 0, "type": "int" },
{ "name": "reverse", "data": 0, "type": "int" }
],
"opset": {
"domain": "",
"version": 11
},
"cases": [
{
"name": "CumSum 5-D; axis = 0; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5],
"dims": [1, 1, 1, 1, 5],
"type": "float32"
},
{
"data": [4],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 3, 6, 10, 15],
"dims": [1, 1, 1, 1, 5],
"type": "float32"
}
]
}
]
[js/webgpu] Fix shader compilation errors in cumsum (#18779) ### Description This PR fixes below shader compilation errors: ``` Tint WGSL reader failure: :39:31 error: no matching overload for operator + (f32, i32) 5 candidate operators: operator + (T, T) -> T where: T is abstract-float, abstract-int, f32, i32, u32 or f16 operator + (vecN<T>, T) -> vecN<T> where: T is abstract-float, abstract-int, f32, i32, u32 or f16 operator + (T, vecN<T>) -> vecN<T> where: T is abstract-float, abstract-int, f32, i32, u32 or f16 operator + (vecN<T>, vecN<T>) -> vecN<T> where: T is abstract-float, abstract-int, f32, i32, u32 or f16 operator + (matNxM<T>, matNxM<T>) -> matNxM<T> where: T is abstract-float, f32 or f16 sum = sum + get_inputByIndices(inputIndices); ^ - While validating [ShaderModuleDescriptor "CumSum"] - While calling [Device].CreateShaderModule([ShaderModuleDescriptor "CumSum"]).
2023-12-12 02:11:38 +00:00
},
{
"name": "CumSum",
"operator": "CumSum",
"attributes": [
{ "name": "exclusive", "data": 0, "type": "int" },
{ "name": "reverse", "data": 0, "type": "int" }
],
"opset": {
"domain": "",
"version": 11
},
"cases": [
{
"name": "CumSum int32; axis = 0; exclusive = 0, reverse = 0",
"inputs": [
{
"data": [1, 2, 3, 4, 5],
"dims": [1, 1, 1, 1, 5],
"type": "int32"
},
{
"data": [4],
"dims": [],
"type": "int32"
}
],
"outputs": [
{
"data": [1, 3, 6, 10, 15],
"dims": [1, 1, 1, 1, 5],
"type": "int32"
}
]
}
]
[JS/Web] Added CumSum operator to JSEP (#18637) ### Description Added CumSum operator ### Motivation and Context Reduce CPU <->GPU data movement.
2023-12-05 15:51:53 +00:00
}
]
Reference in a new issue Copy permalink