summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/webnn/resources/test_data/arg_min.json
diff options
context:
space:
mode:
Diffstat (limited to 'testing/web-platform/tests/webnn/resources/test_data/arg_min.json')
-rw-r--r--testing/web-platform/tests/webnn/resources/test_data/arg_min.json963
1 files changed, 963 insertions, 0 deletions
diff --git a/testing/web-platform/tests/webnn/resources/test_data/arg_min.json b/testing/web-platform/tests/webnn/resources/test_data/arg_min.json
new file mode 100644
index 0000000000..132a2dc3e8
--- /dev/null
+++ b/testing/web-platform/tests/webnn/resources/test_data/arg_min.json
@@ -0,0 +1,963 @@
+{
+ "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": true
+ },
+ "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"
+ }
+ }
+ ]
+} \ No newline at end of file
generated by cgit v1.2.3 (git 2.39.1) at 2024-07-21 14:51:02 +0000