Searched refs:_set_pattern_complex_format (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/ |
| H A D | onednn.py | 191 ._set_pattern_complex_format(
204 ._set_pattern_complex_format(
277 ._set_pattern_complex_format(
290 ._set_pattern_complex_format(
383 ._set_pattern_complex_format(
396 ._set_pattern_complex_format(
479 ._set_pattern_complex_format(
492 ._set_pattern_complex_format(
|
| H A D | backend_config.py | 616 def _set_pattern_complex_format(self, pattern: Pattern) -> BackendPatternConfig: member in BackendPatternConfig
701 conf._set_pattern_complex_format(
|
| H A D | _qnnpack_pt2e.py | 118 ._set_pattern_complex_format(
|
| H A D | README.md | 37 …tuple" pattern format, enabled through `BackendPatternConfig()._set_pattern_complex_format(...)`. …
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 752 … ._set_pattern_complex_format((nn.ReLU, (torch.add, MatchAllNode, (nn.BatchNorm2d, nn.Conv2d)))) \
755 …._set_pattern_complex_format((nn.ReLU, (operator.add, MatchAllNode, (nn.BatchNorm2d, nn.Conv2d))))…
814 … ._set_pattern_complex_format((nn.ReLU, (torch.add, (nn.BatchNorm2d, nn.Conv2d), MatchAllNode))) \
880 ._set_pattern_complex_format((nn.ReLU, (torch.add, nn.Conv2d, MatchAllNode))) \
6262 … ._set_pattern_complex_format((torch.reshape, torch.transpose, MatchAllNode)) # noqa: E131
|