Replace export_for_training with export

Summary: export_for_training will be deprecated soon, swap with export

Differential Revision: D79097629

Author: ethansfng

backends/cadence/aot/compiler_funcs.py

@@ -35,7 +35,7 @@ def trace(
     decomp_table = torch.export.default_decompositions()
     # pyre-fixme[6]: For 1st argument expected `Dict[typing.Callable[..., typing.Any
     remove_decompositions(decomp_table, ops_to_keep)
-    program = torch.export.export_for_training(
+    program = torch.export.export(
         model, inputs, strict=strict
     ).run_decompositions(decomp_table)