- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for applyTensorflowAndCLOptions (0.42 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/graph_optimization_pass.cc
<< ", session flag not enabled"; return absl::OkStatus(); } VLOG(1) << "Run MLIR Graph Optimization Passes"; PassManager pm(module.getContext()); ::tensorflow::applyTensorflowAndCLOptions(pm); // Run island coarsening before shape inference to allow more exact shape // inference using constant folding within islands. pm.addNestedPass<func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/function/function.cc
} mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext()); // Lower MLIR TF Dialect to MLIR TFRT CoreRT dialect. mlir::PassManager pm(module.getContext()); tensorflow::applyTensorflowAndCLOptions(pm); tensorflow::TfrtPipelineOptions pass_options; if (!options.default_device.empty()) { pass_options.default_device = options.default_device; } if (!options.force_data_format.empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 08:13:15 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/tf_dialect_to_executor.cc
} } // namespace tensorflow::Status ExportFromTensorflowDialectToExecutor( ModuleOp module, llvm::StringRef module_name) { PassManager tf_to_executor(module.getContext()); ::tensorflow::applyTensorflowAndCLOptions(tf_to_executor); tf_to_executor.enableVerifier(); AddTfDialectToExecutorPasses(tf_to_executor); mlir::StatusScopedDiagnosticHandler diag_handler( module.getContext(), /*propagate=*/false,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 16 03:41:02 UTC 2024 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dump_mlir_util.h
// tensorflow specific options. // // Note that this function should be in a more appropriate file, but it is // unclear what a proper file would be as no other functions would currently be // in the file also. void applyTensorflowAndCLOptions(mlir::PassManager& pm, llvm::StringRef dir_path = ""); // Prints the pass pipeline of `pass_manager` to `os`. void PrintPassPipeline(const mlir::PassManager& pass_manager,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 04:50:13 UTC 2023 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_dialect_to_executor.cc
} } // namespace tensorflow::Status ExportFromTensorflowDialectToExecutor( ModuleOp module, llvm::StringRef module_name) { PassManager tf_to_executor(module.getContext()); ::tensorflow::applyTensorflowAndCLOptions(tf_to_executor); tf_to_executor.enableVerifier(); AddTfDialectToExecutorPasses(tf_to_executor); if (VLOG_IS_ON(1) || DEBUG_DATA_DUMPER()->ShouldDump(module_name.str(), kDebugGroupMain)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 23:22:50 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf.cc
"1; this could happen if the binary doesn't link the constant fold " "hook registration library."); } PassManager bridge(module.getContext()); bridge.enableVerifier(); ::tensorflow::applyTensorflowAndCLOptions(bridge); // Populate a passmanager with the list of passes that implement the bridge. pipeline_builder(bridge); mlir::StatusScopedDiagnosticHandler diag_handler(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:25:18 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/cluster_tf.cc
"1; this could happen if the binary doesn't link the constant fold " "hook registration library."); } PassManager bridge(module.getContext()); bridge.enableVerifier(); ::tensorflow::applyTensorflowAndCLOptions(bridge); // Populate a passmanager with the list of passes that implement the bridge. pipeline_builder(bridge); mlir::StatusScopedDiagnosticHandler diag_handler(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:25:18 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops.cc
absl::Status RunLowerClusterToRuntimeOpsPassPipeline( mlir::ModuleOp module, tsl::DeviceType xla_device_type, llvm::StringRef module_name) { PassManager runtime_lowering(module.getContext()); ::tensorflow::applyTensorflowAndCLOptions(runtime_lowering); if (xla_device_type == DeviceType(DEVICE_TPU_XLA_JIT)) { AddTPULowerClusterToRuntimeOpsPassPipeline(runtime_lowering, module_name); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dump_mlir_util.cc
path, std::make_unique<WritableFileRawStream>(std::move(file))); }; pm.enableCrashReproducerGeneration(factory, /*genLocalReproducer=*/false); } void applyTensorflowAndCLOptions(mlir::PassManager& pm, llvm::StringRef dir_path) { mlir::registerPassManagerCLOptions(); if (!mlir::succeeded(mlir::applyPassManagerCLOptions(pm))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 03:03:46 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc
// breakup islands passes are common between the export pipeline and here. // Reconsider this if there is more commonality in the future with more // passes. mlir::PassManager manager(module.getContext()); applyTensorflowAndCLOptions(manager); manager.addPass(mlir::TF::CreatePrepareTpuComputationForTfExportPass()); manager.addPass(mlir::TF::CreateTFRegionControlFlowToFunctional()); manager.addPass(mlir::TF::CreateTFShapeInferencePass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K bytes - Viewed (0)