onnxruntime/js/web/test/data/ops/instance-norm.jsonc
Xu Xing 76dfe5347c
[js/webgpu] Support uniforms for instance-norm (#18929)
Co-authored-by: Guenther Schmuelling <guschmue@microsoft.com>
2024-01-09 14:56:00 -08:00

228 lines
6.2 KiB
Text

[
{
"name": "Simple test with NHWC",
"operator": "InstanceNormalization",
"inputShapeDefinitions": "rankOnly",
"opset": { "domain": "", "version": 17 },
"cases": [
{
"name": "Simple test",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 9, 8, 7, 6, 5, 4],
"dims": [1, 4, 2, 2],
"type": "float32"
},
{
"data": [1, 2, 3, 4],
"dims": [4],
"type": "float32"
},
{
"data": [4, 5, 6, 7],
"dims": [4],
"type": "float32"
}
],
"outputs": [
{
"data": [
2.6583645343780518, 3.552788257598877, 4.447211742401123, 5.341635704040527, 2.3167295455932617,
4.105576515197754, 5.8944244384765625, 7.683271408081055, 6, 10.242595672607422, 6, 1.7574005126953125,
12.36654281616211, 8.788846969604492, 5.211153030395508, 1.633458137512207
],
"dims": [1, 4, 2, 2],
"type": "float32"
}
]
}
]
},
{
"name": "Simple test with NHWC, components 1",
"operator": "InstanceNormalization",
"inputShapeDefinitions": "rankOnly",
"opset": { "domain": "", "version": 17 },
"cases": [
{
"name": "Simple test",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 9, 8, 7, 6, 5],
"dims": [1, 5, 3, 1],
"type": "float32"
},
{
"data": [1, 2, 3, 4, 5],
"dims": [5],
"type": "float32"
},
{
"data": [4, 5, 6, 7, 8],
"dims": [5],
"type": "float32"
}
],
"outputs": [
{
"data": [
2.775264263153076, 4, 5.224735260009766, 2.5505285263061523, 5, 7.449470520019531, 2.325794219970703, 6,
9.674205780029297, 11.898944854736328, 7, 2.1010589599609375, 14.123676300048828, 8, 1.876321792602539
],
"dims": [1, 5, 3, 1],
"type": "float32"
}
]
}
]
},
{
"name": "Simple test with NHWC, components 2",
"operator": "InstanceNormalization",
"inputShapeDefinitions": "rankOnly",
"opset": { "domain": "", "version": 17 },
"cases": [
{
"name": "Simple test",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 9, 8],
"dims": [2, 6, 1, 1],
"type": "float32"
},
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [6],
"type": "float32"
},
{
"data": [4, 5, 6, 7, 8, 9],
"dims": [6],
"type": "float32"
}
],
"outputs": [
{
"data": [4, 5, 6, 7, 8, 9, 4, 5, 6, 7, 8, 9],
"dims": [2, 6, 1, 1],
"type": "float32"
}
]
}
]
},
{
"name": "Simple test with NCHW",
"operator": "InstanceNormalization",
"opset": { "domain": "", "version": 17 },
"cases": [
{
"name": "Simple test",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 9, 8, 7, 6, 5, 4],
"dims": [1, 4, 2, 2],
"type": "float32"
},
{
"data": [1, 2, 3, 4],
"dims": [4],
"type": "float32"
},
{
"data": [4, 5, 6, 7],
"dims": [4],
"type": "float32"
}
],
"outputs": [
{
"data": [
2.6583645343780518, 3.552788257598877, 4.447211742401123, 5.341635704040527, 2.3167295455932617,
4.105576515197754, 5.8944244384765625, 7.683271408081055, 6, 10.242595672607422, 6, 1.7574005126953125,
12.36654281616211, 8.788846969604492, 5.211153030395508, 1.633458137512207
],
"dims": [1, 4, 2, 2],
"type": "float32"
}
]
}
]
},
{
"name": "Simple test with NCHW, components 1",
"operator": "InstanceNormalization",
"opset": { "domain": "", "version": 17 },
"cases": [
{
"name": "Simple test",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 9, 8, 7, 6, 5],
"dims": [1, 5, 3, 1],
"type": "float32"
},
{
"data": [1, 2, 3, 4, 5],
"dims": [5],
"type": "float32"
},
{
"data": [4, 5, 6, 7, 8],
"dims": [5],
"type": "float32"
}
],
"outputs": [
{
"data": [
2.775264263153076, 4, 5.224735260009766, 2.5505285263061523, 5, 7.449470520019531, 2.325794219970703, 6,
9.674205780029297, 11.898944854736328, 7, 2.1010589599609375, 14.123676300048828, 8, 1.876321792602539
],
"dims": [1, 5, 3, 1],
"type": "float32"
}
]
}
]
},
{
"name": "Simple test with NCHW, components 2",
"operator": "InstanceNormalization",
"opset": { "domain": "", "version": 17 },
"cases": [
{
"name": "Simple test",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 9, 8, 7, 6, 5, 4, 3, 2],
"dims": [1, 3, 6, 1],
"type": "float32"
},
{
"data": [1, 2, 3],
"dims": [3],
"type": "float32"
},
{
"data": [4, 5, 6],
"dims": [3],
"type": "float32"
}
],
"outputs": [
{
"data": [
2.5361523628234863, 3.1216912269592285, 3.70723032951355, 4.292769432067871, 4.878308296203613,
5.4638471603393555, 1.8666191101074219, 3.9555397033691406, 6.044460296630859, 8.133380889892578,
6.044460296630859, 3.9555397033691406, 10.3915433883667, 8.634925842285156, 6.878308296203613,
5.121691703796387, 3.365074634552002, 1.6084575653076172
],
"dims": [1, 3, 6, 1],
"type": "float32"
}
]
}
]
}
]