Home
last modified time | relevance | path

Searched refs:DynamicDeviceMgr (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddynamic_device_mgr.cc29 DynamicDeviceMgr::DynamicDeviceMgr() : cpu_device_(nullptr) {} in DynamicDeviceMgr() function in tensorflow::DynamicDeviceMgr
31 DynamicDeviceMgr::~DynamicDeviceMgr() { in ~DynamicDeviceMgr()
39 void DynamicDeviceMgr::ListDeviceAttributes( in ListDeviceAttributes()
48 std::vector<Device*> DynamicDeviceMgr::ListDevices() const { in ListDevices()
58 string DynamicDeviceMgr::DebugString() const { in DebugString()
67 string DynamicDeviceMgr::DeviceMappingString() const { in DeviceMappingString()
79 Status DynamicDeviceMgr::LookupDevice(StringPiece name, Device** device) const { in LookupDevice()
95 bool DynamicDeviceMgr::ContainsDevice(int64 device_incarnation) const { in ContainsDevice()
100 void DynamicDeviceMgr::ClearContainers( in ClearContainers()
119 int DynamicDeviceMgr::NumDeviceType(const string& type) const { in NumDeviceType()
[all …]
Ddynamic_device_mgr_test.cc70 auto dm = MakeUnique<DynamicDeviceMgr>(); in TEST()
86 auto dm = MakeUnique<DynamicDeviceMgr>(); in TEST()
110 auto dm = MakeUnique<DynamicDeviceMgr>(); in TEST()
135 auto dm = MakeUnique<DynamicDeviceMgr>(); in TEST()
151 auto dm = MakeUnique<DynamicDeviceMgr>(); in TEST()
170 auto dm = MakeUnique<DynamicDeviceMgr>(); in TEST()
187 auto dm = MakeUnique<DynamicDeviceMgr>(); in TEST()
200 auto dm = MakeUnique<DynamicDeviceMgr>(); in TEST()
Ddevice_mgr.h118 class DynamicDeviceMgr : public DeviceMgr {
121 DynamicDeviceMgr();
123 ~DynamicDeviceMgr() override;
185 TF_DISALLOW_COPY_AND_ASSIGN(DynamicDeviceMgr);
Dprocess_function_library_runtime_test.cc112 device_mgr_ = std::make_unique<DynamicDeviceMgr>(); in ProcessFunctionLibraryRuntimeTest()
320 std::unique_ptr<DynamicDeviceMgr> device_mgr_;
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_session.h44 DynamicDeviceMgr* remote_device_mgr() { return remote_device_mgr_.get(); } in remote_device_mgr()
63 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr);
69 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr);
95 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr);
118 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr_;
Dworker_session.cc111 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) in WorkerSession()
148 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) { in CreateWithBorrowedDeviceMgr()
158 std::unique_ptr<DynamicDeviceMgr> remote_device_mgr) in WorkerSession()
Dsession_mgr.cc145 std::unique_ptr<DynamicDeviceMgr> remote_devices; in CreateSession()
147 remote_devices = MakeUnique<DynamicDeviceMgr>(); in CreateSession()
161 std::unique_ptr<DynamicDeviceMgr> remote_devices; in CreateSession()
163 remote_devices = MakeUnique<DynamicDeviceMgr>(); in CreateSession()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h305 const tensorflow::DynamicDeviceMgr* remote_device_mgr() const { in remote_device_mgr()
309 tensorflow::DynamicDeviceMgr* GetOwnedRemoteDeviceMgr() { in GetOwnedRemoteDeviceMgr()
369 std::unique_ptr<DynamicDeviceMgr> remote_device_manager,
392 DynamicDeviceMgr* remote_device_mgr,
586 OwnedOrUnownedHelper<DynamicDeviceMgr> remote_device_manager_;
671 std::unique_ptr<DynamicDeviceMgr> remote_device_manager,
Dplacement_test.cc96 device_manager_ = new DynamicDeviceMgr(); in InitDeviceManager()
108 DynamicDeviceMgr* device_manager_;
Dcontext_test.cc71 device_manager_ = new DynamicDeviceMgr(); in InitDeviceManager()
81 DynamicDeviceMgr* device_manager_;
Dcontext_distributed_manager.cc65 tensorflow::DynamicDeviceMgr* remote_device_mgr) { in AddRemoteDevicesToMgr()
99 std::unique_ptr<tensorflow::DynamicDeviceMgr>* device_mgr) { in GetAllRemoteDevices()
100 auto remote_device_mgr = std::make_unique<tensorflow::DynamicDeviceMgr>(); in GetAllRemoteDevices()
109 tensorflow::DynamicDeviceMgr* remote_device_mgr) { in RemoveRemoteDevicesFromMgr()
486 std::unique_ptr<tensorflow::DynamicDeviceMgr> new_remote_device_mgr; in UpdateContextWithServerDef()
487 tensorflow::DynamicDeviceMgr* remote_device_mgr = nullptr; in UpdateContextWithServerDef()
Dexecute_node_test.cc54 auto remote_device_mgr = absl::make_unique<DynamicDeviceMgr>(); in TEST()
Dcontext.cc1204 std::unique_ptr<DynamicDeviceMgr> remote_device_manager, in InitializeRemoteMaster()
1302 std::unique_ptr<DynamicDeviceMgr> remote_device_manager, uint64 context_id, in SetMasterContextState()
1421 DynamicDeviceMgr* remote_device_mgr, in InitializeRemoteWorker()
Dtensor_handle_test.cc299 DynamicDeviceMgr remote_device_mgr; in TEST()