Home
last modified time | relevance | path

Searched refs:ParsedKey (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Drendezvous.h57 struct ParsedKey { struct
65 ParsedKey() {} in ParsedKey() argument
66 ParsedKey(const ParsedKey& b) { *this = b; } in ParsedKey() function
68 ParsedKey& operator=(const ParsedKey& b); argument
89 virtual Status Send(const ParsedKey& key, const Args& args, const Tensor& val, argument
102 virtual void RecvAsync(const ParsedKey& key, const Args& args,
106 Status Recv(const ParsedKey& key, const Args& args, Tensor* val,
108 Status Recv(const ParsedKey& key, const Args& args, Tensor* val,
137 static Status ParseKey(StringPiece key, ParsedKey* out);
Drendezvous.cc38 Rendezvous::ParsedKey& Rendezvous::ParsedKey::operator=(const ParsedKey& b) { in operator =()
88 Status Rendezvous::ParseKey(StringPiece key, ParsedKey* out) { in ParseKey()
119 Status RendezvousInterface::Recv(const ParsedKey& key, const Args& recv_args, in Recv()
145 Status RendezvousInterface::Recv(const ParsedKey& key, const Args& args, in Recv()
156 Status Send(const ParsedKey& key, const Args& send_args, const Tensor& val, in Send()
161 void RecvAsync(const ParsedKey& key, const Args& recv_args, in RecvAsync()
Dlocal_rendezvous.h41 Status Send(const Rendezvous::ParsedKey& key,
44 void RecvAsync(const Rendezvous::ParsedKey& key,
Drendezvous_test.cc51 Rendezvous::ParsedKey parsed; in TEST()
100 Rendezvous::ParsedKey MakeKey(const string& name) { in MakeKey()
104 Rendezvous::ParsedKey k; in MakeKey()
109 const Rendezvous::ParsedKey& KeyFoo() { in KeyFoo()
110 static auto* key = new Rendezvous::ParsedKey(MakeKey("foo")); in KeyFoo()
114 const Rendezvous::ParsedKey& KeyBar() { in KeyBar()
115 static auto* key = new Rendezvous::ParsedKey(MakeKey("bar")); in KeyBar()
Dlocal_rendezvous.cc108 Status LocalRendezvous::Send(const Rendezvous::ParsedKey& key, in Send()
156 void LocalRendezvous::RecvAsync(const Rendezvous::ParsedKey& key, in RecvAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.h78 void RecvLocalAsync(int64 step_id, const Rendezvous::ParsedKey& parsed,
82 Status RecvLocal(int64 step_id, const Rendezvous::ParsedKey& parsed,
128 Status Send(const ParsedKey& key, const Rendezvous::Args& args,
135 void RecvAsync(const ParsedKey& key, const Rendezvous::Args& args,
150 void RecvLocalAsync(const ParsedKey& parsed, DoneCallback done);
153 virtual void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
191 const ParsedKey parsed;
194 DeferredCall(const ParsedKey& parsed, DoneCallback done);
212 Status ValidateDevices(const Rendezvous::ParsedKey& parsed, bool is_src);
218 void SameWorkerRecvDone(const Rendezvous::ParsedKey& parsed,
[all …]
Dbase_rendezvous_mgr.cc71 const Rendezvous::ParsedKey& parsed, in RecvLocalAsync()
85 const Rendezvous::ParsedKey& parsed, in RecvLocal()
186 Status BaseRemoteRendezvous::Send(const Rendezvous::ParsedKey& parsed, in Send()
208 Status BaseRemoteRendezvous::ValidateDevices(const ParsedKey& parsed, in ValidateDevices()
235 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& send_args, in SameWorkerRecvDone()
312 void BaseRemoteRendezvous::RecvAsync(const ParsedKey& parsed, in RecvAsync()
359 void BaseRemoteRendezvous::RecvLocalAsync(const ParsedKey& parsed, in RecvLocalAsync()
381 void BaseRemoteRendezvous::RecvLocalAsyncInternal(const ParsedKey& parsed, in RecvLocalAsyncInternal()
462 BaseRemoteRendezvous::DeferredCall::DeferredCall(const ParsedKey& parsed, in DeferredCall()
Drendezvous_mgr_interface.h84 const Rendezvous::ParsedKey& parsed,
88 virtual Status RecvLocal(int64 step_id, const Rendezvous::ParsedKey& parsed,
Dworker.h115 Status PrepareRecvTensor(const Rendezvous::ParsedKey& parsed,
Dworker.cc433 Status Worker::PrepareRecvTensor(const Rendezvous::ParsedKey& parsed, in PrepareRecvTensor()
/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_mgr.h48 Status Send(const ParsedKey& key, const Rendezvous::Args& args,
50 void RecvAsync(const ParsedKey& key, const Rendezvous::Args& args,
77 Status Send(const ParsedKey& key, const Rendezvous::Args& args,
79 void RecvAsync(const ParsedKey& key, const Rendezvous::Args& args,
Drendezvous_mgr.cc37 const Rendezvous::ParsedKey& parsed, in SameWorkerRecvDone()
110 const RendezvousInterface::ParsedKey& parsed, in IntraProcessRecvAsyncImpl()
151 Status RefCountedIntraProcessRendezvous::Send(const ParsedKey& key, in Send()
159 void RefCountedIntraProcessRendezvous::RecvAsync(const ParsedKey& key, in RecvAsync()
176 Status PrivateIntraProcessRendezvous::Send(const ParsedKey& key, in Send()
184 void PrivateIntraProcessRendezvous::RecvAsync(const ParsedKey& key, in RecvAsync()
Drendezvous_util.cc42 Rendezvous::ParsedKey parsed; in SendTensorsToRendezvous()
73 std::tuple<string, Tensor*, Rendezvous::ParsedKey, AllocatorAttributes>> in RecvOutputsFromRendezvousAsync()
76 Rendezvous::ParsedKey parsed; in RecvOutputsFromRendezvousAsync()
95 Rendezvous::ParsedKey parsed = std::get<2>(p); in RecvOutputsFromRendezvousAsync()
125 Rendezvous::ParsedKey parsed; in RecvOutputsFromRendezvous()
Dkernel_benchmark_testlib.cc154 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
160 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
171 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
177 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
Dgraph_runner.cc52 Status Send(const ParsedKey& parsed, const Args& send_args, const Tensor& val, in Send()
67 void RecvAsync(const ParsedKey& parsed, const Args& recv_args, in RecvAsync()
136 Rendezvous::ParsedKey parsed; in Run()
198 Rendezvous::ParsedKey parsed; in Run()
Dexecutor_test.cc138 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key()
140 Rendezvous::ParsedKey result; in Key()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc44 Rendezvous::ParsedKey MakeKey(const string& s) { in MakeKey()
45 Rendezvous::ParsedKey key; in MakeKey()
93 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
113 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
147 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
171 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
197 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
228 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
Drpc_rendezvous_mgr.cc46 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
212 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.h31 Rendezvous::ParsedKey parsed_key_;
44 Rendezvous::ParsedKey parsed_key_;
Dsendrecv_ops_test.cc30 Status Send(const ParsedKey& key, const Args& args, const Tensor& val, in Send()
34 void RecvAsync(const ParsedKey& key, const Args& args, in RecvAsync()
Dsendrecv_ops.cc98 Rendezvous::ParsedKey in_loop_parsed; in Compute()
185 Rendezvous::ParsedKey in_loop_parsed; in ComputeAsync()
/external/tensorflow/tensorflow/c/experimental/
Dnetwork_test.cc156 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key()
158 Rendezvous::ParsedKey result; in Key()
Drendezvous_internal.h93 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
Drendezvous.cc45 void CRemoteRendezvous::RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed, in RecvFromRemoteAsync()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor_test.cc130 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key()
132 Rendezvous::ParsedKey result; in Key()

12