Searched refs:BaseRemoteRendezvous (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 60 BaseRemoteRendezvous* BaseRendezvousMgr::FindOrCreate(int64 step_id) { in FindOrCreate()
119 BaseRemoteRendezvous::BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id) in BaseRemoteRendezvous() function in tensorflow::BaseRemoteRendezvous
125 BaseRemoteRendezvous::~BaseRemoteRendezvous() { in ~BaseRemoteRendezvous()
138 Status BaseRemoteRendezvous::Initialize(WorkerSession* session) { in Initialize()
163 WorkerSession* BaseRemoteRendezvous::session() { in session()
168 bool BaseRemoteRendezvous::is_initialized() { in is_initialized()
173 Status BaseRemoteRendezvous::Send(const Rendezvous::ParsedKey& parsed, in Send()
195 Status BaseRemoteRendezvous::ValidateDevices(const ParsedKey& parsed, in ValidateDevices()
221 void BaseRemoteRendezvous::SameWorkerRecvDone( in SameWorkerRecvDone()
292 bool BaseRemoteRendezvous::IsSameWorker(DeviceNameUtils::ParsedName src, in IsSameWorker()
[all …]
|
| D | base_rendezvous_mgr.h | 39 class BaseRemoteRendezvous; variable
92 virtual BaseRemoteRendezvous* Create(int64 step_id,
97 typedef absl::flat_hash_map<int64, BaseRemoteRendezvous*> Table;
105 BaseRemoteRendezvous* FindOrCreate(int64 step_id);
116 class BaseRemoteRendezvous : public RemoteRendezvous {
118 BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id);
169 ~BaseRemoteRendezvous() override;
219 TF_DISALLOW_COPY_AND_ASSIGN(BaseRemoteRendezvous);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 41 class RpcRemoteRendezvous : public BaseRemoteRendezvous {
44 : BaseRemoteRendezvous(env, step_id) {} in RpcRemoteRendezvous()
313 BaseRemoteRendezvous* RpcRendezvousMgr::Create(int64 step_id, in Create()
|
| D | rpc_rendezvous_mgr.h | 50 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|