onnxruntime/js/web/test/data/ops/conv.jsonc

[
  {
    "name": "conv without bias addition A",
    "operator": "Conv",
    "attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
    "cases": [
      {
        "name": "T[0]",
        "inputs": [
          {
            "data": [10, 20, 30, 40, 50, 60, 70, 80, 90],
            "dims": [1, 1, 3, 3],
            "type": "float32"
          },
          {
            "data": [1, 2, 3, 4],
            "dims": [1, 1, 2, 2],
            "type": "float32"
          }
        ],
        "outputs": [
          {
            "data": [370, 470, 670, 770],
            "dims": [1, 1, 2, 2],
            "type": "float32"
          }
        ]
      }
    ]
  },
  {
    "name": "Conv without bias addition B",
    "operator": "Conv",
    "attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
    "cases": [
      {
        "name": "T[0]",
        "inputs": [
          {
            "data": [10, 20, 30, 40, 50, 60, 70, 80],
            "dims": [1, 2, 2, 2],
            "type": "float32"
          },
          {
            "data": [1, 2, 3, 4, 5, 6, 7, 8],
            "dims": [1, 2, 2, 2],
            "type": "float32"
          }
        ],
        "outputs": [
          {
            "data": [2040],
            "dims": [1, 1, 1, 1],
            "type": "float32"
          }
        ]
      }
    ]
  },
  {
    "name": "Conv with bias addition A",
    "operator": "Conv",
    "attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
    "cases": [
      {
        "name": "T[0]",
        "inputs": [
          {
            "data": [10, 20, 30, 40, 50, 60, 70, 80],
            "dims": [1, 2, 2, 2],
            "type": "float32"
          },
          {
            "data": [
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1
            ],
            "dims": [4, 2, 2, 2],
            "type": "float32"
          },
          {
            "data": [0.1, 0.2, 0.3, 0.4],
            "dims": [4],
            "type": "float32"
          }
        ],
        "outputs": [
          {
            "data": [360.1, 360.2, 360.3, 360.4],
            "dims": [1, 4, 1, 1],
            "type": "float32"
          }
        ]
      }
    ]
  },
  {
    "name": "conv with bias addition B",
    "operator": "Conv",
    "attributes": [{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }],
    "cases": [
      {
        "name": "T[0]",
        "inputs": [
          {
            "data": [1, 2, 3, 4],
            "dims": [1, 1, 2, 2],
            "type": "float32"
          },
          {
            "data": [1, 1, 1, 1],
            "dims": [1, 1, 2, 2],
            "type": "float32"
          },
          {
            "data": [5],
            "dims": [1],
            "type": "float32"
          }
        ],
        "outputs": [
          {
            "data": [15],
            "dims": [1, 1, 1, 1],
            "type": "float32"
          }
        ]
      }
    ]
  },
  {
    "name": "conv - group - A",
    "operator": "Conv",
    "attributes": [
      { "name": "kernel_shape", "data": [1, 1], "type": "ints" },
      { "name": "group", "data": 2, "type": "int" }
    ],
    "cases": [
      {
        "name": "T[0]",
        "inputs": [
          {
            "data": [
              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
            ],
            "dims": [1, 2, 3, 3],
            "type": "float32"
          },
          {
            "data": [1.0, 2.0],
            "dims": [2, 1, 1, 1],
            "type": "float32"
          }
        ],
        "outputs": [
          {
            "data": [
              0.0,
              1.0,
              2.0,
              3.0,
              4.0,
              5.0,
              6.0,
              7.0,
              8.0,
              18.0,
              20.0,
              22.0,
              24.0,
              26.0,
              28.0,
              30.0,
              32.0,
              34.0
            ],
            "dims": [1, 2, 3, 3],
            "type": "float32"
          }
        ]
      }
    ]
  },
  {
    "name": "conv - group - B",
    "operator": "Conv",
    "attributes": [
      { "name": "kernel_shape", "data": [2, 2], "type": "ints" },
      { "name": "group", "data": 3, "type": "int" }
    ],
    "cases": [
      {
        "name": "T[0]",
        "inputs": [
          {
            "data": [
              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
            ],
            "dims": [1, 3, 3, 3],
            "type": "float32"
          },
          {
            "data": [
              1.0,
              2.0,
              3.0,
              4.0,
              5.0,
              6.0,
              7.0,
              8.0,
              9.0,
              10.0,
              11.0,
              12.0
            ],
            "dims": [3, 1, 2, 2],
            "type": "float32"
          },
          {
            "data": [0.1, 0.2, 0.3],
            "dims": [3],
            "type": "float32"
          }
        ],
        "outputs": [
          {
            "data": [
              27.1,
              37.1,
              57.1,
              67.1,
              293.2,
              319.2,
              371.2,
              397.2,
              847.3,
              889.3,
              409.3,
              428.3
            ],
            "dims": [1, 3, 2, 2],
            "type": "float32"
          }
        ]
      }
    ]
  },
  {
    "name": "conv - group - C",
    "operator": "Conv",
    "attributes": [
      { "name": "kernel_shape", "data": [2, 2], "type": "ints" },
      { "name": "group", "data": 3, "type": "int" }
    ],
    "cases": [
      {
        "name": "T[0]",
        "inputs": [
          {
            "data": [
              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,
              24.0,
              25.0,
              26.0,
              27.0,
              28.0,
              29.0,
              30.0,
              31.0,
              32.0,
              33.0,
              34.0,
              35.0
            ],
            "dims": [1, 3, 3, 4],
            "type": "float32"
          },
          {
            "data": [
              1.0,
              2.0,
              3.0,
              4.0,
              5.0,
              6.0,
              7.0,
              8.0,
              9.0,
              10.0,
              11.0,
              12.0
            ],
            "dims": [3, 1, 2, 2],
            "type": "float32"
          }
        ],
        "outputs": [
          {
            "data": [
              34,
              44,
              54,
              74,
              84,
              94,
              386,
              412,
              438,
              490,
              516,
              542,
              1122,
              1164,
              1206,
              1290,
              1332,
              1374
            ],
            "dims": [1, 3, 2, 3],
            "type": "float32"
          }
        ]
      }
    ]
  },

  {
    "name": "conv - pointwise",
    "operator": "Conv",
    "attributes": [{ "name": "kernel_shape", "data": [1, 1], "type": "ints" }],
    "cases": [
      {
        "name": "T[0]",
        "inputs": [
          {
            "data": [
              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,
              24.0,
              25.0,
              26.0,
              27.0,
              28.0,
              29.0,
              30.0,
              31.0
            ],
            "dims": [1, 8, 2, 2],
            "type": "float32"
          },
          {
            "data":[
              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
            ],
            "dims":[2, 8, 1, 1],
            "type":"float32"
          },
          {
            "data": [0.5, 0.4],
            "dims": [2],
            "type": "float32"
          }
        ],
        "outputs":[
          {
            "data": [
              560.5,588.5,616.5,644.5,1456.4,1548.4,1640.4,1732.4
            ],
            "dims": [1, 2, 2, 2],
            "type": "float32"
          }
        ]
      }
    ]
  }
]