(self, graph_module: torch.fx.GraphModule)
| 254 | self.edge_manager_update_quant_config_method(idx, val) |
| 255 | |
| 256 | def call(self, graph_module: torch.fx.GraphModule): |
| 257 | for i, qparams in self.quantized_inputs_idx_dict.items(): |
| 258 | exported_program = ( |
| 259 | self.edge_program_manager.exported_program() |
| 260 | if self.edge_program_manager is not None |
| 261 | else self.exported_program |
| 262 | ) |
| 263 | self.quant_args[i] = quantize_input(exported_program, i, qparams) |
| 264 | self.edge_manager_update_quant_config_method(i, self.quant_args[i]) |
| 265 | |
| 266 | return PassResult(graph_module, True) |
| 267 | |
| 268 | |
| 269 | class QuantizeOutputs(ExportPass): |
nothing calls this directly
no test coverage detected