Home
last modified time | relevance | path

Searched refs:IntraProcessRendezvous (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_mgr.cc34 IntraProcessRendezvous::IntraProcessRendezvous(const DeviceMgr* device_mgr) in IntraProcessRendezvous() function in tensorflow::IntraProcessRendezvous
37 IntraProcessRendezvous::~IntraProcessRendezvous() { local_->Unref(); } in ~IntraProcessRendezvous()
39 Status IntraProcessRendezvous::Send(const ParsedKey& parsed, in Send()
52 Status IntraProcessRendezvous::ParseKey(const string& key, bool is_src, in ParseKey()
62 void IntraProcessRendezvous::SameWorkerRecvDone( in SameWorkerRecvDone()
116 void IntraProcessRendezvous::RecvAsync(const ParsedKey& parsed, in RecvAsync()
156 void IntraProcessRendezvous::StartAbort(const Status& s) { in StartAbort()
Drendezvous_mgr.h40 class IntraProcessRendezvous : public Rendezvous {
42 explicit IntraProcessRendezvous(const DeviceMgr* device_mgr);
67 ~IntraProcessRendezvous() override;
85 TF_DISALLOW_COPY_AND_ASSIGN(IntraProcessRendezvous);
Ddirect_session.h183 IntraProcessRendezvous* rendez = nullptr;
265 IntraProcessRendezvous* rendez);
Dprocess_function_library_runtime_test.cc95 rendezvous_ = new IntraProcessRendezvous(device_mgr_.get()); in Init()
217 IntraProcessRendezvous* rendezvous_;
Ddirect_session.cc482 run_state.rendez = new IntraProcessRendezvous(device_mgr_.get()); in RunInternal()
859 run_state->rendez = new IntraProcessRendezvous(device_mgr_.get()); in PRunSetup()
1041 IntraProcessRendezvous* rendez) { in SendPRunInputs()
1101 IntraProcessRendezvous* rendez = run_state->rendez; in RecvPRunOutputs()
Dfunction.cc1054 Rendezvous* rendezvous = new IntraProcessRendezvous(device_mgr_); in Run()
1130 Rendezvous* rendezvous = new IntraProcessRendezvous(device_mgr_); in Run()
Dfunction_test.cc1591 opts.rendezvous = new IntraProcessRendezvous(device_mgr_.get()); in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_data.cc42 new tensorflow::IntraProcessRendezvous(device_mgr.get()); in Prepare()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc228 Rendezvous* rendez = new IntraProcessRendezvous(lib->device_mgr()); in RunFunction()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc316 Rendezvous* rendezvous = new IntraProcessRendezvous(pflr_->device_mgr()); in Run()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc346 new tensorflow::IntraProcessRendezvous(device_mgr.get()); in TFE_NewContext()
359 new tensorflow::IntraProcessRendezvous(device_mgr); in TFE_NewContextFromSession()