Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Drendezvous.cc119 Status RendezvousInterface::Recv(const ParsedKey& key, const Args& recv_args, in Recv()
126 const bool dead) { in Recv()
161 void RecvAsync(const ParsedKey& key, const Args& recv_args, in RecvAsync()
Drendezvous_test.cc245 const Rendezvous::Args& recv_args, const Tensor& v, in TEST_F()
253 const Rendezvous::Args& recv_args, const Tensor& v, in TEST_F()
261 const Rendezvous::Args& recv_args, const Tensor& v, in TEST_F()
262 const bool dead) { in TEST_F()
425 const Rendezvous::Args& recv_args, const Tensor& val, bool is_dead) { in TEST_F()
467 const Tensor& tensor, bool is_dead) { in BM_RecvSend()
Dlocal_rendezvous.cc157 const Rendezvous::Args& recv_args, in RecvAsync()
244 const Rendezvous::Args& recv_args, const Tensor& v, bool dead) { in RecvAsync()
/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_mgr.cc39 const Rendezvous::Args& recv_args, const Tensor& in, in SameWorkerRecvDone()
111 const Rendezvous::Args& recv_args, in IntraProcessRecvAsyncImpl()
122 bool is_dead) mutable { in IntraProcessRecvAsyncImpl()
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/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::__anondc8fdab30111::RpcRecvTensorCall
212 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
Drpc_rendezvous_mgr_test.cc245 bool is_dead) { in TEST_F()
Dgrpc_worker_service.cc516 const bool is_dead) { in GrpcRecvTensorAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.cc92 const Rendezvous::Args& recv_args, in RecvLocal()
93 const Tensor& v, const bool dead) { in RecvLocal()
236 const Rendezvous::Args& recv_args, const Tensor& in, Tensor* out, in SameWorkerRecvDone()
313 const Rendezvous::Args& recv_args, in RecvAsync()
336 const Rendezvous::Args& recv_args, const Tensor& in, bool is_dead) { in RecvAsync()
/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc148 const Tensor& val, bool is_dead) { in make_recv_callback()
/external/tensorflow/tensorflow/c/experimental/
Drendezvous_internal.h64 const TF_RendezvousArgs* recv_args; member