File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
backends/nxp/tests_models Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change 2626)
2727from executorch .backends .nxp .quantizer .neutron_quantizer import NeutronQuantizer
2828from executorch .backends .nxp .tests_models .model_input_spec import ModelInputSpec
29+ from executorch .backends .transforms .quantize_fused_convbn_bias_pass import (
30+ QuantizeFusedConvBnBiasAtenPass ,
31+ )
2932from executorch .devtools .visualization .visualization_utils import (
3033 visualize_with_clusters ,
3134)
@@ -138,6 +141,11 @@ def to_quantized_edge_program(
138141
139142 exir_program_aten_quant = convert_pt2e (m )
140143
144+ if use_qat :
145+ exir_program_aten_quant = QuantizeFusedConvBnBiasAtenPass (
146+ default_zero_bias = True
147+ )(exir_program_aten_quant ).graph_module
148+
141149 # To ATen
142150 core_aten_ep = _to_core_aten (
143151 exir_program_aten_quant , example_input , None , verbose = True
You can’t perform that action at this time.
0 commit comments