Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Drpc_rendezvous_mgr_test.cc149 RemoteRendezvous* rendez = rmgr_.Find(step_id); in TEST_F() local
170 RemoteRendezvous* rendez = rmgr_.Find(step_id); in TEST_F() local
184 RemoteRendezvous* rendez = rmgr_.Find(step_id); in TEST_F() local
204 RemoteRendezvous* rendez = rmgr_.Find(step_id); in TEST_F() local
228 RemoteRendezvous* rendez = rmgr_.Find(step_id); in TEST_F() local
268 RemoteRendezvous* rendez = rmgr_.Find(step_id); in TEST_F() local
300 RemoteRendezvous* rendez = rmgr_.Find(step_id); in TEST_F() local
319 RemoteRendezvous* rendez = rmgr_.Find(step_id); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dbase_rendezvous_mgr.cc43 static void StartAbortRendevous(Rendezvous* rendez, const Status& s) { in StartAbortRendevous()
53 auto rendez = p.second; in ~BaseRendezvousMgr() local
76 auto rendez = FindOrCreate(step_id); in RecvLocalAsync() local
107 Rendezvous* rendez = nullptr; in Cleanup() local
H A Dworker.cc112 RemoteRendezvous* rendez = env_->rendezvous_mgr->Find(step_id); in AbortStep() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dexecutor_test.cc95 Status Run(Rendezvous* rendez) { in Run()
287 Rendezvous* rendez = NewLocalRendezvous(); in TEST_F() local
402 Rendezvous* rendez = NewLocalRendezvous(); in TEST_F() local
421 Rendezvous* rendez = NewLocalRendezvous(); in TEST_F() local
H A Dgraph_runner.cc129 SimpleRendezvous rendez; in Run() local
H A Dkernel_benchmark_testlib.cc50 Rendezvous* rendez, const char* executor_type, in Benchmark()
H A Ddirect_session.h218 core::RefCountPtr<IntraProcessRendezvous> rendez = nullptr; member
H A Dsingle_threaded_executor_test.cc112 Status Run(Rendezvous* rendez) { in Run()
H A Ddirect_session.cc1165 IntraProcessRendezvous* rendez) { in SendPRunInputs()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drendezvous_test.cc438 Rendezvous* rendez = NewLocalRendezvous(); in BM_SendRecv() local
455 Rendezvous* rendez = NewLocalRendezvous(); in BM_RecvSend() local
492 Rendezvous* rendez = NewLocalRendezvous(); in BM_PingPong() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc1425 PrivateIntraProcessRendezvous rendez(device_mgr_); in InitializeVarOnTPU() local
1592 PrivateIntraProcessRendezvous rendez(device_mgr_); in InitializeShardedVarOnTPU() local
2695 auto* rendez = new RefCountedIntraProcessRendezvous(device_mgr_); in ExecuteFunctions() local