Searched refs:ProvideBuf (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous_test.cc | 53 br_->ProvideBuf( in TEST_F()
95 br_->ProvideBuf( in TEST_F()
111 br_->ProvideBuf("key0", fake_dev_ptr_, fake_dev_ctx_, &a_, aa_, in TEST_F()
117 br_->ProvideBuf("key0", fake_dev_ptr_, fake_dev_ctx_, &a_, aa_, in TEST_F()
154 br_->ProvideBuf("key1", fake_dev_ptr_, fake_dev_ctx_, &a_, aa_, in TEST_F()
183 br_->ProvideBuf("key1", fake_dev_ptr_, fake_dev_ctx_, &a_, aa_, in TEST_F()
|
| D | buf_rendezvous.h | 77 void ProvideBuf(const string& key, Device* dev, DeviceContext* dev_ctx,
|
| D | collective_rma_local.cc | 78 buf_rendezvous_.ProvideBuf(key, from_device, from_device_ctx, from_tensor, in PostToPeer()
|
| D | buf_rendezvous.cc | 65 void BufRendezvous::ProvideBuf(const string& key, Device* dev, in ProvideBuf() function in tensorflow::BufRendezvous
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 266 wi->buf_rendezvous()->ProvideBuf( in TEST_F()
315 wi->buf_rendezvous()->ProvideBuf( in TEST_F()
|