HomeSort by relevance Sort by last modified time
    Searched defs:device_mgr (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/distributed_runtime/
worker_env.h 44 // The local devices of this worker. Devices are owned by the device_mgr.
49 // device_mgr manages local devices (cpu and gpu). The WorkerService
52 // Note: Please use the device_mgr associated with your session if appropriate
53 // instead of this one. Using this device_mgr does not support ClusterSpec
55 DeviceMgr* device_mgr = nullptr; member in struct:tensorflow::WorkerEnv
worker_session.h 21 #include "tensorflow/core/common_runtime/device_mgr.h"
44 // in all except the SessionMgr.legacy_session_. legacy_session_.device_mgr
45 // == worker_env_.device_mgr, which holds the true devices.
46 const std::unique_ptr<DeviceMgr> device_mgr; member in struct:tensorflow::WorkerSession
51 // Note: graph_mgr must be deleted before device_mgr!
58 std::unique_ptr<DeviceMgr> device_mgr,
  /external/tensorflow/tensorflow/core/grappler/clusters/
single_machine.cc 23 #include "tensorflow/core/common_runtime/device_mgr.h"
211 const DeviceMgr* device_mgr; local
212 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr));
213 std::vector<Device*> devices = device_mgr->ListDevices();
400 const DeviceMgr* device_mgr; local
401 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr));
402 std::vector<Device*> devices = device_mgr->ListDevices();
  /external/tensorflow/tensorflow/c/
c_api_internal.h 134 const tensorflow::DeviceMgr* device_mgr; // Owned by session. member in struct:TF_Session
135 std::vector<tensorflow::Device*> devices; // Owned by device_mgr.

Completed in 392 milliseconds