summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/webnn/resources/test_data/gather.json
diff options
context:
space:
mode:
Diffstat (limited to 'testing/web-platform/tests/webnn/resources/test_data/gather.json')
-rw-r--r--testing/web-platform/tests/webnn/resources/test_data/gather.json1268
1 files changed, 1268 insertions, 0 deletions
diff --git a/testing/web-platform/tests/webnn/resources/test_data/gather.json b/testing/web-platform/tests/webnn/resources/test_data/gather.json
new file mode 100644
index 0000000000..a01654637f
--- /dev/null
+++ b/testing/web-platform/tests/webnn/resources/test_data/gather.json
@@ -0,0 +1,1268 @@
+{
+ "tests": [
+ {
+ "name": "gather float32 1D tensor and uint32 0D scalar indices default options",
+ "inputs": {
+ "input": {
+ "shape": [24],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [],
+ "data": [
+ 4
+ ],
+ "type": "uint32"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [],
+ "data": [
+ 89.0337142944336
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 1D tensor and int64 0D scalar indices default options",
+ "inputs": {
+ "input": {
+ "shape": [24],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [],
+ "data": [
+ 0
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [],
+ "data": [
+ -66.05901336669922
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 1D tensor and int64 1D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [24],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [8],
+ "data": [
+ 16,
+ 20,
+ 6,
+ 11,
+ 17,
+ 19,
+ 13,
+ 17
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [8],
+ "data": [
+ 10.829925537109375,
+ 0.9129875898361206,
+ 43.84803771972656,
+ -50.42131042480469,
+ -19.693084716796875,
+ 43.11057662963867,
+ 55.620765686035156,
+ -19.693084716796875
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 1D tensor and int64 2D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [24],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [2, 2],
+ "data": [
+ 14,
+ 9,
+ 21,
+ 17
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [2, 2],
+ "data": [
+ 44.92119598388672,
+ 41.94132614135742,
+ -7.699817180633545,
+ -19.693084716796875
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 1D tensor and int64 3D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [24],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [2, 3, 3],
+ "data": [
+ 17,
+ 19,
+ 14,
+ 16,
+ 13,
+ 0,
+ 5,
+ 15,
+ 18,
+ 18,
+ 6,
+ 20,
+ 7,
+ 22,
+ 5,
+ 1,
+ 4,
+ 19
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [2, 3, 3],
+ "data": [
+ -19.693084716796875,
+ 43.11057662963867,
+ 44.92119598388672,
+ 10.829925537109375,
+ 55.620765686035156,
+ -66.05901336669922,
+ -45.89653396606445,
+ 56.828636169433594,
+ -37.696800231933594,
+ -37.696800231933594,
+ 43.84803771972656,
+ 0.9129875898361206,
+ 48.81806945800781,
+ 25.76774024963379,
+ -45.89653396606445,
+ -68.9197006225586,
+ 89.0337142944336,
+ 43.11057662963867
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 1D tensor and int64 4D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [24],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [1, 2, 2, 3],
+ "data": [
+ 18,
+ 18,
+ 22,
+ 11,
+ 8,
+ 15,
+ 12,
+ 11,
+ 7,
+ 13,
+ 7,
+ 7
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [1, 2, 2, 3],
+ "data": [
+ -37.696800231933594,
+ -37.696800231933594,
+ 25.76774024963379,
+ -50.42131042480469,
+ 51.79948425292969,
+ 56.828636169433594,
+ 90.2870101928711,
+ -50.42131042480469,
+ 48.81806945800781,
+ 55.620765686035156,
+ 48.81806945800781,
+ 48.81806945800781
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 2D tensor and 0D scalar indices default options",
+ "inputs": {
+ "input": {
+ "shape": [12, 2],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [],
+ "data": [
+ 11
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [2],
+ "data": [
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 2D tensor and 1D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [12, 2],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [8],
+ "data": [
+ 1,
+ 10,
+ 9,
+ 0,
+ 3,
+ 5,
+ 3,
+ 8
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [8, 2],
+ "data": [
+ -77.02045440673828,
+ -26.158037185668945,
+ 0.9129875898361206,
+ -7.699817180633545,
+ -37.696800231933594,
+ 43.11057662963867,
+ -66.05901336669922,
+ -68.9197006225586,
+ 43.84803771972656,
+ 48.81806945800781,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 43.84803771972656,
+ 48.81806945800781,
+ 10.829925537109375,
+ -19.693084716796875
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 2D tensor and 2D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [12, 2],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [2, 2],
+ "data": [
+ 4,
+ 8,
+ 9,
+ 10
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [2, 2, 2],
+ "data": [
+ 51.79948425292969,
+ 41.94132614135742,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 2D tensor and 3D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [12, 2],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [2, 3, 3],
+ "data": [
+ 8,
+ 2,
+ 2,
+ 3,
+ 4,
+ 1,
+ 2,
+ 2,
+ 7,
+ 11,
+ 4,
+ 11,
+ 6,
+ 6,
+ 7,
+ 3,
+ 11,
+ 10
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [2, 3, 3, 2],
+ "data": [
+ 10.829925537109375,
+ -19.693084716796875,
+ 89.0337142944336,
+ -45.89653396606445,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 89.0337142944336,
+ -45.89653396606445,
+ 44.92119598388672,
+ 56.828636169433594,
+ 25.76774024963379,
+ 73.60064697265625,
+ 51.79948425292969,
+ 41.94132614135742,
+ 25.76774024963379,
+ 73.60064697265625,
+ 90.2870101928711,
+ 55.620765686035156,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 43.84803771972656,
+ 48.81806945800781,
+ 25.76774024963379,
+ 73.60064697265625,
+ 0.9129875898361206,
+ -7.699817180633545
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 2D tensor and 4D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [12, 2],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [1, 2, 2, 3],
+ "data": [
+ 6,
+ 9,
+ 7,
+ 3,
+ 4,
+ 7,
+ 4,
+ 3,
+ 7,
+ 7,
+ 6,
+ 0
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [1, 2, 2, 3, 2],
+ "data": [
+ 90.2870101928711,
+ 55.620765686035156,
+ -37.696800231933594,
+ 43.11057662963867,
+ 44.92119598388672,
+ 56.828636169433594,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ 44.92119598388672,
+ 56.828636169433594,
+ 51.79948425292969,
+ 41.94132614135742,
+ 43.84803771972656,
+ 48.81806945800781,
+ 44.92119598388672,
+ 56.828636169433594,
+ 44.92119598388672,
+ 56.828636169433594,
+ 90.2870101928711,
+ 55.620765686035156,
+ -66.05901336669922,
+ -68.9197006225586
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 3D tensor and 2D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [3, 4, 2],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [2, 2],
+ "data": [
+ 2,
+ 1,
+ 1,
+ 1
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [2, 2, 4, 2],
+ "data": [
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 4D tensor and 2D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [8, 1, 1, 3],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [2, 2],
+ "data": [
+ 0,
+ 0,
+ 7,
+ 4
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [2, 2, 1, 1, 3],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 5D tensor and 1D indices default options",
+ "inputs": {
+ "input": {
+ "shape": [4, 2, 1, 1, 3],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [3],
+ "data": [
+ 3,
+ 2,
+ 2
+ ],
+ "type": "int64"
+ }
+ },
+ "expected": {
+ "name": "output",
+ "shape": [3, 2, 1, 1, 3],
+ "data": [
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 3D tensor and 1D indices options.axis=1",
+ "inputs": {
+ "input": {
+ "shape": [3, 4, 2],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [3],
+ "data": [
+ 1,
+ 1,
+ 2
+ ],
+ "type": "int64"
+ }
+ },
+ "options": {
+ "axis": 1
+ },
+ "expected": {
+ "name": "output",
+ "shape": [3, 3, 2],
+ "data": [
+ -77.02045440673828,
+ -26.158037185668945,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ -1.1303654909133911,
+ -50.42131042480469,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ -37.696800231933594,
+ 43.11057662963867,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 3D tensor and 2D indices options.axis=2",
+ "inputs": {
+ "input": {
+ "shape": [3, 4, 2],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [2, 2],
+ "data": [
+ 0,
+ 0,
+ 0,
+ 1
+ ],
+ "type": "int64"
+ }
+ },
+ "options": {
+ "axis": 2
+ },
+ "expected": {
+ "name": "output",
+ "shape": [3, 4, 2, 2],
+ "data": [
+ -66.05901336669922,
+ -66.05901336669922,
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -77.02045440673828,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ 89.0337142944336,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 43.84803771972656,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 51.79948425292969,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -1.1303654909133911,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 90.2870101928711,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 44.92119598388672,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ 10.829925537109375,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ -37.696800231933594,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ 0.9129875898361206,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 25.76774024963379,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 4D tensor and 2D indices explict options.axis=0",
+ "inputs": {
+ "input": {
+ "shape": [8, 1, 1, 3],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [2, 2],
+ "data": [
+ 0,
+ 0,
+ 7,
+ 4
+ ],
+ "type": "int64"
+ }
+ },
+ "options": {
+ "axis": 0
+ },
+ "expected": {
+ "name": "output",
+ "shape": [2, 2, 1, 1, 3],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672
+ ],
+ "type": "float32"
+ }
+ },
+ {
+ "name": "gather float32 5D tensor and 0D scalar indices options.axis=4",
+ "inputs": {
+ "input": {
+ "shape": [4, 2, 1, 1, 3],
+ "data": [
+ -66.05901336669922,
+ -68.9197006225586,
+ -77.02045440673828,
+ -26.158037185668945,
+ 89.0337142944336,
+ -45.89653396606445,
+ 43.84803771972656,
+ 48.81806945800781,
+ 51.79948425292969,
+ 41.94132614135742,
+ -1.1303654909133911,
+ -50.42131042480469,
+ 90.2870101928711,
+ 55.620765686035156,
+ 44.92119598388672,
+ 56.828636169433594,
+ 10.829925537109375,
+ -19.693084716796875,
+ -37.696800231933594,
+ 43.11057662963867,
+ 0.9129875898361206,
+ -7.699817180633545,
+ 25.76774024963379,
+ 73.60064697265625
+ ],
+ "type": "float32"
+ },
+ "indices": {
+ "shape": [],
+ "data": [
+ 1
+ ],
+ "type": "int64"
+ }
+ },
+ "options": {
+ "axis": 4
+ },
+ "expected": {
+ "name": "output",
+ "shape": [4, 2, 1, 1],
+ "data": [
+ -68.9197006225586,
+ 89.0337142944336,
+ 48.81806945800781,
+ -1.1303654909133911,
+ 55.620765686035156,
+ 10.829925537109375,
+ 43.11057662963867,
+ 25.76774024963379
+ ],
+ "type": "float32"
+ }
+ }
+ ]
+} \ No newline at end of file
generated by cgit v1.2.3 (git 2.39.1) at 2024-07-21 14:55:45 +0000