Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for HARDWARE_BACKENDS (0.17 sec)

  1. tensorflow/compiler/mlir/lite/experimental/tac/py_wrapper/tac_wrapper.cc

    bool run_tac(const std::string& model_file_path,
                 const std::vector<std::string>& device_specs,
                 const std::string& model_output_path) {
      mlir::TFL::tac::TacModule::Options options;
      options.hardware_backends = device_specs;
      options.enable_inliner = true;
      options.legalize_to_tflite_ops = true;
      mlir::TFL::tac::TacModule tac_module(options);
      mlir::DialectRegistry registry;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 06 03:08:33 UTC 2023
    - 2.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/tac/tac_module.cc

    }
    
    absl::Status TacModule::RunTacPasses(mlir::ModuleOp* module, bool debug_mode) {
      mlir::PassManager pm((*module)->getName(),
                           mlir::OpPassManager::Nesting::Implicit);
      AddTACPass(&pm, options_.hardware_backends);
      if (!debug_mode) {
        AddExportTFLPass(&pm, options_.enable_inliner);
      }
    
      mlir::StatusScopedDiagnosticHandler statusHandler(module->getContext(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 5.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/experimental/tac/tac_translate.cc

    }
    
    absl::Status TargetAwareConversionMain() {
      std::vector<std::string> device_specs_array =
          absl::StrSplit(device_specs, ',', absl::SkipEmpty());
      mlir::TFL::tac::TacModule::Options options;
      options.hardware_backends = device_specs_array;
      options.debug_mode = true;
      if (!output_mlir || inline_subgraphs) {
        options.debug_mode = false;
      }
      options.enable_inliner = true;
      options.legalize_to_tflite_ops = true;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 22 14:25:57 UTC 2022
    - 5.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/experimental/tac/tac_module.h

    class TacModule {
     public:
      // TAC options. Contains knobs to configure TAC as needed.
      struct Options {
        // List of names for the requested Target hardware.
        std::vector<std::string> hardware_backends;
        // Debug mode.
        // This will output different alternative subgraphs in mlir format for debug
        // purpose.
        bool debug_mode = false;
        // Whether to enable inliner passes or not.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 4.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/experimental/tac/README.md

    to use TAC to optimize for your model:
    
    ```
    bazel run -c opt //tensorflow/compiler/mlir/lite/experimental/tac:tac-translate -- <PATH_TO_YOUR_MODEL> -o=<OUTPUT_PATH> -device-specs=<HARDWARE_BACKENDS>
    ```
    
    The devices_specs is a list of the names of the desired hardware backends,
    separated by comma, e.g., "GPU,CPU".
    
    If you're interested in what are the subgraphs being explored for different
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 29 18:32:13 UTC 2022
    - 11.6K bytes
    - Viewed (0)
Back to top