| /external/pytorch/torch/quantization/ |
| D | fuse_modules.py | 12 from torch.ao.quantization.fuse_modules import (
17 fuse_modules,
|
| D | __init__.py | 3 from .fuse_modules import fuse_modules
|
| /external/pytorch/test/quantization/eager/ |
| D | test_equalize_eager.py | 7 from torch.ao.quantization.fuse_modules import fuse_modules
131 … fused_model1 = fuse_modules(model, [['conv1', 'relu1'], ['conv2', 'relu2'], ['conv3', 'relu3']])
177 …fused_model1 = fuse_modules(model, [['linear1', 'relu1'], ['linear2', 'relu2'], ['linear3', 'relu3…
|
| D | test_fuse_eager.py | 17 fuse_modules,
108 model = fuse_modules(
177 model = fuse_modules(
246 fuse_modules(
346 model = fuse_modules(model, [["fc", "bn"]])
360 model = fuse_modules(model, [["conv1", "bn1"], ["conv2", "bn2"], ["conv3", "bn3"]])
375 dummy_model = fuse_modules(dummy_model, [["fc1"]], fuser_func=custom_fuse_func)
441 … fuse_modules(model, [['features.0.0', 'features.0.1', 'features.0.2']], inplace=True)
|
| D | test_model_numerics.py | 28 … torch.ao.quantization.fuse_modules(qModel.module, [['conv1', 'bn1', 'relu1']], inplace=True)
50 … torch.ao.quantization.fuse_modules(q_model.module, [['conv1', 'bn1', 'relu1']], inplace=True)
|
| D | test_quantize_eager_ptq.py | 1319 torch.ao.quantization.fuse_modules(model, [['fc1', 'relu']], inplace=True)
|
| /external/pytorch/torch/ao/quantization/ |
| D | __init__.py | 9 from .fuse_modules import fuse_modules, fuse_modules_qat # noqa: F403
|
| D | fuse_modules.py | 132 def fuse_modules( function
|
| /external/pytorch/test/ao/sparsity/ |
| D | test_composability.py | 187 tq.fuse_modules(mod, [["5", "6"]], inplace=True)
215 tq.fuse_modules(mod, [["5", "6"]], inplace=True)
|
| /external/pytorch/test/ |
| D | test_model_dump.py | 159 fmodel = torch.ao.quantization.fuse_modules(fmodel, [
|
| D | test_mobile_optimizer.py | 429 torch.ao.quantization.fuse_modules(self, [['conv2', 'relu']], inplace=True)
|
| /external/pytorch/docs/source/ |
| D | quantization-accuracy-debugging.rst | 28 this manually with the ``torch.ao.quantization.fuse_modules`` API.
|
| D | quantization-support.rst | 34 fuse_modules.fuse_modules
|
| D | quantization.rst | 276 model_fp32_fused = torch.ao.quantization.fuse_modules(model_fp32, [['conv', 'relu']])
374 model_fp32_fused = torch.ao.quantization.fuse_modules(model_fp32,
429 :func:`~torch.ao.quantization.fuse_modules.fuse_modules` API, which takes in lists of modules
1319 .. py:module:: torch.ao.quantization.fuse_modules
1389 .. py:module:: torch.quantization.fuse_modules
|
| /external/pytorch/test/mobile/model_test/ |
| D | quantization_ops.py | 171 model_fp32_fused = torch.ao.quantization.fuse_modules(
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_quantization.py | 1575 torch.ao.quantization.fuse_modules(self, [['conv', 'bn', 'relu']], inplace=True)
1963 def fuse_modules(self): member in InnerModule
1977 torch.ao.quantization.fuse_modules(self, fusable_layers, inplace=True)
2144 def fuse_modules(self): member in SkipQuantModel
2145 self.sub.fuse_modules()
2162 def fuse_modules(self): member in AnnotatedSkipQuantModel
2163 self.sub.module.fuse_modules()
2514 torch.ao.quantization.fuse_modules(self, [['conv1', 'bn1', 'relu1']], inplace=True)
|
| /external/libfuse/lib/ |
| D | fuse.c | 200 static struct fuse_module *fuse_modules = NULL; variable
224 mod->next = fuse_modules; in fuse_register_module()
225 fuse_modules = mod; in fuse_register_module()
233 for (mp = &fuse_modules; *mp; mp = &(*mp)->next) { in fuse_unregister_module()
294 for (m = fuse_modules; m; m = m->next) { in fuse_find_module()
332 for (mp = &fuse_modules; *mp;) { in fuse_put_module()
5165 while (fuse_modules) { in fuse_destroy()
5166 fuse_put_module(fuse_modules); in fuse_destroy()
|
| /external/pytorch/test/mobile/ |
| D | test_lite_script_module.py | 544 model_fp32_fused = torch.ao.quantization.fuse_modules(
|
| /external/pytorch/test/quantization/fx/ |
| D | test_model_report_fx.py | 462 … model_fp32_fused = torch.ao.quantization.fuse_modules(model_fp32, [["conv", "bn", "relu"]])
1949 model = torch.ao.quantization.fuse_modules(model, model.get_fusion_modules())
|
| D | test_quantize_fx.py | 60 fuse_modules,
1695 fuse_modules(m_eager, fuse_list, inplace=True)
1736 fuse_modules(m_eager, fuse_list, inplace=True)
|
| /external/pytorch/test/quantization/jit/ |
| D | test_quantize_jit.py | 26 fuse_modules,
3714 fuse_modules(conv_model, ["conv", "bn"], inplace=True)
3803 eager_model.fuse_modules()
|
| /external/pytorch/test/export/ |
| D | test_converter.py | 1389 torch.ao.quantization.fuse_modules(
|
| /external/pytorch/test/onnx/ |
| D | test_pytorch_onnx_onnxruntime.py | 13278 model = torch.ao.quantization.fuse_modules(model.eval(), [["conv", "relu"]])
13312 model = torch.ao.quantization.fuse_modules(model.eval(), [["linear", "relu"]])
|