Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 12 of 12 for StaticDeviceMgr (0.17 sec)

  1. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

      if (devices.empty()) {
        return errors::NotFound(
            "Failed to create a CPU device for EncapsulateSubgraphsPass");
      }
    
      std::unique_ptr<DeviceMgr> device_mgr =
          std::make_unique<StaticDeviceMgr>(std::move(devices));
      const auto* config = &options.session_options->config;
      std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(
          new ProcessFunctionLibraryRuntime(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

      std::vector<std::unique_ptr<Device>> devices;
      TF_CHECK_OK(DeviceFactory::AddDevices(
          session_options, "/job:localhost/replica:0/task:0", &devices));
      OptimizerOptions opts;
      auto device_mgr = std::make_unique<StaticDeviceMgr>(std::move(devices));
      auto pflr = std::make_unique<ProcessFunctionLibraryRuntime>(
          device_mgr.get(), Env::Default(), /*config=*/nullptr,
          TF_GRAPH_DEF_VERSION, lib_def.get(), opts,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
Back to top