get_default_qat_qconfig_mapping

函数torch.ao.quantization.qconfig_mapping.get_default_qat_qconfig_mapping(backend='x86', version=1)[源代码]

返回量化感知训练的默认 QConfigMapping。

参数
  • backend (*) – 默认的 qconfig 映射量化后端,应该选择 "x86"(默认)、"fbgemm"、"qnnpack" 或 "onednn" 中的一个。

  • 版本 (*) – 默认 qconfig 映射的版本号

返回类型

QConfigMapping

本页目录