{
  "customize": {
    "extra_dir": "",
    "impl_name": "pytorch",
    "install_env": {
      "PLUGIN_LSTM_HIDDEN_SIZE": 1024,
      "PLUGIN_LSTM_INPUT_SIZE": 2048,
      "PLUGIN_LSTM_NUM_LAYERS": 3,
      "PLUGIN_NAME": "lstm_rnnt_post.py",
      "PLUGIN_SIGNATURE": "LSTM_PYTORCH"
    },
    "env_prefix": "CK_NNTEST_PLUGIN_POST",
    "model_name": "lstm-post",
    "version": "0.1"
  },
  "deps": {
    "model": {
      "local": "yes",
      "name": "PyTorch RNN-T model",
      "sort": 80,
      "tags": "model,pytorch,rnnt"
    },
   "python": {
      "add_to_path": "no",
      "force_target_as_host": "yes",
      "local": "yes",
      "name": "Python interpreter",
      "sort": 80,
      "tags": "compiler,python"
    },
    "pytorch": {
      "local": "yes",
      "name": "PyTorch framework",
      "sort": 90,
      "tags": "lib,pytorch"
    }
  },
  "end_full_path_universal": "plugin/lstm_rnnt_post.py",
  "process_script": "install",
  "soft_uoa": "fee21442c93c5180",
  "suggested_path": "nntest-plugins-lstm-rnnt-post-pytorch",
  "tags": [
    "nntest",
    "plugin",
    "lstm",
    "rnnt-post",
    "pytorch"
  ],
  "variations": {
    "naive": {
      "extra_customize": {
        "package_name": "RNNT LSTM Post naive plugin",
        "version": "naive"
      },
      "extra_env": {
        "PLUGIN_SUBDIR": "naive"
      },
      "on_by_default": "no"
    },
    "orig": {
      "extra_customize": {
        "package_name": "RNNT LSTM Post original plugin",
        "version": "orig"
      },
      "extra_env": {
        "PLUGIN_SUBDIR": "original"
      },
      "on_by_default": "yes"
    },
    "orig-quant": {
      "extra_customize": {
        "package_name": "RNNT LSTM Post original plugin - quantized",
        "version": "orig"
      },
      "extra_env": {
        "PLUGIN_SUBDIR": "orig-quantized"
      },
      "on_by_default": "no"
    }
  }
}