| { |
| "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": "Shape [2,0,4] has an empty time axis and a four-aligned inner dimension. Vectorized coordinate decomposition must handle the resulting zero outer stride and produce the empty output without division by zero." |
| }, |
| "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": "A four-element innermost dimension with float16 data and sequence lengths 2 and 3 over three time steps checks partial and full reversal independently in each batch." |
| }, |
| "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": "Projects representable int64 sequence lengths to int32 storage." |
| } |
| }, |
| { |
| "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": "Projects representable int64 sequence lengths to int32 storage." |
| } |
| }, |
| { |
| "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": "Uint32 payloads include UINT32_MAX and values on both sides of the signed boundary, while sequence lengths reverse different prefixes in each batch." |
| }, |
| "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": "A rank-6 input exercises the ONNX defaults batch_axis=1 and 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 } } |
| }, |
| { |
| "name": "output_vectors_float32_axis0_4x5x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [0, 1, 2, 5] } } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_float32_axis0_2x3x4x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_float32_axis1_4x5x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [5], "data": { "kind": "values", "values": [0, 1, 2, 4, 4] } } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_float32_axis1_2x3x4x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [0, 1, 2] } } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_float16_axis0_4x5x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float16", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [0, 1, 2, 5] } } |
| }, |
| "outputs": { "y": { "dtype": "float16", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_float16_axis0_2x3x4x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float16", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } |
| }, |
| "outputs": { "y": { "dtype": "float16", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_float16_axis1_4x5x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float16", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [5], "data": { "kind": "values", "values": [0, 1, 2, 4, 4] } } |
| }, |
| "outputs": { "y": { "dtype": "float16", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_float16_axis1_2x3x4x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "float16", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [0, 1, 2] } } |
| }, |
| "outputs": { "y": { "dtype": "float16", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int32_axis0_4x5x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "int32", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [-2147483648, 2147483647, 0, -7, 3, 11, 17, -23, 31, 127, -255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [0, 1, 2, 5] } } |
| }, |
| "outputs": { "y": { "dtype": "int32", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int32_axis0_2x3x4x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "int32", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [-2147483648, 2147483647, 0, -7, 3, 11, 17, -23, 31, 127, -255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } |
| }, |
| "outputs": { "y": { "dtype": "int32", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int32_axis1_4x5x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "int32", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [-2147483648, 2147483647, 0, -7, 3, 11, 17, -23, 31, 127, -255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [5], "data": { "kind": "values", "values": [0, 1, 2, 4, 4] } } |
| }, |
| "outputs": { "y": { "dtype": "int32", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int32_axis1_2x3x4x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "int32", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [-2147483648, 2147483647, 0, -7, 3, 11, 17, -23, 31, 127, -255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [0, 1, 2] } } |
| }, |
| "outputs": { "y": { "dtype": "int32", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_uint32_axis0_4x5x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint32", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [4294967295, 0, 2147483648, 7, 3, 11, 17, 23, 31, 127, 255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [0, 1, 2, 5] } } |
| }, |
| "outputs": { "y": { "dtype": "uint32", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_uint32_axis0_2x3x4x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint32", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [4294967295, 0, 2147483648, 7, 3, 11, 17, 23, 31, 127, 255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } |
| }, |
| "outputs": { "y": { "dtype": "uint32", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_uint32_axis1_4x5x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint32", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [4294967295, 0, 2147483648, 7, 3, 11, 17, 23, 31, 127, 255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [5], "data": { "kind": "values", "values": [0, 1, 2, 4, 4] } } |
| }, |
| "outputs": { "y": { "dtype": "uint32", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_uint32_axis1_2x3x4x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint32", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [4294967295, 0, 2147483648, 7, 3, 11, 17, 23, 31, 127, 255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [0, 1, 2] } } |
| }, |
| "outputs": { "y": { "dtype": "uint32", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int16_axis0_4x5x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "int16", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [-32768, 32767, 0, -7, 3, 11, 17, -23, 31, 127, -255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [0, 1, 2, 5] } } |
| }, |
| "outputs": { "y": { "dtype": "int16", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int16_axis0_2x3x4x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "int16", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [-32768, 32767, 0, -7, 3, 11, 17, -23, 31, 127, -255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } |
| }, |
| "outputs": { "y": { "dtype": "int16", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int16_axis1_4x5x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "int16", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [-32768, 32767, 0, -7, 3, 11, 17, -23, 31, 127, -255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [5], "data": { "kind": "values", "values": [0, 1, 2, 4, 4] } } |
| }, |
| "outputs": { "y": { "dtype": "int16", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int16_axis1_2x3x4x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "int16", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [-32768, 32767, 0, -7, 3, 11, 17, -23, 31, 127, -255] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [0, 1, 2] } } |
| }, |
| "outputs": { "y": { "dtype": "int16", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_uint8_axis0_4x5x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint8", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [255, 0, 128, 7, 3, 11, 17, 23, 31, 127, 254] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [0, 1, 2, 5] } } |
| }, |
| "outputs": { "y": { "dtype": "uint8", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_uint8_axis0_2x3x4x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint8", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [255, 0, 128, 7, 3, 11, 17, 23, 31, 127, 254] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } |
| }, |
| "outputs": { "y": { "dtype": "uint8", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_uint8_axis1_4x5x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint8", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [255, 0, 128, 7, 3, 11, 17, 23, 31, 127, 254] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [5], "data": { "kind": "values", "values": [0, 1, 2, 4, 4] } } |
| }, |
| "outputs": { "y": { "dtype": "uint8", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_uint8_axis1_2x3x4x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "uint8", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [255, 0, 128, 7, 3, 11, 17, 23, 31, 127, 254] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [0, 1, 2] } } |
| }, |
| "outputs": { "y": { "dtype": "uint8", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int8_axis0_4x5x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "int8", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [-128, 127, 0, -7, 3, 11, 17, -23, 31, 63, -64] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [0, 1, 2, 5] } } |
| }, |
| "outputs": { "y": { "dtype": "int8", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int8_axis0_2x3x4x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "int8", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [-128, 127, 0, -7, 3, 11, 17, -23, 31, 63, -64] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } |
| }, |
| "outputs": { "y": { "dtype": "int8", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int8_axis1_4x5x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "int8", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [-128, 127, 0, -7, 3, 11, 17, -23, 31, 63, -64] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [5], "data": { "kind": "values", "values": [0, 1, 2, 4, 4] } } |
| }, |
| "outputs": { "y": { "dtype": "int8", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_int8_axis1_2x3x4x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "int8", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [-128, 127, 0, -7, 3, 11, 17, -23, 31, 63, -64] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [0, 1, 2] } } |
| }, |
| "outputs": { "y": { "dtype": "int8", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_bool_axis0_4x5x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "bool", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [0, 1, 1, 0, 1, 0, 0, 1, 1, 1, 0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [0, 1, 2, 5] } } |
| }, |
| "outputs": { "y": { "dtype": "bool", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_bool_axis0_2x3x4x3", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "bool", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [0, 1, 1, 0, 1, 0, 0, 1, 1, 1, 0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } |
| }, |
| "outputs": { "y": { "dtype": "bool", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_bool_axis1_4x5x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "bool", |
| "shape": [4, 5, 3], |
| "data": { "kind": "cycle", "values": [0, 1, 1, 0, 1, 0, 0, 1, 1, 1, 0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [5], "data": { "kind": "values", "values": [0, 1, 2, 4, 4] } } |
| }, |
| "outputs": { "y": { "dtype": "bool", "shape": [4, 5, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_bool_axis1_2x3x4x3", |
| "attrs": { "batch_axis": 1, "time_axis": 0 }, |
| "inputs": { |
| "input": { |
| "dtype": "bool", |
| "shape": [2, 3, 4, 3], |
| "data": { "kind": "cycle", "values": [0, 1, 1, 0, 1, 0, 0, 1, 1, 1, 0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [0, 1, 2] } } |
| }, |
| "outputs": { "y": { "dtype": "bool", "shape": [2, 3, 4, 3], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_special_rank2_float32", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [3, 8], |
| "data": { "kind": "cycle", "values": ["NaN", -0.0, 0.125, -1.0, 2.0, 3.0, 4.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [8, 3, 0] } } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [3, 8], "tolerance": 0, "allowNaN": true } } |
| }, |
| { |
| "name": "output_vectors_fold_float32_batch4_wg8", |
| "provenance": { |
| "notes": "The output-vector grid crosses the dispatch fold. Different sequence lengths and an inner extent not divisible by four expose incorrect source or folded-output offsets." |
| }, |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "tunables": { "WORKGROUP_SIZE": 8 }, |
| "inputs": { |
| "input": { |
| "dtype": "float32", |
| "shape": [4, 8193, 65], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [0, 1, 4096, 8193] } } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [4, 8193, 65], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_special_rank2_float16", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { |
| "dtype": "float16", |
| "shape": [3, 8], |
| "data": { "kind": "cycle", "values": ["NaN", -0.0, 0.125, -1.0, 2.0, 3.0, 4.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [8, 3, 0] } } |
| }, |
| "outputs": { "y": { "dtype": "float16", "shape": [3, 8], "tolerance": 0, "allowNaN": true } } |
| }, |
| { |
| "name": "output_vectors_fold_float16_batch4_wg8", |
| "provenance": { |
| "notes": "The output-vector grid crosses the dispatch fold. Different sequence lengths and an inner extent not divisible by four expose incorrect source or folded-output offsets." |
| }, |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "tunables": { "WORKGROUP_SIZE": 8 }, |
| "inputs": { |
| "input": { |
| "dtype": "float16", |
| "shape": [4, 8193, 65], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { "dtype": "uint32", "shape": [4], "data": { "kind": "values", "values": [0, 1, 4096, 8193] } } |
| }, |
| "outputs": { "y": { "dtype": "float16", "shape": [4, 8193, 65], "tolerance": 0 } } |
| }, |
| { |
| "name": "output_vectors_fold_float16_batch8_wg8", |
| "provenance": { |
| "notes": "The output-vector grid crosses the dispatch fold. Different sequence lengths and an inner extent not divisible by four expose incorrect source or folded-output offsets." |
| }, |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "tunables": { "WORKGROUP_SIZE": 8 }, |
| "inputs": { |
| "input": { |
| "dtype": "float16", |
| "shape": [8, 8193, 65], |
| "data": { "kind": "cycle", "values": [0.125, -0.25, 0.5, 1.0, 2.0, -4.0, 8.0, 0.75, -1.5, 3.0, 7.0] } |
| }, |
| "sequence_lens": { |
| "dtype": "uint32", |
| "shape": [8], |
| "data": { "kind": "values", "values": [0, 1, 4096, 8193, 0, 1, 4096, 8193] } |
| } |
| }, |
| "outputs": { "y": { "dtype": "float16", "shape": [8, 8193, 65], "tolerance": 0 } } |
| } |
| ] |
| } |
|
|
