Searched refs:BufRendezvous (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous.cc | 29 void DeregisterCancellation(BufRendezvous::Hook* h) { in DeregisterCancellation()
38 BufRendezvous::~BufRendezvous() { in ~BufRendezvous()
46 void BufRendezvous::StartAbort(const Status& s) { in StartAbort()
61 void BufRendezvous::PurgeTable(const Status& s, HookTable* table) { in PurgeTable()
78 string BufRendezvous::Hook::DebugString() const { in DebugString()
86 void BufRendezvous::ProvideBuf(const string& key, Device* dev, in ProvideBuf()
149 void BufRendezvous::ConsumeBuf(const string& key, const string& device_name, in ConsumeBuf()
220 void BufRendezvous::CancelHook(const string& key) { in CancelHook()
243 void BufRendezvous::DoneWithHook(Hook* h) { in DoneWithHook()
248 void BufRendezvous::LogContents() { in LogContents()
|
| D | buf_rendezvous_test.cc | 53 br_ = absl::make_unique<BufRendezvous>(123, dev_mgr_.get()); in InitializeDevice()
70 std::unique_ptr<BufRendezvous> br_;
100 BufRendezvous::Hook* h) { in TEST_F()
126 BufRendezvous::Hook* h) { in TEST_F()
180 [&cons_status](const Status& s, BufRendezvous::Hook* h) { in TEST_F()
199 [&cons_note, &cons_status](const Status& s, BufRendezvous::Hook* h) { in TEST_F()
232 [&cons_note, &cons_status](const Status& s, BufRendezvous::Hook* h) { in TEST_F()
263 [¬e, &cons_status](const Status& s, BufRendezvous::Hook* h) { in TEST_F()
315 [&status, ¬e](const Status& s, BufRendezvous::Hook* h) { in TEST_F()
335 [&status, ¬e](const Status& s, BufRendezvous::Hook* h) { in TEST_F()
[all …]
|
| D | buf_rendezvous.h | 42 class BufRendezvous {
44 explicit BufRendezvous(uint64 step_id, const DeviceMgr* dev_mgr) in BufRendezvous() function
47 ~BufRendezvous();
|
| D | collective_rma_local.h | 61 BufRendezvous* buf_rendezvous() override { return &buf_rendezvous_; } in buf_rendezvous()
77 BufRendezvous buf_rendezvous_;
|
| D | collective_rma_local.cc | 51 BufRendezvous::Hook* hook) { in RecvFromPeer()
83 BufRendezvous::DoneWithHook(hook); in RecvFromPeer()
88 BufRendezvous::DoneWithHook(hook); in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 75 BufRendezvous* buf_rendezvous() { return &buf_rendezvous_; } in buf_rendezvous()
115 [opts, response, done](const Status& s, BufRendezvous::Hook* h) { in RecvBufAsync()
128 if (h) BufRendezvous::DoneWithHook(h); in RecvBufAsync()
137 BufRendezvous buf_rendezvous_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 29 class BufRendezvous; variable
289 virtual BufRendezvous* buf_rendezvous() = 0;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 639 BufRendezvous::Hook* hook) { in RecvBufAsync()
682 BufRendezvous::DoneWithHook(hook); in RecvBufAsync()
694 BufRendezvous::DoneWithHook(hook); in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 143 // concurrently so that BufRendezvous entries will make the correct
484 // Used at server side to find the correct BufRendezvous.
487 // Arbitrary string identifying a BufRendezvous entry.
490 // Size of value expected, must agree with BufRendezvous entry.
|