Home
last modified time | relevance | path

Searched defs:rendezvous (Results 1 – 25 of 26) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dgraph_mgr.cc357 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); in SendInputs() local
376 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); in RecvOutputs() local
395 Rendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); in RecvOutputsAsync() local
462 RemoteRendezvous* rendezvous = worker_env_->rendezvous_mgr->Find(step_id); in ExecuteAsync() local
517 const string& handle, int64_t step_id, Item* item, Rendezvous* rendezvous, in StartParallelExecutors()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drendezvous_util.cc23 RendezvousInterface* rendezvous, DeviceContext* device_context, in SendTensorsToRendezvous()
57 RendezvousInterface* rendezvous, DeviceContext* device_context, in RecvOutputsFromRendezvousAsync()
121 Status RecvOutputsFromRendezvous(RendezvousInterface* rendezvous, in RecvOutputsFromRendezvous()
H A Dfunction.cc1038 RendezvousInterface* rendezvous = opts.rendezvous; in RunRemote() local
1137 auto* rendezvous = new RefCountedIntraProcessRendezvous(device_mgr_); in Run() local
1217 auto* rendezvous = new RefCountedIntraProcessRendezvous(device_mgr_); in Run() local
1311 std::unique_ptr<PrivateIntraProcessRendezvous> rendezvous; in RunSync() local
1330 std::unique_ptr<PrivateIntraProcessRendezvous> rendezvous; in RunSync() local
H A Dexecutor.h94 RendezvousInterface* rendezvous = nullptr; member
H A Dprocess_function_library_runtime.cc125 RendezvousInterface* rendezvous) { in SendTensors()
144 RendezvousInterface* rendezvous, std::vector<Tensor>* received_tensors, in ReceiveTensorsAsync()
1794 auto rendezvous = opts.rendezvous; in RunInternal() local
H A Ddirect_session.cc734 PrivateIntraProcessRendezvous rendezvous(device_mgr_.get()); in RunInternal() local
741 core::RefCountPtr<RefCountedIntraProcessRendezvous> rendezvous( in RunInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_compat_request_state.cc66 core::RefCountPtr<Rendezvous> rendezvous, OpKernelRunnerTable* runner_table, in KernelFallbackCompatRequestState()
H A Dkernel_fallback_compat_request_state.h129 RendezvousInterface* rendezvous() const { return rendezvous_.get(); } in rendezvous() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dmkl_eager_op_rewrite_test.cc41 tensorflow::Rendezvous* rendezvous = in CreateOp() local
H A Dcontext.cc150 DeviceMgr* device_mgr, bool device_mgr_owned, Rendezvous* rendezvous, in EagerContext()
1215 IntraProcessRendezvous* rendezvous = in GetGlobalRendezvousForFunctionLocalRendezvousStatus() local
1226 IntraProcessRendezvous* rendezvous = in UpdateGlobalRendezvousDeviceManager() local
H A Dkernel_and_device.h201 tensorflow::Rendezvous* rendezvous, bool log_memory, in KernelAndDeviceOp()
H A Dkernel_and_device.cc385 Rendezvous* rendezvous = rendezvous_creator_(opts->step_id); in PrepareForRun() local
/aosp_15_r20/external/autotest/client/common_lib/
H A Dbarrier.py494 def rendezvous(self, *hosts, **dargs): member in barrier
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Ddelegate_data.cc223 tensorflow::Rendezvous* rendezvous = in Prepare() local
/aosp_15_r20/external/pytorch/torch/distributed/
H A Drendezvous.py99 def rendezvous(url: str, rank: int = -1, world_size: int = -1, **kwargs): function
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Dremote_mgr_test.cc55 tensorflow::Rendezvous* rendezvous = in RemoteMgrTest() local
H A Deager_service_impl_test.cc1232 tensorflow::Rendezvous* rendezvous = in TEST_F() local
/aosp_15_r20/external/cronet/base/memory/
H A Dshared_memory_switch.cc234 auto* rendezvous = MachPortRendezvousClient::GetInstance(); in Deserialize() local
/aosp_15_r20/external/linux-kselftest/tools/testing/selftests/kvm/
H A Dmax_guest_memory_test.c37 static atomic_t rendezvous; variable
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DSymmetricMemory.cpp176 TORCH_API c10::intrusive_ptr<SymmetricMemory> rendezvous( in rendezvous() function
H A Dintra_node_comm.cpp273 bool IntraNodeComm::rendezvous() { in rendezvous() function in c10d::intra_node_comm::IntraNodeComm
H A DCUDASymmetricMemory.cu701 c10::intrusive_ptr<SymmetricMemory> CUDASymmetricMemoryAllocator::rendezvous( in rendezvous() function in c10d::symmetric_memory::CUDASymmetricMemoryAllocator
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcollective_ops_utils.h309 std::shared_ptr<Rendezvous<I, O>> rendezvous = rendezvous_getter(); in SubmitParticipant() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.h621 RendezvousInterface* rendezvous = nullptr; member
1059 RendezvousInterface* rendezvous() const { return params_->rendezvous; } in rendezvous() function
H A Dfunction.h845 RendezvousInterface* rendezvous = nullptr; member

12