Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Drendezvous.h59 struct ParsedKey { struct
67 ParsedKey() {} in ParsedKey() argument
68 ParsedKey(const ParsedKey& b) { *this = b; } in ParsedKey() function
70 ParsedKey& operator=(const ParsedKey& b); argument
91 virtual Status Send(const ParsedKey& key, const Args& args, const Tensor& val, argument
104 virtual void RecvAsync(const ParsedKey& key, const Args& args,
108 Status Recv(const ParsedKey& key, const Args& args, Tensor* val,
110 Status Recv(const ParsedKey& key, const Args& args, Tensor* val,
178 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.h45 Status Send(const Rendezvous::ParsedKey& key,
48 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.cc109 Status LocalRendezvous::Send(const Rendezvous::ParsedKey& key, in Send()
167 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,
125 Status Send(const ParsedKey& key, const Rendezvous::Args& args,
132 void RecvAsync(const ParsedKey& key, const Rendezvous::Args& args,
147 void RecvLocalAsync(const ParsedKey& parsed, DoneCallback done);
150 virtual void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
186 const ParsedKey parsed;
189 DeferredCall(const ParsedKey& parsed, DoneCallback done);
205 Status ValidateDevices(const Rendezvous::ParsedKey& parsed, bool is_src);
211 void SameWorkerRecvDone(const Rendezvous::ParsedKey& parsed,
[all …]
Dbase_rendezvous_mgr.cc72 const Rendezvous::ParsedKey& parsed, in RecvLocalAsync()
86 const Rendezvous::ParsedKey& parsed, in RecvLocal()
173 Status BaseRemoteRendezvous::Send(const Rendezvous::ParsedKey& parsed, in Send()
195 Status BaseRemoteRendezvous::ValidateDevices(const ParsedKey& parsed, in ValidateDevices()
222 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& send_args, in SameWorkerRecvDone()
297 void BaseRemoteRendezvous::RecvAsync(const ParsedKey& parsed, in RecvAsync()
343 void BaseRemoteRendezvous::RecvLocalAsync(const ParsedKey& parsed, in RecvLocalAsync()
365 void BaseRemoteRendezvous::RecvLocalAsyncInternal(const ParsedKey& parsed, in RecvLocalAsyncInternal()
442 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.cc436 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,
Dkernel_benchmark_testlib.cc181 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
187 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
199 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
205 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
233 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
239 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
250 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
256 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
Drendezvous_mgr.cc37 const Rendezvous::ParsedKey& parsed, in SameWorkerRecvDone()
117 const RendezvousInterface::ParsedKey& parsed, in IntraProcessRecvAsyncImpl()
158 Status RefCountedIntraProcessRendezvous::Send(const ParsedKey& key, in Send()
166 void RefCountedIntraProcessRendezvous::RecvAsync(const ParsedKey& key, in RecvAsync()
183 Status PrivateIntraProcessRendezvous::Send(const ParsedKey& key, in Send()
191 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()
Dgraph_runner.cc54 Status Send(const ParsedKey& parsed, const Args& send_args, const Tensor& val, in Send()
69 void RecvAsync(const ParsedKey& parsed, const Args& recv_args, in RecvAsync()
138 Rendezvous::ParsedKey parsed; in Run()
196 Rendezvous::ParsedKey parsed; in Run()
Dexecutor_test.cc143 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key()
145 Rendezvous::ParsedKey result; in Key()
Ddirect_session.cc1150 Rendezvous::ParsedKey parsed; in SendPRunInputs()
1194 Rendezvous::ParsedKey parsed; in RecvPRunOutputs()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc47 Rendezvous::ParsedKey MakeKey(const string& s) { in MakeKey()
48 Rendezvous::ParsedKey key; in MakeKey()
141 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
161 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
195 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
219 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
260 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
292 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
311 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
Drpc_rendezvous_mgr.cc47 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
234 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
Dgrpc_worker_service.cc492 Rendezvous::ParsedKey parsed; in GrpcRecvTensorAsync()
/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.h33 Rendezvous::ParsedKey parsed_key_;
48 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.cc102 Rendezvous::ParsedKey in_loop_parsed; in Compute()
208 Rendezvous::ParsedKey in_loop_parsed; in ComputeAsync()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dhost_compute_ops.cc126 std::vector<Rendezvous::ParsedKey> parsed_key(ctx->num_outputs()); in ComputeAsync()
281 Rendezvous::ParsedKey parsed_key; in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor_test.cc132 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key()
134 Rendezvous::ParsedKey result; in Key()