Searched refs:ConsumeBuf (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous_test.cc | 61 br_->ConsumeBuf("key0", [this, &cons_status, &cons_callback_called]( in TEST_F()
84 br_->ConsumeBuf("key0", [this, &cons_status, &cons_callback_called]( in TEST_F()
132 br_->ConsumeBuf("key0", in TEST_F()
149 br_->ConsumeBuf("key0", [&cons_note, &cons_status](const Status& s, in TEST_F()
178 br_->ConsumeBuf("key0", [&cons_note, &cons_status](const Status& s, in TEST_F()
|
| D | buf_rendezvous.h | 83 void ConsumeBuf(const string& key, const ConsumerCallback& done);
|
| D | collective_rma_local.cc | 40 buf_rendezvous_.ConsumeBuf( in RecvFromPeer()
|
| D | buf_rendezvous.cc | 111 void BufRendezvous::ConsumeBuf(const string& key, in ConsumeBuf() function in tensorflow::BufRendezvous
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_worker.cc | 162 rma->buf_rendezvous()->ConsumeBuf( in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 98 buf_rendezvous_.ConsumeBuf( in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 567 rma->buf_rendezvous()->ConsumeBuf( in RecvBufAsync()
|