- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for CreateRenameEntrypointToMainPass (0.29 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) -
tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.cc
void AddTFToStablehloPasses( mlir::PassManager& pm, llvm::ArrayRef<llvm::ArrayRef<int64_t>> input_arg_shapes) { pm.addPass(mlir::odml::CreateRenameEntrypointToMainPass()); // TODO: b/230572023 - Consider improving shape inference for While op instead // of dropping the attribute. This need not be correct for models not trained // on TPU.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 9.8K bytes - Viewed (0)