xnnpack/config.json

{
  "$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
  "model": "deeplab_v3",
  "family": "deeplab",
  "capabilities": [
    "semantic-segmentation"
  ],
  "backend": "xnnpack",
  "license": "bsd-3-clause",
  "variants": [
    {
      "file": "deeplab_v3_mobilenet_v3_large_xnnpack_fp32.pte",
      "precision": "fp32",
      "size": "mobilenet_v3_large",
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                1,
                3,
                520,
                520
              ],
              "dtype": "float32"
            }
          ],
          "outputs": [
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            },
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    },
    {
      "file": "deeplab_v3_mobilenet_v3_large_xnnpack_int8.pte",
      "precision": "int8",
      "size": "mobilenet_v3_large",
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                1,
                3,
                520,
                520
              ],
              "dtype": "float32"
            }
          ],
          "outputs": [
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            },
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    },
    {
      "file": "deeplab_v3_resnet50_xnnpack_fp32.pte",
      "precision": "fp32",
      "size": "resnet50",
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                1,
                3,
                520,
                520
              ],
              "dtype": "float32"
            }
          ],
          "outputs": [
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            },
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    },
    {
      "file": "deeplab_v3_resnet50_xnnpack_int8.pte",
      "precision": "int8",
      "size": "resnet50",
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                1,
                3,
                520,
                520
              ],
              "dtype": "float32"
            }
          ],
          "outputs": [
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            },
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    },
    {
      "file": "deeplab_v3_resnet101_xnnpack_fp32.pte",
      "precision": "fp32",
      "size": "resnet101",
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                1,
                3,
                520,
                520
              ],
              "dtype": "float32"
            }
          ],
          "outputs": [
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            },
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    },
    {
      "file": "deeplab_v3_resnet101_xnnpack_int8.pte",
      "precision": "int8",
      "size": "resnet101",
      "methods": {
        "forward": {
          "inputs": [
            {
              "shape": [
                1,
                3,
                520,
                520
              ],
              "dtype": "float32"
            }
          ],
          "outputs": [
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            },
            {
              "shape": [
                1,
                21,
                520,
                520
              ],
              "dtype": "float32"
            }
          ]
        }
      }
    }
  ]
}