Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetTargetHardwareFactory (0.21 sec)

  1. tensorflow/compiler/mlir/lite/experimental/tac/hardwares/target_hardware.h

    // If not found nullptr is returned.
    // DEPRECATED: Do not use, prefer GetTargetHardwareFactory instead.
    const TargetHardware* GetTargetHardware(const std::string& hardware_name);
    
    // Returns the factory method for the requested hardware if present.
    std::function<std::unique_ptr<TargetHardware>()> GetTargetHardwareFactory(
        const std::string& hardware_name);
    
    namespace internal {
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 09 21:39:59 UTC 2023
    - 7.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/tac/tac_module.cc

    TacModule::InstantiateBackends() {
      std::vector<std::unique_ptr<tac::TargetHardware>> backends;
      for (const auto& hardware_name : options_.hardware_backends) {
        auto factory = tac::GetTargetHardwareFactory(hardware_name);
        backends.emplace_back(factory());
        backends.back()->Init();
      }
      return backends;
    }
    
    absl::Status TacModule::Run() {
      // Construct all backends.
    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/hardwares/target_hardware.cc

        if (hardware.unique_name == canonical_name) {
          return hardware.target_hardware.get();
        }
      }
      return nullptr;
    }
    
    std::function<std::unique_ptr<TargetHardware>()> GetTargetHardwareFactory(
        const std::string& hardware_name) {
      const std::string canonical_name = GetCanonicalHardwareName(hardware_name);
      // Just loop for now, we don't expect number of hardwares to be huge.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 09 21:39:59 UTC 2023
    - 9.9K bytes
    - Viewed (0)
Back to top