| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 38 const Rendezvous::Args& send_args, in SameWorkerRecvDone() argument
44 (send_args.alloc_attrs.on_host() || parsed.src.type == "CPU"); in SameWorkerRecvDone()
80 attr.set_gpu_compatible(send_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()
127 const Status& status, const Rendezvous::Args& send_args, in IntraProcessRecvAsyncImpl()
135 auto final_callback = [send_args, recv_args, out, is_dead, in IntraProcessRecvAsyncImpl()
137 done(s, send_args, recv_args, *out, is_dead); in IntraProcessRecvAsyncImpl()
142 SameWorkerRecvDone(device_mgr, parsed, send_args, recv_args, in, out, in IntraProcessRecvAsyncImpl()
|
| D | rendezvous_util.cc | 103 const Rendezvous::Args& send_args, in RecvOutputsFromRendezvousAsync()
|
| D | graph_runner.cc | 54 Status Send(const ParsedKey& parsed, const Args& send_args, const Tensor& val, in Send() argument
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 76 const Status& s, const Rendezvous::Args& send_args, in RecvLocalAsync()
80 done(s, send_args, recv_args, v, dead); in RecvLocalAsync()
92 const Rendezvous::Args& send_args, in RecvLocal() argument
222 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& send_args, in SameWorkerRecvDone() argument
228 (send_args.alloc_attrs.on_host() || parsed.src.type == "CPU"); in SameWorkerRecvDone()
265 attr.set_gpu_compatible(send_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()
319 const Status& status, const Rendezvous::Args& send_args, in RecvAsync()
324 StatusCallback final_callback = [done, send_args, recv_args, out, in RecvAsync()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | local_rendezvous.cc | 36 Item(Rendezvous::Args send_args, const Tensor& value, bool is_dead) in Item()
37 : Item(send_args, kSend) { in Item()
110 const Rendezvous::Args& send_args, in Send() argument
141 queue->push_back(new Item(send_args, val, is_dead)); in Send()
162 (*item->recv_state.waiter)(Status::OK(), send_args, item->args, val, is_dead); in Send()
272 const Status& s, const Rendezvous::Args& send_args, in RecvAsync()
282 done(s, send_args, recv_args, v, dead); in RecvAsync()
|
| D | rendezvous.cc | 124 [&ret, &n, val, is_dead](const Status& s, const Args& send_args, in Recv()
156 Status Send(const ParsedKey& key, const Args& send_args, const Tensor& val, in Send() argument
158 return impl_.Send(key, send_args, val, is_dead); in Send()
|
| D | rendezvous_test.cc | 236 [&n0, &s0](const Status& s, const Rendezvous::Args& send_args, in TEST_F()
244 [&n1, &s1](const Status& s, const Rendezvous::Args& send_args, in TEST_F() argument
252 [&n2, &s2](const Status& s, const Rendezvous::Args& send_args, in TEST_F() argument
260 [&n3, &s3](const Status& s, const Rendezvous::Args& send_args, in TEST_F() argument
424 [&n](const Status& s, const Rendezvous::Args& send_args, in TEST_F()
427 send_args.device_context) in TEST_F()
|
| D | local_rendezvous.h | 46 const Rendezvous::Args& send_args, const Tensor& val,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr_test.cc | 275 [&n](const Status& s, const Rendezvous::Args send_args, in TEST_F()
279 static_cast<DummyDeviceContext*>(send_args.device_context); in TEST_F()
|
| D | grpc_worker_service.cc | 514 const Status& status, const Rendezvous::Args& send_args, in GrpcRecvTensorAsync()
525 const bool on_host = send_args.alloc_attrs.on_host(); in GrpcRecvTensorAsync()
529 DeviceContext* send_dev_context = send_args.device_context; in GrpcRecvTensorAsync()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 152 const Rendezvous::Args& send_args, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sendrecv_ops.cc | 173 const Rendezvous::Args& send_args, in make_recv_callback()
|