{ "op": "ai.onnx.ReverseSequence", "cases": [ { "name": "vec4_inner8_partial_reverse", "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [2, 5, 8], "data": { "kind": "fillFloat32", "sinStep": 0.11, "cosStep": 0.23 } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [3, 5] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 5, 8], "tolerance": 0 } } }, { "name": "vec4_empty_time_dim_inner4", "provenance": { "notes": "An empty tensor that still reaches the vec4 route. The existing empty_zero_dim cases have an innermost dimension of 1, so they fail the `dim(input, rank-1) % 4 == 0` gate and take the scalar fallback -- 0 satisfies that gate, but only the innermost dimension is tested by it. With the zero on the TIME axis and a four-aligned innermost dimension the outer stride is a product containing that zero, which is the case the `stride.value if stride.value > 0 else 1` guards in the coordinate decode and the source recomposition exist for; nothing rendered them before." }, "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [2, 0, 4], "data": { "kind": "values", "values": [] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 0] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 0, 4], "tolerance": 0 } } }, { "name": "vec4_f16_inner4_partial_reverse", "provenance": { "notes": "The f16 half of the vec4 route. float16_default_axes has an innermost dimension of 1 and so takes the scalar fallback, leaving `enable f16` and the f16 element types unrendered on this shader even though the type constraints admit float16." }, "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "float16", "shape": [2, 3, 4], "data": { "kind": "cycle", "values": [0.5, -1.25, 2.0, -0.75, 3.5, -2.5] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [2, 3] } } }, "outputs": { "y": { "dtype": "float16", "shape": [2, 3, 4], "tolerance": 0 } } }, { "name": "dispatch_cliff_rankn_f32", "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [1, 16777216], "data": { "kind": "cycle", "values": [-1.0, 0.0, 1.0, 2.0] } }, "sequence_lens": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [1] } } }, "outputs": { "y": { "dtype": "float32", "shape": [1, 16777216], "tolerance": 0 } } }, { "name": "batch_major_int32", "attrs": { "batch_axis": 0, "time_axis": 1 }, "provenance": { "source": "onnxruntime/test/providers/cpu/tensor/reverse_sequence_test.cc", "test": "ReverseSequenceTest.BatchMajor", "notes": "Adapted from int64 to supported int32 data and lengths." }, "inputs": { "input": { "dtype": "int32", "shape": [2, 4, 1], "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [4, 3] } } }, "outputs": { "y": { "dtype": "int32", "shape": [2, 4, 1] } } }, { "name": "time_major_int32", "attrs": { "batch_axis": 1, "time_axis": 0 }, "provenance": { "source": "onnxruntime/test/providers/cpu/tensor/reverse_sequence_test.cc", "test": "ReverseSequenceTest.TimeMajor", "notes": "Adapted from int64 to supported int32 data and lengths." }, "inputs": { "input": { "dtype": "int32", "shape": [4, 2, 1], "data": { "kind": "values", "values": [0, 4, 1, 5, 2, 6, 3, 7] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [4, 3] } } }, "outputs": { "y": { "dtype": "int32", "shape": [4, 2, 1] } } }, { "name": "larger_trailing_dim", "attrs": { "batch_axis": 0, "time_axis": 1 }, "provenance": { "source": "onnxruntime/test/providers/cpu/tensor/reverse_sequence_test.cc", "test": "ReverseSequenceTest.LargerDim2" }, "inputs": { "input": { "dtype": "float32", "shape": [2, 3, 2], "data": { "kind": "values", "values": [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [2, 3] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 3, 2] } } }, { "name": "zero_sequence_length", "attrs": { "batch_axis": 1, "time_axis": 0 }, "provenance": { "source": "onnxruntime/test/providers/cpu/tensor/reverse_sequence_test.cc", "test": "ReverseSequenceTest.ZeroSequenceLength" }, "inputs": { "input": { "dtype": "float32", "shape": [5, 3, 1], "data": { "kind": "values", "values": [10.0, 20.0, 30.0, 11.0, 21.0, 31.0, 12.0, 22.0, 32.0, 13.0, 23.0, 33.0, 14.0, 24.0, 34.0] } }, "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [4, 0, 2] } } }, "outputs": { "y": { "dtype": "float32", "shape": [5, 3, 1] } } }, { "name": "float16_default_axes", "inputs": { "input": { "dtype": "float16", "shape": [3, 2, 1], "data": { "kind": "values", "values": [1.0, 10.0, 2.0, 11.0, 3.0, 12.0] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [3, 1] } } }, "outputs": { "y": { "dtype": "float16", "shape": [3, 2, 1], "tolerance": 0.001 } } }, { "name": "onnx_backend_rank2_time_axis0", "attrs": { "time_axis": 0, "batch_axis": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [4, 4], "data": { "kind": "values", "values": [0.0, 4.0, 8.0, 12.0, 1.0, 5.0, 9.0, 13.0, 2.0, 6.0, 10.0, 14.0, 3.0, 7.0, 11.0, 15.0] } }, "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [4, 3, 2, 1] } } }, "outputs": { "y": { "dtype": "float32", "shape": [4, 4] } }, "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_reversesequence_time", "notes": "Official fixture sequence_lens is int64; this framework stores lengths as int32." } }, { "name": "onnx_backend_rank2_time_axis1", "attrs": { "time_axis": 1, "batch_axis": 0 }, "inputs": { "input": { "dtype": "float32", "shape": [4, 4], "data": { "kind": "values", "values": [0.0, 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] } }, "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [1, 2, 3, 4] } } }, "outputs": { "y": { "dtype": "float32", "shape": [4, 4] } }, "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_reversesequence_batch", "notes": "Official fixture sequence_lens is int64; this framework stores lengths as int32." } }, { "name": "uint8_batch_major_edge_values", "attrs": { "batch_axis": 0, "time_axis": 1 }, "provenance": { "source": "onnxruntime/test/providers/cpu/tensor/reverse_sequence_test.cc", "test": "ReverseSequenceTest.BatchMajor", "notes": "Additional valid type edge: logical uint8 data is reversed using the same batch-major addressing as ORT's BatchMajor case." }, "inputs": { "input": { "dtype": "uint8", "shape": [2, 4, 1], "data": { "kind": "values", "values": [250, 1, 2, 3, 4, 5, 6, 255] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [4, 2] } } }, "outputs": { "y": { "dtype": "uint8", "shape": [2, 4, 1], "tolerance": 0 } } }, { "name": "int8_time_major_negative_values", "attrs": { "batch_axis": 1, "time_axis": 0 }, "provenance": { "source": "onnxruntime/test/providers/cpu/tensor/reverse_sequence_test.cc", "test": "ReverseSequenceTest.TimeMajor", "notes": "Additional valid type edge: logical int8 data, including -128 and 127, follows ORT's TimeMajor addressing." }, "inputs": { "input": { "dtype": "int8", "shape": [4, 2, 1], "data": { "kind": "values", "values": [-128, 10, -2, 20, 0, 30, 127, 40] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [3, 1] } } }, "outputs": { "y": { "dtype": "int8", "shape": [4, 2, 1], "tolerance": 0 } } }, { "name": "ort_bool_batch_major", "attrs": { "batch_axis": 0, "time_axis": 1 }, "provenance": { "source": "onnxruntime/test/providers/cpu/tensor/reverse_sequence_test.cc", "test": "ReverseSequenceTest.BatchMajor", "notes": "ONNX ReverseSequence permits bool payload tensors, which this fixture exercises batch-major." }, "inputs": { "input": { "dtype": "bool", "shape": [2, 4, 1], "data": { "kind": "values", "values": [1, 0, 1, 0, 0, 1, 0, 1] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [3, 2] } } }, "outputs": { "y": { "dtype": "bool", "shape": [2, 4, 1], "tolerance": 0 } } }, { "name": "ort_int16_time_major", "attrs": { "batch_axis": 1, "time_axis": 0 }, "provenance": { "source": "onnxruntime/test/providers/cpu/tensor/reverse_sequence_test.cc", "test": "ReverseSequenceTest.TimeMajor", "notes": "ONNX ReverseSequence int16 coverage with min/max sentinels on the time-major axis." }, "inputs": { "input": { "dtype": "int16", "shape": [4, 2, 1], "data": { "kind": "values", "values": [-32768, 10, -2, 20, 0, 30, 32767, 40] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [4, 2] } } }, "outputs": { "y": { "dtype": "int16", "shape": [4, 2, 1], "tolerance": 0 } } }, { "name": "uint32_batch_major_extreme_payloads", "provenance": { "source": "onnx/onnx/docs/Operators.md#ReverseSequence-10", "notes": "Covers standard uint32 data with UINT32_MAX and values across the signed boundary while using the existing runtime index projection." }, "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "uint32", "shape": [2, 3, 1], "data": { "kind": "values", "values": [0, 4294967295, 2147483648, 1, 16777217, 4000000000] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [3, 2] } } }, "outputs": { "y": { "dtype": "uint32", "shape": [2, 3, 1], "tolerance": 0 } } }, { "name": "rank6_default_axes", "provenance": { "source": "onnxruntime/test/providers/cpu/tensor/reverse_sequence_test.cc", "notes": "Additional rank edge inside this framework's supported rank range; uses ONNX defaults batch_axis=1,time_axis=0." }, "inputs": { "input": { "dtype": "float32", "shape": [3, 2, 1, 2, 1, 2], "data": { "kind": "values", "values": [0.0, 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] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [2, 3] } } }, "outputs": { "y": { "dtype": "float32", "shape": [3, 2, 1, 2, 1, 2], "tolerance": 0 } } }, { "name": "empty_zero_dim", "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "int32", "shape": [2, 0, 1], "data": { "kind": "values", "values": [] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 0] } } }, "outputs": { "y": { "dtype": "int32", "shape": [2, 0, 1], "tolerance": 0 } } }, { "name": "empty_zero_dim_f32", "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [2, 0, 1], "data": { "kind": "values", "values": [] } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 0] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 0, 1], "tolerance": 0 } } }, { "name": "scalar_fallback_inner_not_mul4_partial_reverse", "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [2, 4, 5], "data": { "kind": "fillFloat32", "sinStep": 0.11, "cosStep": 0.23 } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [3, 4] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 4, 5], "tolerance": 0 } } }, { "name": "scalar_fallback_rank2_time_is_inner_axis", "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [3, 6], "data": { "kind": "fillFloat32", "sinStep": 0.07, "cosStep": 0.19 } }, "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [6, 4, 2] } } }, "outputs": { "y": { "dtype": "float32", "shape": [3, 6], "tolerance": 0 } } }, { "name": "vec4_len_exceeds_timedim_clamp", "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [2, 5, 8], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29 } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [99, 100] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 5, 8], "tolerance": 0 } } }, { "name": "rank7_batch_major", "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [2, 3, 1, 1, 1, 1, 2], "data": { "kind": "linspace", "start": 0.0, "end": 11.0 } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [2, 3] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 3, 1, 1, 1, 1, 2], "tolerance": 0 } } }, { "name": "rank8_batch_major", "attrs": { "batch_axis": 0, "time_axis": 1 }, "inputs": { "input": { "dtype": "float32", "shape": [2, 3, 1, 1, 1, 1, 2, 2], "data": { "kind": "linspace", "start": 1.0, "end": 24.0 } }, "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [3, 2] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 3, 1, 1, 1, 1, 2, 2], "tolerance": 0 } } } ] }