- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CreateRenameEntrypointToMainPass (0.5 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/rename_entrypoint_to_main.cc
} // Found entrypoint Builder builder(candidate_funcs[0]); candidate_funcs[0].setName(builder.getStringAttr("main")); } }; std::unique_ptr<Pass> CreateRenameEntrypointToMainPass() { return std::make_unique<RenameEntrypointToMainPass>(); } static PassRegistration<RenameEntrypointToMainPass> pass; } // namespace odml
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/transforms.cc
namespace mlir { namespace odml { void AddTFToStablehloPasses(OpPassManager& pm, bool skip_resize, bool smuggle_disallowed_ops) { pm.addPass(CreateRenameEntrypointToMainPass()); // if the input is a call_xla_module, then unwrap the content pm.addPass(mlir::odml::CreateLegalizeTFXlaCallModuleToStablehloPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 04:34:23 UTC 2024 - 5.8K bytes - Viewed (0)