{ "cases": [ { "name": "f32_values", "attrs": { "bias": 0.25, "lambd": 0.5 }, "inputs": { "x": { "dtype": "float32", "shape": [6], "data": { "kind": "values", "values": [-2.0, -0.75, -0.25, 0.25, 0.75, 2.0] } } }, "outputs": { "y": { "dtype": "float32", "shape": [6], "tolerance": 0.000001 } } }, { "name": "f32_zero_threshold_subnormal_gpu_gap", "skipGpu": { "category": "permanent", "reason": "Portable WGSL floating-point semantics do not guarantee preservation of the subnormal values required by this fixture. Backend evidence: WebGPU/Metal flushes subnormals to zero (f32 and f16); the kernel cannot preserve denormal inputs/outputs bit-exactly." }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "ShrinkOpTest", "notes": "With lambd=0 and bias=0, every nonzero value should pass through, including subnormal values on both sides of zero." }, "attrs": { "bias": 0, "lambd": 0 }, "inputs": { "x": { "dtype": "float32", "shape": [5], "data": { "kind": "values", "values": [-1e-39, -1e-40, 0.0, 1e-40, 1e-39] } } }, "outputs": { "y": { "dtype": "float32", "shape": [5], "tolerance": 0 } } }, { "name": "f32_zero_threshold_subnormal_vec4_gpu_gap", "skipGpu": { "category": "permanent", "reason": "Portable WGSL floating-point semantics do not guarantee preservation of the subnormal values required by this fixture. Backend evidence: WebGPU/Metal flushes subnormals to zero (f32 and f16); the kernel cannot preserve denormal inputs/outputs bit-exactly." }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "ShrinkOpTest", "notes": "On the vectorized path with lambd=0 and bias=0, nonzero subnormal values should pass through unchanged." }, "attrs": { "bias": 0, "lambd": 0 }, "inputs": { "x": { "dtype": "float32", "shape": [4], "data": { "kind": "values", "values": [-1e-39, -1e-40, 1e-40, 1e-39] } } }, "outputs": { "y": { "dtype": "float32", "shape": [4], "tolerance": 0 } } }, { "name": "f16_values", "attrs": { "bias": 0.25, "lambd": 0.5 }, "inputs": { "x": { "dtype": "float16", "shape": [2, 3], "data": { "kind": "values", "values": [-2.0, -0.75, -0.25, 0.25, 0.75, 2.0] } } }, "outputs": { "y": { "dtype": "float16", "shape": [2, 3] } }, "tolerance": 0.001 }, { "name": "onnx_hard_shrink", "attrs": { "lambd": 1.5 }, "inputs": { "x": { "dtype": "float32", "shape": [5], "data": { "kind": "values", "values": [-2.0, -1.0, 0.0, 1.0, 2.0] } } }, "outputs": { "y": { "dtype": "float32", "shape": [5], "tolerance": 0.000001 } }, "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_shrink_hard" } }, { "name": "onnx_soft_shrink", "attrs": { "bias": 1.5, "lambd": 1.5 }, "inputs": { "x": { "dtype": "float32", "shape": [5], "data": { "kind": "values", "values": [-2.0, -1.0, 0.0, 1.0, 2.0] } } }, "outputs": { "y": { "dtype": "float32", "shape": [5], "tolerance": 0.000001 } }, "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_shrink_soft" } }, { "name": "default_boundary_nonfinite", "inputs": { "x": { "dtype": "float32", "shape": [7], "data": { "kind": "values", "values": ["-Infinity", -0.5, -0.50000006, 0.0, 0.5, 0.50000006, "Infinity"] } } }, "outputs": { "y": { "dtype": "float32", "shape": [7], "tolerance": 0.000001 } } }, { "name": "ort_float_default_attrs_2x2", "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkFloatType", "notes": "GenerateSignedTestCases(), default attributes." }, "inputs": { "x": { "dtype": "float32", "shape": [2, 2], "data": { "kind": "values", "values": [-1.0, 0.0, 0.0, 1.0] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 } } }, { "name": "ort_float_nondefault_attrs_2x2", "attrs": { "bias": 10, "lambd": 2 }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkFloatType", "notes": "GenerateSignedTestCases(), non-default attributes." }, "inputs": { "x": { "dtype": "float32", "shape": [2, 2], "data": { "kind": "values", "values": [-3.0, -1.0, 1.0, 4.0] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 2], "tolerance": 0.000001 } } }, { "name": "ort_f16_nondefault_attrs_2x2", "attrs": { "bias": 10, "lambd": 2 }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkMLFloat16Type", "notes": "MLFloat16 non-default attributes." }, "inputs": { "x": { "dtype": "float16", "shape": [2, 2], "data": { "kind": "values", "values": [-3.0, -1.0, 1.0, 4.0] } } }, "outputs": { "y": { "dtype": "float16", "shape": [2, 2], "tolerance": 0.001 } } }, { "name": "ort_f16_default_attrs_2x2", "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkMLFloat16Type", "notes": "MLFloat16 default attributes." }, "inputs": { "x": { "dtype": "float16", "shape": [2, 2], "data": { "kind": "values", "values": [-1.0, 0.0, 0.0, 1.0] } } }, "outputs": { "y": { "dtype": "float16", "shape": [2, 2], "tolerance": 0.001 } } }, { "name": "ort_int32_default_attrs_2x2", "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkInt32Type", "notes": "GenerateSignedTestCases(), default attributes." }, "inputs": { "x": { "dtype": "int32", "shape": [2, 2], "data": { "kind": "values", "values": [-1, 0, 0, 1] } } }, "outputs": { "y": { "dtype": "int32", "shape": [2, 2], "tolerance": 0 } } }, { "name": "ort_int32_nondefault_attrs_2x2", "attrs": { "bias": 10, "lambd": 2 }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkInt32Type", "notes": "GenerateSignedTestCases(), non-default attributes." }, "inputs": { "x": { "dtype": "int32", "shape": [2, 2], "data": { "kind": "values", "values": [-3, -1, 1, 4] } } }, "outputs": { "y": { "dtype": "int32", "shape": [2, 2], "tolerance": 0 } } }, { "name": "ort_int16_default_attrs_2x2", "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkInt16Type", "notes": "GenerateSignedTestCases(), default attributes." }, "inputs": { "x": { "dtype": "int16", "shape": [2, 2], "data": { "kind": "values", "values": [-1, 0, 0, 1] } } }, "outputs": { "y": { "dtype": "int16", "shape": [2, 2], "tolerance": 0 } } }, { "name": "ort_int16_nondefault_attrs_2x2", "attrs": { "bias": 10, "lambd": 2 }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkInt16Type", "notes": "GenerateSignedTestCases(), non-default attributes." }, "inputs": { "x": { "dtype": "int16", "shape": [2, 2], "data": { "kind": "values", "values": [-3, -1, 1, 4] } } }, "outputs": { "y": { "dtype": "int16", "shape": [2, 2], "tolerance": 0 } } }, { "name": "ort_uint32_nondefault_attrs_2x2", "attrs": { "bias": 10, "lambd": 2 }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkUint32Type", "notes": "GenerateUnsignedTestCases(), non-default attributes." }, "inputs": { "x": { "dtype": "uint32", "shape": [2, 2], "data": { "kind": "values", "values": [37, 1, 1, 11] } } }, "outputs": { "y": { "dtype": "uint32", "shape": [2, 2], "tolerance": 0 } } }, { "name": "ort_uint32_default_attrs_2x2", "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkUint32Type", "notes": "GenerateUnsignedTestCases(), default attributes." }, "inputs": { "x": { "dtype": "uint32", "shape": [2, 2], "data": { "kind": "values", "values": [0, 0, 0, 1] } } }, "outputs": { "y": { "dtype": "uint32", "shape": [2, 2], "tolerance": 0 } } }, { "name": "ort_int8_default_attrs_2x2", "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkInt8Type", "notes": "GenerateSignedTestCases(), default attributes using logical int8 storage." }, "inputs": { "x": { "dtype": "int8", "shape": [2, 2], "data": { "kind": "values", "values": [-1, 0, 0, 1] } } }, "outputs": { "y": { "dtype": "int8", "shape": [2, 2], "tolerance": 0 } } }, { "name": "ort_int8_nondefault_attrs_2x2", "attrs": { "bias": 10, "lambd": 2 }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkInt8Type", "notes": "GenerateSignedTestCases(), non-default attributes using logical int8 storage." }, "inputs": { "x": { "dtype": "int8", "shape": [2, 2], "data": { "kind": "values", "values": [-3, -1, 1, 4] } } }, "outputs": { "y": { "dtype": "int8", "shape": [2, 2], "tolerance": 0 } } }, { "name": "ort_uint8_nondefault_attrs_2x2", "attrs": { "bias": 10, "lambd": 2 }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkUint8Type", "notes": "GenerateUnsignedTestCases(), non-default attributes using logical uint8 storage." }, "inputs": { "x": { "dtype": "uint8", "shape": [2, 2], "data": { "kind": "values", "values": [37, 1, 1, 11] } } }, "outputs": { "y": { "dtype": "uint8", "shape": [2, 2], "tolerance": 0 } } }, { "name": "ort_uint8_default_attrs_2x2", "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkUint8Type", "notes": "GenerateUnsignedTestCases(), default attributes using logical uint8 storage." }, "inputs": { "x": { "dtype": "uint8", "shape": [2, 2], "data": { "kind": "values", "values": [0, 0, 0, 1] } } }, "outputs": { "y": { "dtype": "uint8", "shape": [2, 2], "tolerance": 0 } } }, { "name": "ort_int32_float_attr_rounding_boundary", "attrs": { "bias": 1, "lambd": 1 }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkInt32Type", "notes": "Integer Shrink uses float-valued attributes; this documents ORT's f32 boundary behavior above 2^24." }, "inputs": { "x": { "dtype": "int32", "shape": [7], "data": { "kind": "values", "values": [-16777218, -16777217, -16777216, 0, 16777216, 16777217, 16777218] } } }, "outputs": { "y": { "dtype": "int32", "shape": [7], "data": { "kind": "values", "values": [-16777216, -16777215, -16777215, 0, 16777215, 16777215, 16777216] }, "tolerance": 0 } } }, { "name": "ort_uint32_float_attr_rounding_boundary", "attrs": { "bias": 1, "lambd": 1 }, "provenance": { "source": "onnxruntime/test/providers/cpu/nn/shrink_test.cc", "test": "MathOpTest.ShrinkUint32Type", "notes": "Unsigned integer Shrink uses float-valued attributes; this documents ORT's f32 boundary behavior above 2^24." }, "inputs": { "x": { "dtype": "uint32", "shape": [7], "data": { "kind": "values", "values": [0, 1, 2, 16777216, 16777217, 16777218, 4000000000] } } }, "outputs": { "y": { "dtype": "uint32", "shape": [7], "data": { "kind": "values", "values": [0, 0, 1, 16777215, 16777215, 16777216, 4000000000] }, "tolerance": 0 } } }, { "name": "onnx_backend_shrink_hard", "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_shrink_hard" }, "attrs": { "lambd": 1.5 }, "inputs": { "x": { "dtype": "float32", "shape": [5], "data": { "kind": "values", "values": [-2.0, -1.0, 0.0, 1.0, 2.0] } } }, "outputs": { "y": { "dtype": "float32", "shape": [5], "tolerance": 0.00001 } } }, { "name": "onnx_backend_shrink_soft", "provenance": { "source": "cmake/external/onnx/onnx/backend/test/data/node/test_shrink_soft" }, "attrs": { "bias": 1.5, "lambd": 1.5 }, "inputs": { "x": { "dtype": "float32", "shape": [5], "data": { "kind": "values", "values": [-2.0, -1.0, 0.0, 1.0, 2.0] } } }, "outputs": { "y": { "dtype": "float32", "shape": [5], "tolerance": 0.00001 } } }, { "name": "empty_input_zero_dim", "attrs": { "bias": 0.25, "lambd": 0.5 }, "inputs": { "x": { "dtype": "float32", "shape": [0], "data": { "kind": "values", "values": [] } } }, "outputs": { "y": { "dtype": "float32", "shape": [0], "tolerance": 0 } } }, { "name": "negative_bias_threshold_boundary_vec4", "attrs": { "bias": -0.25, "lambd": 0.5 }, "inputs": { "x": { "dtype": "float32", "shape": [8], "data": { "kind": "values", "values": [-0.75, -0.5, -0.49999997, 0.0, 0.49999997, 0.5, 0.75, 1.5] } } }, "outputs": { "y": { "dtype": "float32", "shape": [8], "tolerance": 0.000001 } } }, { "name": "f16_threshold_boundary_vec4", "attrs": { "bias": 0.25, "lambd": 0.5 }, "inputs": { "x": { "dtype": "float16", "shape": [8], "data": { "kind": "values", "values": [-1.5, -0.5, -0.4990234375, 0.0, 0.4990234375, 0.5, 0.50048828125, 1.5] } } }, "outputs": { "y": { "dtype": "float16", "shape": [8], "tolerance": 0.001 } } }, { "name": "f32_rank4_nondefault_attrs", "attrs": { "bias": 1.5, "lambd": 1.5 }, "inputs": { "x": { "dtype": "float32", "shape": [2, 3, 4, 5], "data": { "kind": "values", "values": [-3.0, -2.5, -1.5, -1.0, -0.5, 0.0, 0.5, 1.0, 1.5, 2.5, 3.0, -3.0, -2.5, -1.5, -1.0, -0.5, 0.0, 0.5, 1.0, 1.5, 2.5, 3.0, -2.0, 2.0, -1.6, 1.6, -1.4, 1.4, -0.1, 0.1, -3.0, -2.5, -1.5, -1.0, -0.5, 0.0, 0.5, 1.0, 1.5, 2.5, 3.0, -3.0, -2.5, -1.5, -1.0, -0.5, 0.0, 0.5, 1.0, 1.5, 2.5, 3.0, -2.0, 2.0, -1.6, 1.6, -1.4, 1.4, -0.1, 0.1, -3.0, -2.5, -1.5, -1.0, -0.5, 0.0, 0.5, 1.0, 1.5, 2.5, 3.0, -3.0, -2.5, -1.5, -1.0, -0.5, 0.0, 0.5, 1.0, 1.5, 2.5, 3.0, -2.0, 2.0, -1.6, 1.6, -1.4, 1.4, -0.1, 0.1, -3.0, -2.5, -1.5, -1.0, -0.5, 0.0, 0.5, 1.0, 1.5, 2.5, 3.0, -3.0, -2.5, -1.5, -1.0, -0.5, 0.0, 0.5, 1.0, 1.5, 2.5, 3.0, -2.0, 2.0, -1.6, 1.6, -1.4, 1.4, -0.1, 0.1] } } }, "outputs": { "y": { "dtype": "float32", "shape": [2, 3, 4, 5], "tolerance": 0.000001 } } }, { "name": "int8_nonmult4_scalar_path_boundary", "attrs": { "bias": 0.5, "lambd": 0.5 }, "inputs": { "x": { "dtype": "int8", "shape": [5], "data": { "kind": "values", "values": [-2, -1, 0, 1, 2] } } }, "outputs": { "y": { "dtype": "int8", "shape": [5], "tolerance": 0 } } }, { "name": "f32_nan_input_with_nonzero_bias", "attrs": { "bias": 1, "lambd": 0.5 }, "inputs": { "x": { "dtype": "float32", "shape": [8], "data": { "kind": "values", "values": ["NaN", "-Infinity", -1.0, -0.5, 0.5, 1.0, "Infinity", "NaN"] } } }, "outputs": { "y": { "dtype": "float32", "shape": [8], "tolerance": 0.000001, "allowNaN": true } } }, { "name": "uint8_empty_zero_dim", "inputs": { "x": { "dtype": "uint8", "shape": [0], "data": { "kind": "values", "values": [] } } }, "outputs": { "y": { "dtype": "uint8", "shape": [0], "tolerance": 0 } } }, { "name": "f32_vec4_sustained_default_1024", "inputs": { "x": { "dtype": "float32", "shape": [1024], "data": { "kind": "linspace", "start": -2.0, "end": 2.0 } } }, "outputs": { "y": { "dtype": "float32", "shape": [1024], "tolerance": 0.000001 } } } ] }