onnxruntime/js/web/test/data/ops/skip-simplified-layer-norm.jsonc
Yulong Wang 50bd4571ac
[js/web] support SimplifiedLayerNorm and SkipSimplifiedLayerNorm (#20277)
### Description
Support operator `SimplifiedLayerNorm` and `SkipSimplifiedLayerNorm` for
WebGPU backend.
2024-04-11 14:08:50 -07:00

53 lines
1.4 KiB
Text

[
{
"name": "SkipSimplifiedLayerNormalization",
"operator": "SkipSimplifiedLayerNormalization",
"opset": { "domain": "com.microsoft", "version": 1 },
"attributes": [
{
"name": "epsilon",
"data": 1e-5,
"type": "float"
}
],
"inputShapeDefinitions": "rankOnly",
"cases": [
{
"name": "default",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16],
"dims": [1, 2, 8],
"type": "float32"
},
{
"data": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15],
"dims": [1, 2, 8],
"type": "float32"
},
{
"data": [2, 2, 2, 2, 2, 2, 2, 2],
"dims": [8],
"type": "float32"
}
],
"outputs": [
{
"data": [
0.21693046391010284, 0.650791347026825, 1.084652304649353, 1.5185132026672363, 1.9523741006851196,
2.386234998703003, 2.820096015930176, 3.2539567947387695, 1.3915272951126099, 1.5552364587783813,
1.7189455032348633, 1.8826546669006348, 2.046363592147827, 2.2100727558135986, 2.37378191947937,
2.5374910831451416
],
"dims": [1, 2, 8],
"type": "float32"
},
{
"data": null,
"type": "float32"
}
]
}
]
}
]