Home
last modified time | relevance | path

Searched defs:local_device_mgr (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dserver_lib.h93 tensorflow::DeviceMgr* local_device_mgr; member
Dsession_mgr.cc256 const DeviceMgr* local_device_mgr = worker_session->device_mgr(); in UpdateSession() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle_test.cc254 StaticDeviceMgr local_device_mgr(std::move(d0)); in TEST() local
286 StaticDeviceMgr local_device_mgr(std::move(d_local)); in TEST() local
414 StaticDeviceMgr local_device_mgr(std::move(devices)); in TEST() local
Dcontext.h349 tensorflow::DeviceMgr* local_device_mgr() const { in local_device_mgr() function
Dcontext.cc1463 Rendezvous* r, DeviceMgr* local_device_mgr, int keep_alive_secs, in InitializeRemoteMaster()
1560 uint64 context_view_id, Rendezvous* r, DeviceMgr* local_device_mgr, in SetMasterContextState()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.h72 DeviceMgr* local_device_mgr = nullptr; member
Dgrpc_server_lib.cc571 DeviceMgr* local_device_mgr, in Create()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.cc1002 auto* local_device_mgr = dynamic_cast<tensorflow::DynamicDeviceMgr*>( in EnableCollectiveOps() local