Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 for applyTensorflowAndCLOptions (0.42 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top