Home
last modified time | relevance | path

Searched refs:_WrapperModule (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/representation/
H A Drewrite.py9 from torch.ao.quantization.pt2e.export_utils import _WrapperModule
739 _WrapperModule(_qdq_dynamic_quantized_linear),
740 _WrapperModule(_reference_dynamic_quantized_linear),
752 _WrapperModule(_qdq_quantized_linear),
753 _WrapperModule(_reference_quantized_linear),
759 _WrapperModule(_qdq_quantized_conv2d),
760 _WrapperModule(_reference_quantized_conv2d),
766 _WrapperModule(_qdq_quantized_add_relu),
767 _WrapperModule(_reference_quantized_add_relu),
771 _WrapperModule(_qdq_quantized_add),
[all …]
/aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/
H A Dexport_utils.py14 class _WrapperModule(torch.nn.Module): class
67 _WrapperModule(dropout_train), example_inputs
70 _WrapperModule(dropout_eval), example_inputs
74 _WrapperModule(dropout_eval), example_inputs
77 _WrapperModule(dropout_train), example_inputs
144 _WrapperModule(bn_train),
149 _WrapperModule(bn_eval),
H A Dqat_utils.py10 from torch.ao.quantization.pt2e.export_utils import _WrapperModule
102 return _WrapperModule(_conv_bn_pattern)
146 return _WrapperModule(_qat_conv_bn_pattern)
186 return _WrapperModule(_qat_conv_bn_pattern_no_conv_bias)
282 return _WrapperModule(_quantized_qat_conv_bn_pattern)
336 return _WrapperModule(_folded_quantized_qat_conv_bn_pattern)
/aosp_15_r20/external/pytorch/torch/ao/quantization/quantizer/
H A Dxnnpack_quantizer_utils.py11 from torch.ao.quantization.pt2e.export_utils import _WrapperModule
497 return _WrapperModule(_conv_bn)
/aosp_15_r20/external/pytorch/torch/export/
H A D_trace.py1165 class _WrapperModule(torch.nn.Module): class
1203 _WrapperModule(traced_callable),