| { |
| "op": "ai.onnx.Flatten", |
| "fixtureArrays": { |
| "onnx_backend_flatten_input_input": [0.54881352186203, 0.7151893377304077, 0.6027633547782898, 0.5448831915855408, 0.42365479469299316, 0.6458941102027893, 0.4375872015953064, 0.891772985458374, 0.9636627435684204, 0.3834415078163147, 0.7917250394821167, 0.5288949012756348, 0.5680445432662964, 0.9255966544151306, 0.07103605568408966, 0.08712930232286453, 0.020218396559357643, 0.832619845867157, 0.7781567573547363, 0.8700121641159058, 0.978618323802948, 0.7991585731506348, 0.4614793658256531, 0.7805292010307312, 0.11827442795038223, 0.6399210095405579, 0.14335328340530396, 0.9446688890457153, 0.5218483209609985, 0.4146619439125061, 0.26455560326576233, 0.7742336988449097, 0.4561503231525421, 0.568433940410614, 0.018789799883961678, 0.6176354885101318, 0.6120957136154175, 0.6169340014457703, 0.9437480568885803, 0.681820273399353, 0.35950788855552673, 0.43703195452690125, 0.6976311802864075, 0.0602254718542099, 0.6667667031288147, 0.670637845993042, 0.21038256585597992, 0.12892629206180573, 0.31542834639549255, 0.36371076107025146, 0.5701967477798462, 0.4386015236377716, 0.9883738160133362, 0.10204481333494186, 0.20887675881385803, 0.16130951046943665, 0.6531082987785339, 0.25329160690307617, 0.4663107693195343, 0.24442559480667114, 0.15896958112716675, 0.11037514358758926, 0.6563295722007751, 0.13818295300006866, 0.1965823620557785, 0.3687251806259155, 0.8209932446479797, 0.09710127860307693, 0.8379449248313904, 0.0960984081029892, 0.9764594435691833, 0.4686512053012848, 0.9767611026763916, 0.6048455238342285, 0.7392635941505432, 0.03918779268860817, 0.28280696272850037, 0.12019655853509903, 0.296140193939209, 0.11872772127389908, 0.3179831802845001, 0.414262980222702, 0.06414749473333359, 0.6924721002578735, 0.5666014552116394, 0.26538950204849243, 0.5232480764389038, 0.09394051134586334, 0.5759465098381042, 0.9292961955070496, 0.3185689449310303, 0.6674103736877441, 0.13179786503314972, 0.7163271903991699, 0.28940609097480774, 0.18319135904312134, 0.5865129232406616, 0.02010754682123661, 0.8289400339126587, 0.004695476032793522, 0.6778165102005005, 0.2700079679489136, 0.7351940274238586, 0.9621885418891907, 0.2487531453371048, 0.5761573314666748, 0.5920419096946716, 0.5722519159317017, 0.22308163344860077, 0.9527490139007568, 0.4471253752708435, 0.8464086651802063, 0.6994792819023132, 0.2974369525909424, 0.8137978315353394, 0.396505743265152, 0.8811032176017761, 0.5812729001045227, 0.8817353844642639, 0.6925315856933594], |
| "ort_uint32_bool_like_axis2_input_input": [0, 1, 0, 1, 1, 0, 1, 0, 0, 0, 0, 1, 1, 0, 1, 1, 1, 0, 0, 1, 1, 1, 0, 1] |
| }, |
| "cases": [ |
| { |
| "name": "int16_copy_boundaries", |
| "attrs": { "axis": 1 }, |
| "inputs": { |
| "input": { "dtype": "int16", "shape": [2, 2], "data": { "kind": "values", "values": [-32768, -1, 0, 32767] } } |
| }, |
| "outputs": { |
| "output": { |
| "dtype": "int16", |
| "shape": [2, 2], |
| "tolerance": 0, |
| "data": { "kind": "values", "values": [-32768, -1, 0, 32767] } |
| } |
| } |
| }, |
| { |
| "name": "rank3_to_2d", |
| "attrs": { "axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4], |
| "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29 } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 12], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "uint32_bool_like_rank3_to_2d", |
| "attrs": { "axis": 2 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint32", |
| "shape": [2, 3, 2], |
| "data": { "kind": "values", "values": [1, 0, 1, 1, 0, 0, 2, 0, 2, 2, 0, 2] } |
| } |
| }, |
| "outputs": { "output": { "dtype": "uint32", "shape": [6, 2], "tolerance": 0 } } |
| }, |
| { |
| "name": "backend_axis0", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4], |
| "data": { "kind": "fillFloat32", "sinStep": 0.11, "cosStep": 0.23 } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 24], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "backend_axis3", |
| "attrs": { "axis": 3 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4], |
| "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29 } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [24, 1], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "negative_axis_minus1", |
| "attrs": { "axis": -1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4], |
| "data": { "kind": "fillFloat32", "sinStep": 0.07, "cosStep": 0.19 } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [6, 4], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "f16_axis0", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { "dtype": "float16", "shape": [2, 2], "data": { "kind": "values", "values": [1.0, -2.0, 3.0, -4.0] } } |
| }, |
| "outputs": { "output": { "dtype": "float16", "shape": [1, 4], "tolerance": 0.001 } } |
| }, |
| { |
| "name": "scalar_axis0_to_1x1", |
| "attrs": { "axis": 0 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [], "data": { "kind": "values", "values": [3.25] } } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 1], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "zero_sized_axis1", |
| "attrs": { "axis": 1 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [2, 0, 3], "data": { "kind": "values", "values": [] } } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 0], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "zero_sized_negative_axis", |
| "attrs": { "axis": -1 }, |
| "inputs": { "input": { "dtype": "float16", "shape": [2, 0, 3], "data": { "kind": "values", "values": [] } } }, |
| "outputs": { "output": { "dtype": "float16", "shape": [0, 3], "tolerance": 0.001 } } |
| }, |
| { |
| "name": "ort_rank4_default_axis", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpTest.Flatten_default_axis" |
| }, |
| "inputs": { "input": { "dtype": "float32", "shape": [2, 3, 4, 5], "data": { "kind": "constant", "value": 1.0 } } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 60], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "ort_rank4_axis0", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpTest.Flatten_axis0" |
| }, |
| "attrs": { "axis": 0 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [2, 3, 4, 5], "data": { "kind": "constant", "value": 1.0 } } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 120], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "ort_rank4_axis3", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpTest.Flatten_axis3" |
| }, |
| "attrs": { "axis": 3 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [2, 3, 4, 5], "data": { "kind": "constant", "value": 1.0 } } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [24, 5], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "ort_rank4_negative_axis_minus1", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpTest.Flatten_neg_axis3" |
| }, |
| "attrs": { "axis": -1 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [2, 3, 4, 5], "data": { "kind": "constant", "value": 1.0 } } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [24, 5], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "ort_rank4_axis4_trailing_unit", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpTest.Flatten_axis4" |
| }, |
| "attrs": { "axis": 4 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [1, 2, 4, 2], |
| "data": { |
| "kind": "values", |
| "values": [0.0, 0.1, 0.2, 0.3, 1.0, 1.1, 1.2, 1.3, 2.0, 2.1, 2.2, 2.3, 3.0, 3.1, 3.2, 3.3] |
| } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [16, 1], "tolerance": 0.000001 } } |
| }, |
| { |
| "name": "ort_uint32_bool_like_axis2", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpTest.Flatten_axis2_bool", |
| "notes": "Adapts ORT's bool fixture to logical uint32 values because this framework stores bool-like tensors as u32 slots." |
| }, |
| "attrs": { "axis": 2 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint32", |
| "shape": [2, 2, 2, 3], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/ort_uint32_bool_like_axis2_input_input" } } |
| } |
| }, |
| "outputs": { "output": { "dtype": "uint32", "shape": [4, 6], "tolerance": 0 } } |
| }, |
| { |
| "name": "ort_bool_axis2", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpTest.Flatten_axis2_bool", |
| "notes": "Direct bool version of ORT's axis=2 flatten fixture." |
| }, |
| "attrs": { "axis": 2 }, |
| "inputs": { |
| "input": { |
| "dtype": "bool", |
| "shape": [2, 2, 2, 3], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/ort_uint32_bool_like_axis2_input_input" } } |
| } |
| }, |
| "outputs": { |
| "output": { |
| "dtype": "bool", |
| "shape": [4, 6], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/ort_uint32_bool_like_axis2_input_input" } }, |
| "tolerance": 0 |
| } |
| } |
| }, |
| { |
| "name": "ort_default_axis_int32_dtype_path", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpTest.Flatten_default_axis", |
| "notes": "Same default-axis reshape as ORT's float fixture, using supported int32 data storage." |
| }, |
| "inputs": { "input": { "dtype": "int32", "shape": [2, 3, 4, 5], "data": { "kind": "constant", "value": 1 } } }, |
| "outputs": { "output": { "dtype": "int32", "shape": [2, 60], "tolerance": 0 } } |
| }, |
| { |
| "name": "ort_model_projection_rank1_axis0_for_broadcast", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpModelTest.Flatten_broadcast", |
| "notes": "Node-level projection of the serialized-model regression: Flatten axis=0 turns a rank-1 vector into a [1,N] row tensor used by a following broadcast Mul." |
| }, |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [4], "data": { "kind": "values", "values": [0.0, 1.0, 2.0, 3.0] } } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 4], "tolerance": 0 } } |
| }, |
| { |
| "name": "onnx_backend_flatten_axis0", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4, 5], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/onnx_backend_flatten_input_input" } } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 120] } }, |
| "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_flatten_axis0" } |
| }, |
| { |
| "name": "onnx_backend_flatten_axis1", |
| "attrs": { "axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4, 5], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/onnx_backend_flatten_input_input" } } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 60] } }, |
| "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_flatten_axis1" } |
| }, |
| { |
| "name": "onnx_backend_flatten_axis2", |
| "attrs": { "axis": 2 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4, 5], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/onnx_backend_flatten_input_input" } } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [6, 20] } }, |
| "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_flatten_axis2" } |
| }, |
| { |
| "name": "onnx_backend_flatten_axis3", |
| "attrs": { "axis": 3 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4, 5], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/onnx_backend_flatten_input_input" } } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [24, 5] } }, |
| "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_flatten_axis3" } |
| }, |
| { |
| "name": "onnx_backend_flatten_default_axis", |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [5, 4, 3, 2], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/onnx_backend_flatten_input_input" } } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [5, 24] } }, |
| "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_flatten_default_axis" } |
| }, |
| { |
| "name": "onnx_backend_flatten_negative_axis1", |
| "attrs": { "axis": -1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4, 5], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/onnx_backend_flatten_input_input" } } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [24, 5] } }, |
| "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_flatten_negative_axis1" } |
| }, |
| { |
| "name": "onnx_backend_flatten_negative_axis2", |
| "attrs": { "axis": -2 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4, 5], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/onnx_backend_flatten_input_input" } } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [6, 20] } }, |
| "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_flatten_negative_axis2" } |
| }, |
| { |
| "name": "onnx_backend_flatten_negative_axis3", |
| "attrs": { "axis": -3 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4, 5], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/onnx_backend_flatten_input_input" } } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [2, 60] } }, |
| "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_flatten_negative_axis3" } |
| }, |
| { |
| "name": "onnx_backend_flatten_negative_axis4", |
| "attrs": { "axis": -4 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4, 5], |
| "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/onnx_backend_flatten_input_input" } } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 120] } }, |
| "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_flatten_negative_axis4" } |
| }, |
| { |
| "name": "ort_int8_axis0_edge_values", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpTest.Flatten_axis0", |
| "notes": "Same axis-0 reshape as ORT's float case, using ONNX-valid signed byte edge values." |
| }, |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { "dtype": "int8", "shape": [2, 3], "data": { "kind": "values", "values": [-128, -1, 0, 1, 42, 127] } } |
| }, |
| "outputs": { "output": { "dtype": "int8", "shape": [1, 6], "tolerance": 0 } } |
| }, |
| { |
| "name": "ort_uint8_negative_axis_edge_values", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/nn/flatten_op_test.cc", |
| "test": "FlattenOpTest.Flatten_neg_axis3", |
| "notes": "Same negative-axis reshape behavior as ORT's rank-4 case, using ONNX-valid unsigned byte edge values." |
| }, |
| "attrs": { "axis": -1 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint8", |
| "shape": [1, 2, 2, 2], |
| "data": { "kind": "values", "values": [0, 1, 127, 128, 200, 250, 254, 255] } |
| } |
| }, |
| "outputs": { "output": { "dtype": "uint8", "shape": [4, 2], "tolerance": 0 } } |
| }, |
| { |
| "name": "scalar_copy_dispatch_clamp", |
| "provenance": { |
| "notes": "Odd numel forces the scalar copy variant; WORKGROUP_SIZE=4 makes ceil(numel/WG)=65536 exceed the 65535 workgroups-per-dimension limit, pinning the min(...,65535) dispatch clamp + grid-stride loop (some threads copy two elements)." |
| }, |
| "attrs": { "axis": 1 }, |
| "tunables": { "WORKGROUP_SIZE": 4 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [262141], |
| "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31 } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [262141, 1], "tolerance": 0 } } |
| }, |
| { |
| "name": "int8_numel2_scalar_copy_path", |
| "attrs": { "axis": 1 }, |
| "inputs": { "input": { "dtype": "int8", "shape": [2, 1], "data": { "kind": "values", "values": [-128, 127] } } }, |
| "outputs": { "output": { "dtype": "int8", "shape": [2, 1], "tolerance": 0 } } |
| }, |
| { |
| "name": "uint8_numel3_scalar_copy_path", |
| "attrs": { "axis": 1 }, |
| "inputs": { |
| "input": { "dtype": "uint8", "shape": [3, 1], "data": { "kind": "values", "values": [0, 128, 255] } } |
| }, |
| "outputs": { "output": { "dtype": "uint8", "shape": [3, 1], "tolerance": 0 } } |
| }, |
| { |
| "name": "f16_numel6_scalar_copy_path", |
| "attrs": { "axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float16", |
| "shape": [2, 3], |
| "data": { "kind": "values", "values": [-65504.0, 0.0, 1.0, -1.0, 65504.0, 0.5] } |
| } |
| }, |
| "outputs": { "output": { "dtype": "float16", "shape": [2, 3], "tolerance": 0 } } |
| } |
| ] |
| } |
|
|
