{
  "op": "ai.onnx.Size",
  "cases": [
    {
      "name": "int16_rank2_boundaries",
      "inputs": {
        "data": {
          "dtype": "int16",
          "shape": [2, 3],
          "data": { "kind": "values", "values": [-32768, -1, 0, 1, 32766, 32767] }
        }
      },
      "outputs": {
        "size": { "dtype": "uint32", "shape": [], "tolerance": 0, "data": { "kind": "values", "values": [6] } }
      }
    },
    {
      "name": "rank3",
      "inputs": { "data": { "dtype": "float32", "shape": [2, 3, 4] } },
      "outputs": { "size": { "dtype": "uint32", "shape": [] } }
    },
    {
      "name": "true_scalar_output",
      "inputs": { "data": { "dtype": "float32", "shape": [2, 0, 4] } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "scalar_input_size_one",
      "inputs": { "data": { "dtype": "float32", "shape": [] } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "uint32_rank2",
      "inputs": { "data": { "dtype": "uint32", "shape": [3, 3], "data": { "kind": "constant", "value": 0 } } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "ort_float_1d_1000",
      "provenance": {
        "source": "onnxruntime/test/providers/cpu/tensor/size_test.cc",
        "test": "SizeOpTest.Float1000Test"
      },
      "inputs": { "data": { "dtype": "float32", "shape": [1000], "data": { "kind": "constant", "value": 0.0 } } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "ort_float_3x3x4",
      "provenance": {
        "source": "onnxruntime/test/providers/cpu/tensor/size_test.cc",
        "test": "SizeOpTest.Float3x3x4Test"
      },
      "inputs": { "data": { "dtype": "float32", "shape": [3, 3, 4], "data": { "kind": "constant", "value": 0.0 } } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "ort_float_3x3",
      "provenance": {
        "source": "onnxruntime/test/providers/cpu/tensor/size_test.cc",
        "test": "SizeOpTest.Float3x3Test"
      },
      "inputs": { "data": { "dtype": "float32", "shape": [3, 3], "data": { "kind": "constant", "value": 0.0 } } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "ort_int32_3x3",
      "provenance": { "source": "onnxruntime/test/providers/cpu/tensor/size_test.cc", "test": "SizeOpTest.Int3x3Test" },
      "inputs": { "data": { "dtype": "int32", "shape": [3, 3], "data": { "kind": "constant", "value": 0 } } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "ort_int32_scalar",
      "provenance": {
        "source": "onnxruntime/test/providers/cpu/tensor/size_test.cc",
        "test": "SizeOpTest.IntScalarTest"
      },
      "inputs": { "data": { "dtype": "int32", "shape": [], "data": { "kind": "values", "values": [0] } } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "bool_rank2_size",
      "provenance": {
        "source": "onnxruntime/test/providers/cpu/tensor/size_test.cc",
        "test": "SizeOpTest.Int3x3Test",
        "notes": "Same shape-only Size behavior as ORT's int case, using ONNX-valid bool input values."
      },
      "inputs": {
        "data": { "dtype": "bool", "shape": [2, 3], "data": { "kind": "values", "values": [1, 0, 1, 1, 0, 0] } }
      },
      "outputs": {
        "size": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [6] }, "tolerance": 0 }
      }
    },
    {
      "name": "ort_int8_3x3_adapted",
      "provenance": {
        "source": "onnxruntime/test/providers/cpu/tensor/size_test.cc",
        "test": "SizeOpTest.Int3x3Test",
        "notes": "Same shape as ORT's int case, using ONNX-valid int8 payload values."
      },
      "inputs": {
        "data": {
          "dtype": "int8",
          "shape": [3, 3],
          "data": { "kind": "values", "values": [-4, -3, -2, -1, 0, 1, 2, 3, 4] }
        }
      },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "uint8_zero_sized_product",
      "inputs": { "data": { "dtype": "uint8", "shape": [2, 0, 5], "data": { "kind": "values", "values": [] } } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "onnx_backend_size",
      "provenance": {
        "source": "cmake/external/onnx/onnx/backend/test/data/node/test_size",
        "notes": "ONNX shape/size outputs are int64; this framework uses uint32 shape/index storage."
      },
      "inputs": { "data": { "dtype": "float32", "shape": [3, 4, 5] } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "onnx_backend_size_example",
      "provenance": {
        "source": "cmake/external/onnx/onnx/backend/test/data/node/test_size_example",
        "notes": "ONNX shape/size outputs are int64; this framework uses uint32 shape/index storage."
      },
      "inputs": { "data": { "dtype": "float32", "shape": [2, 3] } },
      "outputs": { "size": { "dtype": "uint32", "shape": [], "tolerance": 0 } }
    },
    {
      "name": "ort_caseB_empty",
      "inputs": { "data": { "dtype": "float32", "shape": [0, 3, 4], "data": { "kind": "values", "values": [] } } },
      "outputs": {
        "size": { "dtype": "uint32", "shape": [], "data": { "kind": "values", "values": [0] }, "tolerance": 0 }
      }
    },
    {
      "name": "rank3_f16",
      "inputs": { "data": { "dtype": "float16", "shape": [2, 3, 4] } },
      "outputs": { "size": { "dtype": "uint32", "shape": [] } }
    },
    {
      "name": "rank6_numel",
      "inputs": { "data": { "dtype": "float32", "shape": [2, 3, 4, 5, 6, 7] } },
      "outputs": {
        "size": { "dtype": "uint32", "shape": [], "tolerance": 0, "data": { "kind": "values", "values": [5040] } }
      }
    }
  ]
}