File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -119,14 +119,15 @@ def to_quantized_edge_program(
119119 operators_not_to_delegate = operators_not_to_delegate ,
120120 neutron_converter_flavor = neutron_converter_flavor ,
121121 )
122- partitioner = NeutronPartitioner (compile_spec , custom_delegation_options )
123- edge_program_manager = edge_program_manager .to_backend (partitioner )
124122
125123 if remove_quant_io_ops :
126124 edge_program_manager = edge_program_manager .transform (
127125 [RemoveIOQuantOpsPass (edge_program_manager = edge_program_manager )]
128126 )
129127
128+ partitioner = NeutronPartitioner (compile_spec , custom_delegation_options )
129+ edge_program_manager = edge_program_manager .to_backend (partitioner )
130+
130131 return edge_program_manager
131132
132133
You can’t perform that action at this time.
0 commit comments