| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 39 const Rendezvous::Args& recv_args, const Tensor& in, in SameWorkerRecvDone() argument
46 (recv_args.alloc_attrs.on_host() || parsed.dst.type == "CPU"); in SameWorkerRecvDone()
79 AllocatorAttributes attr = recv_args.alloc_attrs; in SameWorkerRecvDone()
81 recv_args.alloc_attrs.gpu_compatible()); in SameWorkerRecvDone()
110 parsed.edge_name, send_args.device_context, recv_args.device_context, in SameWorkerRecvDone()
111 src_device, dst_device, send_args.alloc_attrs, recv_args.alloc_attrs, &in, in SameWorkerRecvDone()
118 const Rendezvous::Args& recv_args, in IntraProcessRecvAsyncImpl() argument
125 parsed, recv_args, in IntraProcessRecvAsyncImpl()
128 const Rendezvous::Args& recv_args, const Tensor& in, in IntraProcessRecvAsyncImpl()
135 auto final_callback = [send_args, recv_args, out, is_dead, in IntraProcessRecvAsyncImpl()
[all …]
|
| D | graph_runner.cc | 69 void RecvAsync(const ParsedKey& parsed, const Args& recv_args, in RecvAsync() argument
82 done(status, Args{}, recv_args, tensor, false); in RecvAsync()
|
| D | rendezvous_util.cc | 104 const Rendezvous::Args& recv_args, in RecvOutputsFromRendezvousAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 77 const Rendezvous::Args& recv_args, const Tensor& v, in RecvLocalAsync()
80 done(s, send_args, recv_args, v, dead); in RecvLocalAsync()
93 const Rendezvous::Args& recv_args, in RecvLocal() argument
223 const Rendezvous::Args& recv_args, const Tensor& in, Tensor* out, in SameWorkerRecvDone() argument
230 (recv_args.alloc_attrs.on_host() || parsed.dst.type == "CPU"); in SameWorkerRecvDone()
264 AllocatorAttributes attr = recv_args.alloc_attrs; in SameWorkerRecvDone()
266 recv_args.alloc_attrs.gpu_compatible()); in SameWorkerRecvDone()
287 parsed.edge_name, send_args.device_context, recv_args.device_context, in SameWorkerRecvDone()
288 src_device, dst_device, send_args.alloc_attrs, recv_args.alloc_attrs, &in, in SameWorkerRecvDone()
298 const Rendezvous::Args& recv_args, in RecvAsync() argument
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | local_rendezvous.cc | 42 Item(Rendezvous::Args recv_args, Rendezvous::DoneCallback waiter, in Item()
44 : Item(recv_args, kRecv) { in Item()
168 const Rendezvous::Args& recv_args, in RecvAsync() argument
178 done(s, Rendezvous::Args(), recv_args, Tensor(), false); in RecvAsync()
186 CancellationManager* cm = recv_args.cancellation_manager; in RecvAsync()
257 Rendezvous::Args(), recv_args, Tensor(), /*is_dead=*/false); in RecvAsync()
270 recv_args, in RecvAsync()
273 const Rendezvous::Args& recv_args, const Tensor& v, bool dead) { in RecvAsync() argument
282 done(s, send_args, recv_args, v, dead); in RecvAsync()
286 queue->push_back(new Item(recv_args, std::move(done), token)); in RecvAsync()
[all …]
|
| D | rendezvous.cc | 119 Status RendezvousInterface::Recv(const ParsedKey& key, const Args& recv_args, in Recv() argument
123 RecvAsync(key, recv_args, in Recv()
125 const Args& recv_args, const Tensor& v, in Recv()
161 void RecvAsync(const ParsedKey& key, const Args& recv_args, in RecvAsync() argument
163 impl_.RecvAsync(key, recv_args, std::move(done)); in RecvAsync()
|
| D | rendezvous_test.cc | 237 const Rendezvous::Args& recv_args, const Tensor& v, in TEST_F()
245 const Rendezvous::Args& recv_args, const Tensor& v, in TEST_F() argument
253 const Rendezvous::Args& recv_args, const Tensor& v, in TEST_F() argument
261 const Rendezvous::Args& recv_args, const Tensor& v, in TEST_F() argument
425 const Rendezvous::Args& recv_args, const Tensor& val, bool is_dead) { in TEST_F() argument
|
| D | local_rendezvous.h | 49 const Rendezvous::Args& recv_args,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 64 const Rendezvous::Args& recv_args, Rendezvous::DoneCallback done) { in Init() argument
68 recv_args_ = recv_args; in Init()
131 const Rendezvous::Args& recv_args() const { return recv_args_; } in recv_args() function in tensorflow::__anona9090a8f0111::RpcRecvTensorCall
234 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync() argument
268 done(s, Args(), recv_args, Tensor{}, false); in RecvFromRemoteAsync()
272 call->Init(rwi, step_id_, parsed.FullKey(), recv_args.alloc_attrs, dst_device, in RecvFromRemoteAsync()
273 recv_args, std::move(done)); in RecvFromRemoteAsync()
276 RegisterCall(call, recv_args); in RecvFromRemoteAsync()
302 call->done()(s, Args(), call->recv_args(), call->tensor(), call->is_dead()); in RecvFromRemoteAsync()
|
| D | rpc_rendezvous_mgr_test.cc | 276 const Rendezvous::Args recv_args, const Tensor& val, in TEST_F()
|
| D | grpc_worker_service.cc | 515 const Rendezvous::Args& recv_args, const Tensor& val, in GrpcRecvTensorAsync()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 153 const Rendezvous::Args& recv_args, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sendrecv_ops.cc | 174 const Rendezvous::Args& recv_args, in make_recv_callback()
|