Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for DeviceMgr (0.05 sec)

  1. tensorflow/c/c_api_internal.h

    #include "tensorflow/core/platform/mutex.h"
    #include "tensorflow/core/platform/status.h"
    #include "tensorflow/core/platform/types.h"
    #include "tensorflow/core/public/session.h"
    
    namespace tensorflow {
    class Device;
    class DeviceMgr;
    class ServerInterface;
    }  // namespace tensorflow
    
    // Internal structures used by the C API. These are likely to change and should
    // not be depended on.
    
    struct TF_SessionOptions {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat May 13 00:49:12 UTC 2023
    - 7.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

      status->status = tensorflow::DeviceFactory::AddDevices(
          opts->session_options.options, "/job:localhost/replica:0/task:0",
          &devices);
      if (!status->status.ok()) return nullptr;
      std::unique_ptr<tensorflow::DeviceMgr> device_mgr(
          new tensorflow::DynamicDeviceMgr(std::move(devices)));
    
      auto r = tsl::core::RefCountPtr<tensorflow::IntraProcessRendezvous>(
          new tensorflow::IntraProcessRendezvous(device_mgr.get()));
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top