/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
torch
/
quantization
/
fx
/
backend_config_dict
/
/usr/local/lib64/python3.6/site-packages/torch/quantization/fx/backend_config_dict
mkdir
upload
Name
Size
Mode
Actions
__pycache__/
-
0755
rm
fbgemm.py
427
0644
edit
dl
rm
tensorrt.py
553
0644
edit
dl
rm
__init__.py
213
0644
edit
dl
rm
Edit:
/usr/local/lib64/python3.6/site-packages/torch/quantization/fx/backend_config_dict/tensorrt.py
(553B)
import torch from ..quantization_patterns import ConvReLUQuantizeHandlerNew, LinearReLUQuantizeHandlerNew def get_tensorrt_backend_config_dict(): """ Get the backend config dictionary for tensorrt backend NOTE: Current api will change in the future, it's just to unblock experimentation for new backends, please don't use it right now. """ quant_patterns = { torch.nn.Conv2d: ConvReLUQuantizeHandlerNew, torch.nn.Linear: LinearReLUQuantizeHandlerNew } return { "quant_patterns": quant_patterns }
Save
cmd:
run