| { |
| "op": "ai.onnx.SplitToSequence", |
| "fixtureArrays": { |
| "ort_positive_axis_scalar_split_rank3_input_input": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24], |
| "onnx_backend_split_scalar_axis1_three_outputs_input_input": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17] |
| }, |
| "cases": [ |
| { |
| "name": "dispatch_cliff_two_outputs_axis0_f32", |
| "attrs": { "axis": 0, "keepdims": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [16777217, 1], |
| "data": { "kind": "linspace", "start": 0.0, "end": 1.0 } |
| }, |
| "split": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [16777216, 1] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [16777216, 1], "tolerance": 0 }, |
| "y1": { "dtype": "float32", "shape": [1, 1], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "split_1d_three_chunks_f32", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [6, 2], |
| "data": { "kind": "fillFloat32", "sinStep": 0.21, "cosStep": 0.37 } |
| }, |
| "split": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [2, 2, 2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "y2": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "split_1d_unequal_two_chunks_f32", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [5, 3], |
| "data": { "kind": "fillFloat32", "sinStep": 0.11, "cosStep": 0.29 } |
| }, |
| "split": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [2, 3] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 3], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [3, 3], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "scalar_split_two_chunks_f32", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { "kind": "fillFloat32", "sinStep": 0.33, "cosStep": 0.17 } |
| }, |
| "split": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [3] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [1, 2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "no_split_keepdims1_three_f32", |
| "attrs": { "axis": 0, "keepdims": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [3, 2], |
| "data": { "kind": "fillFloat32", "sinStep": 0.41, "cosStep": 0.23 } |
| } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [1, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [1, 2], "tolerance": 0.000001 }, |
| "y2": { "dtype": "float32", "shape": [1, 2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "no_split_keepdims0_three_f32", |
| "attrs": { "axis": 0, "keepdims": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [3, 2], |
| "data": { "kind": "fillFloat32", "sinStep": 0.41, "cosStep": 0.23 } |
| } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [2], "tolerance": 0.000001 }, |
| "y2": { "dtype": "float32", "shape": [2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "axis1_two_chunks_f32", |
| "attrs": { "axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 4], |
| "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.31 } |
| }, |
| "split": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [2, 2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "split_1d_three_chunks_f16", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float16", |
| "shape": [6, 2], |
| "data": { "kind": "fillFloat32", "sinStep": 0.21, "cosStep": 0.37 } |
| }, |
| "split": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [2, 2, 2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float16", "shape": [2, 2], "tolerance": 0.002 }, |
| "y1": { "dtype": "float16", "shape": [2, 2], "tolerance": 0.002 }, |
| "y2": { "dtype": "float16", "shape": [2, 2], "tolerance": 0.002 } |
| } |
| }, |
| { |
| "name": "ort_default_axis0_equal_split_float", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_DefaultAxis0EqualSplitFloat" |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { "kind": "values", "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0] } |
| }, |
| "split": { "dtype": "uint32", "shape": [1, 2], "data": { "kind": "values", "values": [2, 2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "ort_default_axis0_equal_split_float16", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_DefaultAxis0EqualSplitMLFloat16" |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float16", |
| "shape": [4, 2], |
| "data": { "kind": "values", "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0] } |
| }, |
| "split": { "dtype": "uint32", "shape": [1, 2], "data": { "kind": "values", "values": [2, 2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float16", "shape": [2, 2], "tolerance": 0.002 }, |
| "y1": { "dtype": "float16", "shape": [2, 2], "tolerance": 0.002 } |
| } |
| }, |
| { |
| "name": "ort_default_axis0_scalar_split_float", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_DefaultAxis0EqualSplitFloatScalarSplit" |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { "kind": "values", "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0] } |
| }, |
| "split": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "ort_default_axis0_uneven_scalar_split_float", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_DefaultAxis0UnevenSplitFloat" |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [5, 2], |
| "data": { "kind": "values", "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0] } |
| }, |
| "split": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "y2": { "dtype": "float32", "shape": [1, 2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "ort_positive_axis_scalar_split_rank3", |
| "attrs": { "axis": 2 }, |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_PositiveAxisScalarSplit" |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 2, 6], |
| "data": { |
| "kind": "values", |
| "values": { "$ref": "#/fixtureArrays/ort_positive_axis_scalar_split_rank3_input_input" } |
| } |
| }, |
| "split": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 2, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [2, 2, 2], "tolerance": 0.000001 }, |
| "y2": { "dtype": "float32", "shape": [2, 2, 2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "ort_positive_axis_uneven_scalar_split", |
| "attrs": { "axis": 1 }, |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_PositiveAxisUnevenSplit" |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 5], |
| "data": { "kind": "values", "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0] } |
| }, |
| "split": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 }, |
| "y2": { "dtype": "float32", "shape": [2, 1], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "ort_axis0_keepdims0_rank3", |
| "attrs": { "axis": 0, "keepdims": 0 }, |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_Axis0DefaultSplitFloatSetAxisExplicitlyDontKeepDims3Dim" |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4], |
| "data": { |
| "kind": "values", |
| "values": { "$ref": "#/fixtureArrays/ort_positive_axis_scalar_split_rank3_input_input" } |
| } |
| } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [3, 4], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [3, 4], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "ort_axis0_keepdims0_rank2", |
| "attrs": { "axis": 0, "keepdims": 0 }, |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_Axis0DefaultSplitFloatSetAxisExplicitlyDontKeepDims2Dim" |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3], |
| "data": { "kind": "values", "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0] } |
| } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [3], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [3], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "ort_default_axis0_four_outputs", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_Axis0DefaultSplitFloatSetAxisExplicitly", |
| "notes": "Valid ONNX sequence output modelled as four fixed tensors." |
| }, |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [4, 2], |
| "data": { "kind": "values", "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0] } |
| } |
| }, |
| "outputs": { |
| "y0": { |
| "dtype": "float32", |
| "shape": [1, 2], |
| "tolerance": 0.000001, |
| "data": { "kind": "values", "values": [1.0, 2.0] } |
| }, |
| "y1": { |
| "dtype": "float32", |
| "shape": [1, 2], |
| "tolerance": 0.000001, |
| "data": { "kind": "values", "values": [3.0, 4.0] } |
| }, |
| "y2": { |
| "dtype": "float32", |
| "shape": [1, 2], |
| "tolerance": 0.000001, |
| "data": { "kind": "values", "values": [5.0, 6.0] } |
| }, |
| "y3": { |
| "dtype": "float32", |
| "shape": [1, 2], |
| "tolerance": 0.000001, |
| "data": { "kind": "values", "values": [7.0, 8.0] } |
| } |
| } |
| }, |
| { |
| "name": "ort_bool_split_axis0_four_outputs", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_BoolSplit", |
| "notes": "Valid ONNX bool sequence output modelled as four fixed tensors." |
| }, |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { "dtype": "bool", "shape": [4, 2], "data": { "kind": "values", "values": [1, 1, 1, 1, 0, 0, 0, 0] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "bool", "shape": [1, 2], "tolerance": 0, "data": { "kind": "values", "values": [1, 1] } }, |
| "y1": { "dtype": "bool", "shape": [1, 2], "tolerance": 0, "data": { "kind": "values", "values": [1, 1] } }, |
| "y2": { "dtype": "bool", "shape": [1, 2], "tolerance": 0, "data": { "kind": "values", "values": [0, 0] } }, |
| "y3": { "dtype": "bool", "shape": [1, 2], "tolerance": 0, "data": { "kind": "values", "values": [0, 0] } } |
| } |
| }, |
| { |
| "name": "ort_positive_axis_keepdims0_four_outputs", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_PositiveAxisDontKeepDims", |
| "notes": "Valid ONNX keepdims=0 split along a non-leading axis, modelled as four fixed tensors." |
| }, |
| "attrs": { "axis": 2, "keepdims": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4], |
| "data": { |
| "kind": "values", |
| "values": { "$ref": "#/fixtureArrays/ort_positive_axis_scalar_split_rank3_input_input" } |
| } |
| } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 3], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [2, 3], "tolerance": 0.000001 }, |
| "y2": { "dtype": "float32", "shape": [2, 3], "tolerance": 0.000001 }, |
| "y3": { "dtype": "float32", "shape": [2, 3], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "ort_default_axis0_uneven_six_outputs", |
| "provenance": { |
| "source": "onnxruntime/test/providers/cpu/sequence/sequence_ops_test.cc", |
| "test": "SequenceOpsTest.SplitToSequence_DefaultAxis0UnevenSplitFloat2", |
| "notes": "Valid ONNX uneven scalar split producing six sequence elements." |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [17, 2], |
| "data": { |
| "kind": "values", |
| "values": [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0] |
| } |
| }, |
| "split": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [3] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "y2": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "y3": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "y4": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "y5": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "onnx_backend_split_scalar_axis1_three_outputs", |
| "attrs": { "axis": 1 }, |
| "provenance": { |
| "source": "cmake/external/onnx/onnx/backend/test/data/node/test_split_to_sequence_1", |
| "notes": "Official sequence output is represented as fixed tensor outputs y0/y1/y2; int64 split scalar is stored as uint32." |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [3, 6], |
| "data": { |
| "kind": "values", |
| "values": { "$ref": "#/fixtureArrays/onnx_backend_split_scalar_axis1_three_outputs_input_input" } |
| } |
| }, |
| "split": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 }, |
| "y2": { "dtype": "float32", "shape": [3, 2], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "onnx_backend_split_vector_axis0_two_outputs", |
| "attrs": { "axis": 0 }, |
| "provenance": { |
| "source": "cmake/external/onnx/onnx/backend/test/data/node/test_split_to_sequence_2", |
| "notes": "Official sequence output is represented as fixed tensor outputs y0/y1; int64 split vector is stored as uint32." |
| }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [3, 6], |
| "data": { |
| "kind": "values", |
| "values": { "$ref": "#/fixtureArrays/onnx_backend_split_scalar_axis1_three_outputs_input_input" } |
| } |
| }, |
| "split": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [1, 2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [1, 6], "tolerance": 0.000001 }, |
| "y1": { "dtype": "float32", "shape": [2, 6], "tolerance": 0.000001 } |
| } |
| }, |
| { |
| "name": "empty_zero_dim", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [6, 0], "data": { "kind": "values", "values": [] } }, |
| "split": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [2, 2, 2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 0], "tolerance": 0 }, |
| "y1": { "dtype": "float32", "shape": [2, 0], "tolerance": 0 }, |
| "y2": { "dtype": "float32", "shape": [2, 0], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "two_outputs_zero_dim", |
| "provenance": { |
| "notes": "Two-chunk split of a tensor with a zero-width trailing axis. The two-output kernel floors each chunk stride at one so a zero-width output cannot divide by zero; the existing zero-dim case splits into three chunks, which is a different kernel." |
| }, |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [4, 0], "data": { "kind": "values", "values": [] } }, |
| "split": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [2, 2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 0], "tolerance": 0 }, |
| "y1": { "dtype": "float32", "shape": [2, 0], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "two_outputs_zero_dim_squeeze", |
| "provenance": { |
| "notes": "Squeezed twin of the case above: keepdims=0 drops the split axis, which is a separate coordinate walk with its own chunk-stride floor. The zero-width trailing axis makes that floor fire." |
| }, |
| "attrs": { "axis": 0, "keepdims": 0 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [2, 3, 0], "data": { "kind": "values", "values": [] } } }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [3, 0], "tolerance": 0 }, |
| "y1": { "dtype": "float32", "shape": [3, 0], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "ort_caseB_empty", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [6, 2], |
| "data": { |
| "kind": "values", |
| "values": [0.15191493928432465, 0.14055313169956207, 0.12134978920221329, 0.10187792778015137, 0.08903708308935165, 0.08793208003044128, 0.1010342463850975, 0.1277448534965515, 0.1644245833158493, 0.20488715171813965, 0.24129025638103485, 0.2653009593486786] |
| } |
| }, |
| "split": { "dtype": "uint32", "shape": [0], "data": { "kind": "values", "values": [] } } |
| }, |
| "outputs": { |
| "y0": { |
| "dtype": "float32", |
| "shape": [1, 2], |
| "data": { "kind": "values", "values": [0.15191493928432465, 0.14055313169956207] }, |
| "tolerance": 0.001 |
| }, |
| "y1": { |
| "dtype": "float32", |
| "shape": [1, 2], |
| "data": { "kind": "values", "values": [0.12134978920221329, 0.10187792778015137] }, |
| "tolerance": 0.001 |
| }, |
| "y2": { |
| "dtype": "float32", |
| "shape": [1, 2], |
| "data": { "kind": "values", "values": [0.08903708308935165, 0.08793208003044128] }, |
| "tolerance": 0.001 |
| } |
| } |
| }, |
| { |
| "name": "keepdims0_axis2_three_outputs_squeeze_f32", |
| "attrs": { "axis": 2, "keepdims": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 3], |
| "data": { |
| "kind": "values", |
| "values": { "$ref": "#/fixtureArrays/onnx_backend_split_scalar_axis1_three_outputs_input_input" } |
| } |
| } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 3], "tolerance": 0 }, |
| "y1": { "dtype": "float32", "shape": [2, 3], "tolerance": 0 }, |
| "y2": { "dtype": "float32", "shape": [2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "negative_axis_neg1_scalar_split_rank3_f32", |
| "attrs": { "axis": -1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 2, 6], |
| "data": { |
| "kind": "values", |
| "values": { "$ref": "#/fixtureArrays/ort_positive_axis_scalar_split_rank3_input_input" } |
| } |
| }, |
| "split": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [2] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 2, 2], "tolerance": 0 }, |
| "y1": { "dtype": "float32", "shape": [2, 2, 2], "tolerance": 0 }, |
| "y2": { "dtype": "float32", "shape": [2, 2, 2], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "negative_axis_neg2_keepdims0_three_outputs_squeeze_f32", |
| "attrs": { "axis": -2, "keepdims": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4], |
| "data": { |
| "kind": "values", |
| "values": { "$ref": "#/fixtureArrays/ort_positive_axis_scalar_split_rank3_input_input" } |
| } |
| } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 4], "tolerance": 0 }, |
| "y1": { "dtype": "float32", "shape": [2, 4], "tolerance": 0 }, |
| "y2": { "dtype": "float32", "shape": [2, 4], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank6_middle_axis_two_outputs_deep_strides_f32", |
| "attrs": { "axis": 2 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 2, 4, 2, 2, 3], |
| "data": { "kind": "linspace", "start": 0.0, "end": 1.0 } |
| }, |
| "split": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [3, 1] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [2, 2, 3, 2, 2, 3], "tolerance": 0 }, |
| "y1": { "dtype": "float32", "shape": [2, 2, 1, 2, 2, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank7_two_outputs", |
| "attrs": { "axis": 5, "keepdims": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [1, 1, 1, 1, 1, 2, 3], |
| "data": { "kind": "linspace", "start": 0.0, "end": 5.0 } |
| }, |
| "split": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [1, 1] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [1, 1, 1, 1, 1, 1, 3], "tolerance": 0 }, |
| "y1": { "dtype": "float32", "shape": [1, 1, 1, 1, 1, 1, 3], "tolerance": 0 } |
| } |
| }, |
| { |
| "name": "rank8_two_outputs", |
| "attrs": { "axis": 6, "keepdims": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [1, 1, 1, 1, 1, 1, 2, 3], |
| "data": { "kind": "linspace", "start": 1.0, "end": 6.0 } |
| }, |
| "split": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [1, 1] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [1, 1, 1, 1, 1, 1, 1, 3], "tolerance": 0 }, |
| "y1": { "dtype": "float32", "shape": [1, 1, 1, 1, 1, 1, 1, 3], "tolerance": 0 } |
| } |
| } |
| ] |
| } |
|
|
