Searched refs:DynamicDeviceMgr (Results 1 – 14 of 14) sorted by relevance
29 DynamicDeviceMgr::DynamicDeviceMgr() : cpu_device_(nullptr) {} in DynamicDeviceMgr() function in tensorflow::DynamicDeviceMgr31 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 …]
70 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()
118 class DynamicDeviceMgr : public DeviceMgr {121 DynamicDeviceMgr();123 ~DynamicDeviceMgr() override;185 TF_DISALLOW_COPY_AND_ASSIGN(DynamicDeviceMgr);
112 device_mgr_ = std::make_unique<DynamicDeviceMgr>(); in ProcessFunctionLibraryRuntimeTest()320 std::unique_ptr<DynamicDeviceMgr> device_mgr_;
44 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_;
111 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()
145 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()
305 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,
96 device_manager_ = new DynamicDeviceMgr(); in InitDeviceManager()108 DynamicDeviceMgr* device_manager_;
71 device_manager_ = new DynamicDeviceMgr(); in InitDeviceManager()81 DynamicDeviceMgr* device_manager_;
65 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()
54 auto remote_device_mgr = absl::make_unique<DynamicDeviceMgr>(); in TEST()
1204 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()
299 DynamicDeviceMgr remote_device_mgr; in TEST()