Home
last modified time | relevance | path

Searched defs:send_args (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drendezvous_test.cc244 [&n1, &s1](const Status& s, const Rendezvous::Args& send_args, in TEST_F()
252 [&n2, &s2](const Status& s, const Rendezvous::Args& send_args, in TEST_F()
260 [&n3, &s3](const Status& s, const Rendezvous::Args& send_args, 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()
H A Drendezvous.cc127 const bool dead) { in Recv()
157 Status Send(const ParsedKey& key, const Args& send_args, const Tensor& val, in Send()
H A Dlocal_rendezvous.cc120 const Rendezvous::Args& send_args, in Send()
299 const Rendezvous::Args& recv_args, const Tensor& v, bool dead) { in RecvAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dbase_rendezvous_mgr.cc94 const Rendezvous::Args& send_args, in RecvLocal()
96 const Tensor& v, const bool dead) { in RecvLocal()
235 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& send_args, in SameWorkerRecvDone()
334 const Rendezvous::Args& recv_args, const Tensor& in, bool is_dead) { in RecvAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drendezvous_mgr.cc40 const Rendezvous::Args& send_args, in SameWorkerRecvDone()
146 bool is_dead) mutable { in IntraProcessRecvAsyncImpl()
H A Drendezvous_util.cc105 const Tensor& v, const bool is_dead) { in RecvOutputsFromRendezvousAsync()
H A Dgraph_runner.cc54 Status Send(const ParsedKey& parsed, const Args& send_args, const Tensor& val, in Send()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dhost_compute_ops.cc154 const Tensor& val, bool is_dead) { in ComputeAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsendrecv_ops.cc181 const Tensor& val, bool is_dead) { in make_recv_callback()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Drpc_rendezvous_mgr_test.cc281 bool is_dead) { in TEST_F()
H A Dgrpc_worker_service.cc521 const bool is_dead) { in GrpcRecvTensorAsync()