Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for RegisterExtraDialects (0.2 sec)

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

      options.enable_inliner = true;
      options.legalize_to_tflite_ops = true;
      mlir::TFL::tac::TacModule tac_module(options);
      mlir::DialectRegistry registry;
      mlir::RegisterAllTensorFlowDialects(registry);
      tac_module.RegisterExtraDialects(registry);
      tac_module.SetImporter(CreateTfLiteImporter(model_file_path));
      tac_module.SetExporter(
          CreateTfLiteExporter(model_output_path, options.hardware_backends));
      return tac_module.Run().ok();
    }
    
    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.h

        return const_backends_;
      }
    
      // Registers all dialects in 'registry' with the module.
      // This to allow clients to register extra dialects required.
      void RegisterExtraDialects(mlir::DialectRegistry& registry);
    
     protected:
      // Adds TAC passes to the 'pass_manager'.
      virtual void AddTACPass(mlir::OpPassManager* pass_manager,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 4.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/experimental/tac/tac_module.cc

      // Run TAC passes.
      auto status = RunTacPasses(&module, options_.debug_mode);
    
      if (!status.ok()) {
        return status;
      }
    
      return exporter_->Export(module);
    }
    
    void TacModule::RegisterExtraDialects(mlir::DialectRegistry& registry) {
      registry.appendTo(registry_);
    }
    }  // namespace tac
    }  // namespace TFL
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 5.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/experimental/tac/tac_translate.cc

      options.legalize_to_tflite_ops = true;
      mlir::TFL::tac::TacModule tac_module(options);
      mlir::DialectRegistry registry;
      mlir::RegisterAllTensorFlowDialects(registry);
      tac_module.RegisterExtraDialects(registry);
      tac_module.SetImporter(CreateTfLiteImporter());
      tac_module.SetExporter(CreateTfLiteExporter(options.hardware_backends));
      return tac_module.Run();
    }
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 22 14:25:57 UTC 2022
    - 5.9K bytes
    - Viewed (0)
Back to top