| /external/tensorflow/tensorflow/contrib/gdr/ |
| gdr_rendezvous_mgr.cc | 43 const Rendezvous::Args& recv_args, int64 step_id,
48 recv_args_(recv_args) {
109 const Rendezvous::Args& recv_args() const { return recv_args_; } function in class:tensorflow::__anon39175::GdrRecvTensorCall
135 const Rendezvous::Args& recv_args,
145 done(s, Args(), recv_args, Tensor{}, false);
153 done(s, Args(), recv_args, Tensor{}, false);
161 done(s, Args(), recv_args, Tensor{}, false);
168 recv_args, step_id_, parsed.FullKey());
181 done(s, Args(), call->recv_args(), call->tensor(), call->is_dead());
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| rpc_rendezvous_mgr.cc | 63 const Rendezvous::Args& recv_args, Rendezvous::DoneCallback done) {
67 recv_args_ = recv_args;
121 const Rendezvous::Args& recv_args() const { return recv_args_; } function in class:tensorflow::__anon39544::RpcRecvTensorCall
208 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args,
237 done(s, Args(), recv_args, Tensor{}, false);
241 call->Init(rwi, step_id_, parsed.FullKey(), recv_args.alloc_attrs, dst_device,
242 recv_args, std::move(done));
255 call->done()(s, Args(), call->recv_args(), call->tensor(), call->is_dead());
|
| /external/tensorflow/tensorflow/core/framework/ |
| rendezvous.cc | 117 Status Rendezvous::Recv(const ParsedKey& key, const Args& recv_args,
121 RecvAsync(key, recv_args,
123 const Args& recv_args, const Tensor& v,
191 item->waiter(Status::OK(), send_args, item->recv_args, val, is_dead);
196 void RecvAsync(const ParsedKey& key, const Args& recv_args,
206 done(s, Args(), recv_args, Tensor(), false);
216 item->recv_args = recv_args;
217 if (item->recv_args.device_context) {
218 item->recv_args.device_context->Ref()
234 done(Status::OK(), item->send_args, recv_args, item->value, item->is_dead); variable
264 Args recv_args; member in struct:tensorflow::LocalRendezvousImpl::Item
[all...] |
| rendezvous_test.cc | 297 const Rendezvous::Args& recv_args, const Tensor& val, bool is_dead) { local
|
