mozilla-central/testing/web-platform/tests/webnn/resources/test_data/arg_min.json

Source code

Revision control

Copy as Markdown

Other Tools

{
"tests": [
{
"name": "argMin float32 1D constant tensor default options",
"inputs": {
"input": {
"shape": [24],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32",
"constant": true
}
},
"expected": {
"name": "output",
"shape": [],
"data": [
7
],
"type": "int64"
}
},
{
"name": "argMin float32 1D tensor default options",
"inputs": {
"input": {
"shape": [24],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"expected": {
"name": "output",
"shape": [],
"data": [
7
],
"type": "int64"
}
},
{
"name": "argMin float32 2D tensor default options",
"inputs": {
"input": {
"shape": [4, 6],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"expected": {
"name": "output",
"shape": [],
"data": [
7
],
"type": "int64"
}
},
{
"name": "argMin float32 3D tensor default options",
"inputs": {
"input": {
"shape": [2, 3, 4],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"expected": {
"name": "output",
"shape": [],
"data": [
7
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor default options",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"expected": {
"name": "output",
"shape": [],
"data": [
7
],
"type": "int64"
}
},
{
"name": "argMin float32 5D tensor default options",
"inputs": {
"input": {
"shape": [2, 1, 4, 1, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"expected": {
"name": "output",
"shape": [],
"data": [
7
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.axes=[2]",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"axes": [2]
},
"expected": {
"name": "output",
"shape": [2, 1, 3],
"data": [
2,
2,
0,
0,
0,
0
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.axes=[]",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"axes": []
},
"expected": {
"name": "output",
"shape": [2, 1, 4, 3],
"data": [
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0,
0
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.keepDimensions=true",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"keepDimensions": true
},
"expected": {
"name": "output",
"shape": [1, 1, 1, 1],
"data": [
7
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.keepDimensions=false",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"keepDimensions": false
},
"expected": {
"name": "output",
"shape": [],
"data": [
7
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.selectLastIndex=true",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"selectLastIndex": true
},
"expected": {
"name": "output",
"shape": [],
"data": [
19
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.selectLastIndex=false",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"selectLastIndex": false
},
"expected": {
"name": "output",
"shape": [],
"data": [
7
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.axes=[0, 2] options.keepDimensions=false",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"axes": [0, 2],
"keepDimensions": false
},
"expected": {
"name": "output",
"shape": [1, 3],
"data": [
2,
2,
4
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.axes=[3, 0, 1] options.keepDimensions=true",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"axes": [3, 0, 1],
"keepDimensions": true
},
"expected": {
"name": "output",
"shape": [1, 1, 4, 1],
"data": [
4,
0,
1,
1
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.axes=[0, 2] options.selectLastIndex=false",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"axes": [0, 2],
"selectLastIndex": false
},
"expected": {
"name": "output",
"shape": [1, 3],
"data": [
2,
2,
4
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.axes=[0, 2] options.selectLastIndex=true",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"axes": [0, 2],
"selectLastIndex": true
},
"expected": {
"name": "output",
"shape": [1, 3],
"data": [
6,
6,
4
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.axes=[3, 0, 1] options.selectLastIndex=false",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"axes": [3, 0, 1],
"selectLastIndex": false
},
"expected": {
"name": "output",
"shape": [4],
"data": [
4,
0,
1,
1
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor options.axes=[3, 0, 1] options.selectLastIndex=true",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"axes": [3, 0, 1],
"selectLastIndex": true
},
"expected": {
"name": "output",
"shape": [4],
"data": [
4,
0,
4,
1
],
"type": "int64"
}
},
{
"name": "argMin float32 4D tensor all options",
"inputs": {
"input": {
"shape": [2, 1, 4, 3],
"data": [
3.830124090690262,
-24.986487937638074,
5.299982630691289,
-48.5486590218902,
40.30886781808215,
60.184293919409726,
-82.78385618759043,
-96.50904103637833,
71.87028201591897,
38.866394268784035,
-39.143725517854435,
31.444366685561903,
-82.78385618759043,
-96.50904103637833,
-25.53388886326502,
-16.142265850469343,
66.63677406472371,
82.5119815304117,
-82.78385618759043,
-96.50904103637833,
39.7687246127592,
42.15040238450999,
82.66863662444459,
85.4526923278379
],
"type": "float32"
}
},
"options": {
"axes": [3, 0],
"keepDimensions": true,
"selectLastIndex": false
},
"expected": {
"name": "output",
"shape": [1, 1, 4, 1],
"data": [
4,
0,
1,
1
],
"type": "int64"
}
},
{
"name": "argMin float32 0D scalar options.axes=[]",
"inputs": {
"input": {
"shape": [],
"data": [
3.830124090690262
],
"type": "float32"
}
},
"options": {
"axes": []
},
"expected": {
"name": "output",
"shape": [],
"data": [
0
],
"type": "int64"
}
},
{
"name": "argMin float32 0D scalar options.axes=[] no effect by both keepDimensions and selectLastIndex being true",
"inputs": {
"input": {
"shape": [],
"data": [
3.830124090690262
],
"type": "float32"
}
},
"options": {
"axes": [],
"keepDimensions": true,
"selectLastIndex": true
},
"expected": {
"name": "output",
"shape": [],
"data": [
0
],
"type": "int64"
}
}
]
}