File tree Expand file tree Collapse file tree
backends/nxp/tests_models Expand file tree Collapse file tree Original file line number Diff line number Diff line change 3333)
3434from executorch .backends .nxp .quantizer .neutron_quantizer import NeutronQuantizer
3535from executorch .backends .nxp .tests_models .model_input_spec import ModelInputSpec
36+ from executorch .backends .transforms .quantize_fused_convbn_bias_pass import (
37+ QuantizeFusedConvBnBiasAtenPass ,
38+ )
3639from executorch .devtools .visualization .visualization_utils import (
3740 visualize_with_clusters ,
3841)
@@ -149,6 +152,11 @@ def to_quantized_edge_program(
149152
150153 exir_program_aten_quant = convert_pt2e (m )
151154
155+ if use_qat :
156+ exir_program_aten_quant = QuantizeFusedConvBnBiasAtenPass (
157+ default_zero_bias = True
158+ )(exir_program_aten_quant ).graph_module
159+
152160 # To ATen
153161 core_aten_ep = _to_core_aten (
154162 exir_program_aten_quant , example_input , None , verbose = True
You can’t perform that action at this time.
0 commit comments