Home
last modified time | relevance | path

Searched refs:BufRendezvous (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dbuf_rendezvous.cc24 BufRendezvous::~BufRendezvous() { in ~BufRendezvous()
32 void BufRendezvous::StartAbort(const Status& s) { in StartAbort()
43 void BufRendezvous::PurgeTable(const Status& s, HookTable* table) { in PurgeTable()
57 string BufRendezvous::Hook::DebugString() const { in DebugString()
65 void BufRendezvous::ProvideBuf(const string& key, Device* dev, in ProvideBuf()
111 void BufRendezvous::ConsumeBuf(const string& key, in ConsumeBuf()
151 void BufRendezvous::DoneWithHook(Hook* h) { in DoneWithHook()
156 void BufRendezvous::LogContents() { in LogContents()
Dbuf_rendezvous_test.cc32 br_.reset(new BufRendezvous(123)); in BufRendezvousTest()
44 std::unique_ptr<BufRendezvous> br_;
62 const Status& s, BufRendezvous::Hook* h) { in TEST_F()
85 const Status& s, BufRendezvous::Hook* h) { in TEST_F()
133 [&cons_status](const Status& s, BufRendezvous::Hook* h) { in TEST_F()
150 BufRendezvous::Hook* h) { in TEST_F()
179 BufRendezvous::Hook* h) { in TEST_F()
Dbuf_rendezvous.h39 class BufRendezvous {
41 explicit BufRendezvous(uint64 step_id) : step_id_(step_id) {} in BufRendezvous() function
43 ~BufRendezvous();
Dcollective_rma_local.h71 BufRendezvous* buf_rendezvous() override { return &buf_rendezvous_; } in buf_rendezvous()
87 BufRendezvous buf_rendezvous_;
Dcollective_rma_local.cc43 done](const Status& s, BufRendezvous::Hook* hook) { in RecvFromPeer()
65 BufRendezvous::DoneWithHook(hook); in RecvFromPeer()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed_test.cc72 BufRendezvous* buf_rendezvous() { return &buf_rendezvous_; } in buf_rendezvous()
101 BufRendezvous::Hook* h) { in RecvBufAsync()
114 if (h) BufRendezvous::DoneWithHook(h); in RecvBufAsync()
122 BufRendezvous buf_rendezvous_;
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_worker.cc165 BufRendezvous::Hook* hook) { in RecvBufAsync()
181 BufRendezvous::DoneWithHook(hook); in RecvBufAsync()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h28 class BufRendezvous; variable
328 virtual BufRendezvous* buf_rendezvous() = 0;
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc570 BufRendezvous::Hook* hook) { in RecvBufAsync()
606 BufRendezvous::DoneWithHook(hook); in RecvBufAsync()
619 BufRendezvous::DoneWithHook(hook); in RecvBufAsync()
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto130 // concurrently so that BufRendezvous entries will make the correct
446 // Used at server side to find the correct BufRendezvous.
449 // Arbitrary string identifying a BufRendezvous entry.
452 // Size of value expected, must agree with BufRendezvous entry.