| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 64 const Rendezvous::Args& recv_args, const Tensor& in, Tensor* out, in SameWorkerRecvDone() argument
71 (recv_args.alloc_attrs.on_host() || parsed.dst.type == "CPU"); in SameWorkerRecvDone()
100 AllocatorAttributes attr = recv_args.alloc_attrs; in SameWorkerRecvDone()
102 recv_args.alloc_attrs.gpu_compatible()); in SameWorkerRecvDone()
111 recv_args.device_context, src_device, dst_device, in SameWorkerRecvDone()
112 send_args.alloc_attrs, recv_args.alloc_attrs, &in, out, in SameWorkerRecvDone()
117 const Rendezvous::Args& recv_args, in RecvAsync() argument
123 parsed, recv_args, in RecvAsync()
129 const Rendezvous::Args& recv_args, const Tensor& in, in RecvAsync()
137 [send_args, recv_args, out, is_dead](DoneCallback done, in RecvAsync()
[all …]
|
| D | graph_runner.cc | 67 void RecvAsync(const ParsedKey& parsed, const Args& recv_args, in RecvAsync() argument
80 done(status, Args{}, recv_args, tensor, false); in RecvAsync()
|
| D | rendezvous_mgr.h | 82 const Rendezvous::Args& recv_args, const Tensor& in,
|
| D | rendezvous_util.cc | 104 const Rendezvous::Args& recv_args, in RecvOutputsFromRendezvousAsync()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.cc | 117 Status Rendezvous::Recv(const ParsedKey& key, const Args& recv_args, in Recv() argument
121 RecvAsync(key, recv_args, in Recv()
123 const Args& recv_args, const Tensor& v, in Recv()
191 item->waiter(Status::OK(), send_args, item->recv_args, val, is_dead); in Send()
196 void RecvAsync(const ParsedKey& key, const Args& recv_args, in RecvAsync() argument
206 done(s, Args(), recv_args, Tensor(), false); in RecvAsync()
216 item->recv_args = recv_args; in RecvAsync()
217 if (item->recv_args.device_context) { in RecvAsync()
218 item->recv_args.device_context->Ref(); in RecvAsync()
234 done(Status::OK(), item->send_args, recv_args, item->value, item->is_dead); in RecvAsync()
[all …]
|
| D | rendezvous_test.cc | 303 const Rendezvous::Args& recv_args, const Tensor& val, bool is_dead) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 77 const Rendezvous::Args& recv_args, const Tensor& v, bool dead) { in RecvLocalAsync() argument
79 done(s, send_args, recv_args, v, dead); in RecvLocalAsync()
93 const Rendezvous::Args& recv_args, in RecvLocal()
231 const Rendezvous::Args& recv_args, const Tensor& in, Tensor* out, in SameWorkerRecvDone() argument
238 (recv_args.alloc_attrs.on_host() || parsed.dst.type == "CPU"); in SameWorkerRecvDone()
268 AllocatorAttributes attr = recv_args.alloc_attrs; in SameWorkerRecvDone()
270 recv_args.alloc_attrs.gpu_compatible()); in SameWorkerRecvDone()
282 recv_args.device_context, src_device, dst_device, in SameWorkerRecvDone()
283 send_args.alloc_attrs, recv_args.alloc_attrs, &in, out, in SameWorkerRecvDone()
293 const Rendezvous::Args& recv_args, in RecvAsync() argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_rendezvous_mgr.cc | 43 const Rendezvous::Args& recv_args, int64 step_id, in GdrRecvTensorCall() argument
48 recv_args_(recv_args) { in GdrRecvTensorCall()
104 const Rendezvous::Args& recv_args() const { return recv_args_; } in recv_args() function in tensorflow::__anon5c78cdd40111::GdrRecvTensorCall
130 const Rendezvous::Args& recv_args, in RecvFromRemoteAsync() argument
140 done(s, Args(), recv_args, Tensor{}, false); in RecvFromRemoteAsync()
148 done(s, Args(), recv_args, Tensor{}, false); in RecvFromRemoteAsync()
156 done(s, Args(), recv_args, Tensor{}, false); in RecvFromRemoteAsync()
163 recv_args, step_id_, parsed.FullKey()); in RecvFromRemoteAsync()
191 done(s, Args(), call->recv_args(), call->tensor(), call->is_dead()); in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 63 const Rendezvous::Args& recv_args, Rendezvous::DoneCallback done) { in Init() argument
67 recv_args_ = recv_args; in Init()
130 const Rendezvous::Args& recv_args() const { return recv_args_; } in recv_args() function in tensorflow::__anon9a114d310111::RpcRecvTensorCall
217 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync() argument
249 done(s, Args(), recv_args, Tensor{}, false); in RecvFromRemoteAsync()
253 call->Init(rwi, step_id_, parsed.FullKey(), recv_args.alloc_attrs, dst_device, in RecvFromRemoteAsync()
254 recv_args, std::move(done)); in RecvFromRemoteAsync()
282 call->done()(s, Args(), call->recv_args(), call->tensor(), call->is_dead()); in RecvFromRemoteAsync()
|
| D | rpc_rendezvous_mgr_test.cc | 189 const Rendezvous::Args recv_args, const Tensor& val, in TEST_F()
|
| D | grpc_worker_service.cc | 481 const Rendezvous::Args& recv_args, const Tensor& val, in GrpcRecvTensorAsync()
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.cc | 63 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync() argument
80 done(s, Args(), recv_args, Tensor{}, false); in RecvFromRemoteAsync()
105 [this, parsed, recv_args, done, dst, temp1, in RecvFromRemoteAsync()
121 recv_args.alloc_attrs, &val); in RecvFromRemoteAsync()
124 tr.InitAlloc(dst_device, recv_args.alloc_attrs); in RecvFromRemoteAsync()
134 done(s, Args(), recv_args, val, mpi_response.response().is_dead()); in RecvFromRemoteAsync()
163 const Rendezvous::Args& recv_args, in AddRequest()
201 const Rendezvous::Args& recv_args, const Tensor& val, bool is_dead) { in AddRequest() argument
259 send_cb, status, send_args, recv_args, val, is_dead, mpi_send_call); in AddRequest()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_rendezvous_mgr.cc | 48 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync() argument
61 done(s, Args(), recv_args, Tensor{}, false); in RecvFromRemoteAsync()
73 done(s, Args(), recv_args, Tensor(), true); in RecvFromRemoteAsync()
78 rc->InsertTensorRequest(key, step_id_, dst_dev, recv_args, done); in RecvFromRemoteAsync()
|
| D | rdma.h | 241 const Rendezvous::Args recv_args,
340 const Rendezvous::Args& recv_args, const Tensor& in,
433 const Rendezvous::Args recv_args,
|
| D | rdma.cc | 669 const Rendezvous::Args recv_args, in InsertTensorRequest() argument
677 recv_args, done); in InsertTensorRequest()
1006 const Rendezvous::Args& recv_args, const Tensor& in, in Start()
1010 RecvHandler(parsed, send_args, recv_args, in, is_dead); in Start()
1042 const Rendezvous::Args& recv_args, in RecvHandler() argument
1478 Device* dst_dev, const Rendezvous::Args recv_args, in RdmaTensorRequest() argument
1485 recv_args_(recv_args), in RdmaTensorRequest()
1510 Rendezvous::Args recv_args = std::move(recv_args_); in Done() local
1515 done(s, Rendezvous::Args(), recv_args, val, is_dead); in Done()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sendrecv_ops.cc | 153 const Rendezvous::Args& recv_args, const Tensor& val, in make_recv_callback()
|