- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for pm_ (0.07 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/input_lowering_metrics_pass_test.cc
protected: void CreateModule(const char* module_string) { TF_ASSERT_OK_AND_ASSIGN(module_, GetMlirModuleFromString(module_string, &context_)); pm_ = std::make_unique<mlir::PassManager>(&context_); pm_->addNestedPass<mlir::func::FuncOp>(CreateInputLoweringMetricsPass()); } bool ModulesEqual(const ModuleOp& module_before, const ModuleOp& module_after) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 23 19:42:09 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types_test.cc
pm_ = std::make_unique<mlir::PassManager>(&context_); pm_->addNestedPass<mlir::func::FuncOp>( quant::stablehlo::CreateConvertTFQuantTypesPass()); } mlir::LogicalResult Run() { return pm_->run(module_.get()); } private: MLIRContext context_; OwningOpRef<ModuleOp> module_; std::unique_ptr<mlir::PassManager> pm_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 09:05:02 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
if (failed(pm_.run(user_module))) { return errors::Internal("Failed to run the decompose passes."); } return absl::OkStatus(); } // Constructor of the decompose context. TFRDecomposeContext::TFRDecomposeContext(mlir::ModuleOp tfr_module) : tfr_module_(tfr_module), pm_(tfr_module_.getContext()) { mlir::OpPassManager& func_pm = pm_.nest<mlir::func::FuncOp>();
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/tfr_decompose_ctx.h
Status DecomposeGraph(mlir::ModuleOp user_module); // Erases the tfr_module created. void Destroy(); private: mlir::ModuleOp tfr_module_; mlir::PassManager pm_; GraphExportConfig export_confs_; }; // Decomposes the NodeDef to a set of primitive ops according to the decompose // library loaded. Wrap the decomposed result in a FunctionDef.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 11:12:54 UTC 2024 - 3.2K bytes - Viewed (0)