inswapper_128_onnx2pytorch_bf16_dfloat11/config.json

{
  "dfloat11_config": {
    "version": "0.5.0",
    "threads_per_block": [
      512
    ],
    "bytes_per_thread": 8,
    "pattern_dict": {
      "Conv_input\\.7": [
        "weight",
        "bias"
      ],
      "Conv_input\\.15": [
        "weight",
        "bias"
      ],
      "Conv_input\\.23": [
        "weight",
        "bias"
      ],
      "Conv_input\\.31": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_159": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_170": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_213": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_224": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_267": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_278": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_321": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_332": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_375": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_386": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_429": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_440": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_483": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_494": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_537": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_548": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_591": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_602": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_645": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_656": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_699": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_710": [
        "weight",
        "bias"
      ],
      "Conv_onnx::ReduceMean_753": [
        "weight",
        "bias"
      ],
      "Gemm_onnx::Unsqueeze_764": [
        "weight",
        "bias"
      ],
      "Conv_input\\.119": [
        "weight",
        "bias"
      ],
      "Conv_input\\.131": [
        "weight",
        "bias"
      ],
      "Conv_input\\.139": [
        "weight",
        "bias"
      ],
      "Conv_onnx::Tanh_826": [
        "weight",
        "bias"
      ],
      "^$": [
        "_initializer_onnx::Resize_918",
        "_initializer_onnx::Add_164",
        "_initializer_onnx::Div_167",
        "_initializer_onnx::Resize_787",
        "_initializer_onnx::Div_830",
        "_initializer_buff2fs"
      ]
    }
  }
}