Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for StandardPipelineOptions (0.78 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops.cc

    }
    
    void CreateTPULowerClusterToRuntimeOpsPassPipeline(
        OpPassManager& pm, const StandardPipelineOptions& options) {
      AddTPULowerClusterToRuntimeOpsPassPipeline(pm, /*module_name=*/"");
    }
    
    void CreateNonTPULowerClusterToRuntimeOpsPassPipeline(
        OpPassManager& pm, const StandardPipelineOptions& options) {
      AddNonTPULowerClusterToRuntimeOpsPassPipeline(pm, /*module_name=*/"");
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 18:52:57 UTC 2024
    - 9.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc

                                                     bool enable_logging,
                                                     bool enable_inliner) {
      PassManager bridge(module.getContext());
    
      StandardPipelineOptions pipeline_options;
      pipeline_options.enable_inliner.setValue(enable_inliner);
      CreateTFStandardPipeline(bridge, pipeline_options);
    
      mlir::StatusScopedDiagnosticHandler diag_handler(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 09 17:16:05 UTC 2023
    - 2.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc

          signalPassFailure();
      }
    
      FrozenRewritePatternSet patterns;
    };
    
    }  // namespace
    
    void CreateTFStandardPipeline(OpPassManager &pm,
                                  const StandardPipelineOptions &options) {
      OpPassManager &func_pm = pm.nest<func::FuncOp>();
    
      // First operates on the executor dialect:
      // - remove dead islands.
      // - fuse islands as much as possible.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/passes.h

    void CreateLayoutOptimizationPipeline(
        OpPassManager& pm,  // NOLINT - MLIR contract is pass by mutable reference.
        const LayoutOptimizationPipelineOptions& options);
    
    struct StandardPipelineOptions
        : public PassPipelineOptions<StandardPipelineOptions> {
      Option<bool> enable_inliner{*this, "enable-inliner",
                                  llvm::cl::desc("Enable inliner."),
                                  llvm::cl::init(false)};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 31.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.cc

      mlir::StatusScopedDiagnosticHandler statusHandler(module_op.getContext(),
                                                        /*propagate=*/true);
    
      mlir::TF::StandardPipelineOptions standard_pipeline_options;
      standard_pipeline_options.enable_inliner = false;
      standard_pipeline_options.form_clusters = false;
      mlir::TF::CreateTFStandardPipeline(pm_before_freezing_variables,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 12:49:45 UTC 2024
    - 9.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/python/mlir.cc

      mlir::OwningOpRef<mlir::ModuleOp> module = std::move(module_or).value();
      mlir::PassManager pm(&context);
      std::string error;
      llvm::raw_string_ostream error_stream(error);
    
      mlir::TF::StandardPipelineOptions tf_options;
      mlir::TF::CreateTFStandardPipeline(pm, tf_options);
    
      mlir::StatusScopedDiagnosticHandler diagnostic_handler(&context);
      if (failed(pm.run(*module))) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:16:49 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/tf_tfl_passes.cc

      auto wrapped_ops = mlir::TFL::AllTfFakeQuantOps();
      pass_manager->addNestedPass<mlir::func::FuncOp>(
          mlir::TFL::CreateRaiseCustomOpsPass(wrapped_ops));
    
      mlir::TF::StandardPipelineOptions standard_pipeline_options;
      standard_pipeline_options.enable_inliner = false;
      standard_pipeline_options.form_clusters = pass_config.form_clusters;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 18:45:51 UTC 2024
    - 25.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

                              /*is_resource=*/arg.kind == XlaArgument::kResource});
      }
    
      mlir::PassManager pm(module_op.getContext());
      applyTensorflowAndCLOptions(pm);
      mlir::TF::StandardPipelineOptions tf_options;
      mlir::TF::CreateTFStandardPipeline(pm, tf_options);
    
      if (VLOG_IS_ON(1))
        tensorflow::DumpMlirOpToFile("compile_graph_setup_before", module_op);
      if (VLOG_IS_ON(2)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
Back to top