Home
last modified time | relevance | path

Searched defs:recv_args (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_mgr.cc64 const Rendezvous::Args& recv_args, const Tensor& in, Tensor* out, in SameWorkerRecvDone()
117 const Rendezvous::Args& recv_args, in RecvAsync()
130 bool is_dead) { in RecvAsync()
Drendezvous_util.cc105 const Tensor& v, const bool is_dead) { in RecvOutputsFromRendezvousAsync()
Dgraph_runner.cc67 void RecvAsync(const ParsedKey& parsed, const Args& recv_args, in RecvAsync()
/external/tensorflow/tensorflow/core/framework/
Drendezvous.cc117 Status Rendezvous::Recv(const ParsedKey& key, const Args& recv_args, in Recv()
124 const bool dead) { in Recv()
196 void RecvAsync(const ParsedKey& key, const Args& recv_args, in RecvAsync()
264 Args recv_args; member
Drendezvous_test.cc303 const Rendezvous::Args& recv_args, const Tensor& val, bool is_dead) { in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.cc77 const Rendezvous::Args& recv_args, const Tensor& v, bool dead) { in RecvLocalAsync()
94 const Tensor& v, const bool dead) { in RecvLocal()
231 const Rendezvous::Args& recv_args, const Tensor& in, Tensor* out, in SameWorkerRecvDone()
293 const Rendezvous::Args& recv_args, in RecvAsync()
313 const Rendezvous::Args& recv_args, const Tensor& in, bool is_dead) { in RecvAsync()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_rendezvous_mgr.cc43 const Rendezvous::Args& recv_args, int64 step_id, 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()
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_rendezvous_mgr.cc63 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
165 MPISendTensorCall* mpi_send_call) { in AddRequest()
201 const Rendezvous::Args& recv_args, const Tensor& val, bool is_dead) { in AddRequest()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr.cc63 const Rendezvous::Args& recv_args, Rendezvous::DoneCallback done) { 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()
Drpc_rendezvous_mgr_test.cc190 bool is_dead) { in TEST_F()
Dgrpc_worker_service.cc482 const bool is_dead) { in GrpcRecvTensorAsync()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma_rendezvous_mgr.cc48 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
Drdma.cc669 const Rendezvous::Args recv_args, in InsertTensorRequest()
1007 bool is_dead) { in Start()
1042 const Rendezvous::Args& recv_args, in RecvHandler()
1478 Device* dst_dev, const Rendezvous::Args recv_args, in RdmaTensorRequest()
1510 Rendezvous::Args recv_args = std::move(recv_args_); in Done() local
/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc154 bool is_dead) { in make_recv_callback()