@@ -199,7 +199,7 @@ def export_program(
199199 return expo_program
200200
201201
202- def lower_ep_to_edge (
202+ def _lower_ep_to_edge (
203203 expo_program : ExportedProgram ,
204204 dump_graphs : bool = False ,
205205 constant_methods : Optional [dict [str , object ]] = None ,
@@ -250,7 +250,7 @@ def export_to_edge(
250250 expo_program = export_program (model , inputs )
251251
252252 # Lower the model to edge IR.
253- edge_prog_manager = lower_ep_to_edge (expo_program , dump_graphs , constant_methods )
253+ edge_prog_manager = _lower_ep_to_edge (expo_program , dump_graphs , constant_methods )
254254
255255 return edge_prog_manager
256256
@@ -272,22 +272,22 @@ def quantize_and_export_to_edge(
272272 dump_graphs = dump_graphs ,
273273 )
274274
275- return lower_ep_to_edge (
275+ return _lower_ep_to_edge (
276276 quantized_model ,
277277 dump_graphs = dump_graphs ,
278278 constant_methods = constant_methods ,
279279 )
280280
281281
282- def lower_ep_to_cadence (
282+ def _lower_ep_to_cadence (
283283 program : ExportedProgram ,
284284 dump_graphs : bool = False ,
285285 opt_level : int = 1 ,
286286) -> EdgeProgramManager :
287287 """
288288 Lower an existing ExportedProgram to edge IR and apply frontend optimization passes.
289289 """
290- edge_prog_manager = lower_ep_to_edge (program , dump_graphs = dump_graphs )
290+ edge_prog_manager = _lower_ep_to_edge (program , dump_graphs = dump_graphs )
291291 cadence_passes = get_cadence_passes (opt_level )
292292
293293 # Run a couple required passes for quant/dequant ops
@@ -329,7 +329,7 @@ def quantize_and_export_to_cadence(
329329 """
330330 quantized_model = quantize_pt2 (model , inputs )
331331
332- return lower_ep_to_cadence (
332+ return _lower_ep_to_cadence (
333333 quantized_model ,
334334 opt_level = opt_level ,
335335 dump_graphs = dump_graphs ,
0 commit comments