- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for DecomposeGraph (0.16 sec)
-
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.h
// FunctionDef. absl::StatusOr<FunctionDef> ExpandNode(const NodeDef& node_def, StringPiece func_name); // Runs the decompose passes on the user_module. Status DecomposeGraph(mlir::ModuleOp user_module); // Erases the tfr_module created. void Destroy(); private: mlir::ModuleOp tfr_module_; mlir::PassManager pm_; GraphExportConfig export_confs_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 11:12:54 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
// Run the decompose passes on the module TF_RETURN_IF_ERROR(DecomposeGraph(module)); // Export the result as a FunctionDef. FunctionDef func_def; TF_RETURN_IF_ERROR( ConvertMlirFunctionToFunctionLibraryDef(func, export_confs_, &func_def)); module.erase(); return func_def; } Status TFRDecomposeContext::DecomposeGraph(mlir::ModuleOp user_module) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/graph_decompose_pass.cc
return absl::OkStatus(); } tf_core_op_expansion_graph_counter->GetCell()->IncrementBy(1); LOG_FIRST_N(INFO, 1) << "Run Graph Decomposition Passes"; TF_RETURN_IF_ERROR(DecomposeGraph(module)); LOG_FIRST_N(INFO, 1) << "Finish Graph Decomposition Passes"; return absl::OkStatus(); } namespace { constexpr int kMlirGraphDecomposePassPriority = -1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Feb 25 16:22:36 UTC 2024 - 2.8K bytes - Viewed (0)