stablehlo_sample.json

{
  "functions": [
    {
      "name": "@main",
      "inputs": [
        "%image",
        "%weights",
        "%bias"
      ],
      "inputTypes": [
        "tensor<28x28xf32>",
        "tensor<784x10xf32>",
        "tensor<1x10xf32>"
      ],
      "outputTypes": {
        "0": {
          "type": "tensor<1x10xf32>"
        }
      },
      "operations": [
        {
          "name": "stablehlo.reshape",
          "attributes": {},
          "inputs": [
            "%image"
          ],
          "inputTypes": [
            "tensor<28x28xf32>"
          ],
          "outputs": [
            "%0"
          ],
          "outputTypes": [
            "tensor<1x784xf32>"
          ],
          "body": null
        },
        {
          "name": "stablehlo.dot",
          "attributes": {},
          "inputs": [
            "%0",
            "%weights"
          ],
          "inputTypes": [
            "tensor<1x784xf32>",
            "tensor<784x10xf32>"
          ],
          "outputs": [
            "%1"
          ],
          "outputTypes": [
            "tensor<1x10xf32>"
          ],
          "body": null
        },
        {
          "name": "stablehlo.add",
          "attributes": {},
          "inputs": [
            "%1",
            "%bias"
          ],
          "inputTypes": [
            "tensor<1x10xf32>",
            "tensor<1x10xf32>"
          ],
          "outputs": [
            "%2"
          ],
          "outputTypes": [
            "tensor<1x10xf32>"
          ],
          "body": null
        },
        {
          "name": "stablehlo.constant",
          "attributes": {
            "value": "dense<0.0> : tensor<1x10xf32>"
          },
          "inputs": [],
          "inputTypes": [],
          "outputs": [
            "%3"
          ],
          "outputTypes": [
            "tensor<1x10xf32>"
          ],
          "body": null
        },
        {
          "name": "stablehlo.maximum",
          "attributes": {},
          "inputs": [
            "%2",
            "%3"
          ],
          "inputTypes": [
            "tensor<1x10xf32>",
            "tensor<1x10xf32>"
          ],
          "outputs": [
            "%4"
          ],
          "outputTypes": [
            "tensor<1x10xf32>"
          ],
          "body": null
        },
        {
          "name": "stablehlo.return",
          "attributes": {},
          "inputs": [
            "%4"
          ],
          "inputTypes": [
            "tensor<1x10xf32>"
          ],
          "outputs": [],
          "outputTypes": [],
          "body": null
        }
      ]
    }
  ],
  "operations": [],
  "attributes": {}
}