Searched refs:CollectiveRemoteAccessLocal (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 22 void CollectiveRemoteAccessLocal::StartAbort(const Status& s) { in StartAbort()
26 void CollectiveRemoteAccessLocal::RecvFromPeer( in RecvFromPeer()
71 void CollectiveRemoteAccessLocal::PostToPeer( in PostToPeer()
83 void CollectiveRemoteAccessLocal::MemCpyAsync( in MemCpyAsync()
|
| D | collective_rma_local.h | 25 class CollectiveRemoteAccessLocal : public PerStepCollectiveRemoteAccess {
27 CollectiveRemoteAccessLocal(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessLocal() function
35 virtual ~CollectiveRemoteAccessLocal() {} in ~CollectiveRemoteAccessLocal()
|
| D | collective_executor_mgr.cc | 59 CollectiveRemoteAccessLocal* rma = in Create()
60 new CollectiveRemoteAccessLocal(dev_mgr_, dev_resolver_.get(), step_id); in Create()
|
| D | collective_rma_local_test.cc | 51 rma_.reset(new CollectiveRemoteAccessLocal(device_mgr_.get(), drl_.get(), in CollectiveRemoteAccessLocalTest()
58 std::unique_ptr<CollectiveRemoteAccessLocal> rma_;
|
| D | ring_gatherer_test.cc | 44 class FailTestRMA : public CollectiveRemoteAccessLocal {
48 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
74 CollectiveRemoteAccessLocal::RecvFromPeer( in RecvFromPeer()
88 CollectiveRemoteAccessLocal::PostToPeer( in PostToPeer()
518 CollectiveRemoteAccessLocal* rma_;
|
| D | ring_reducer_test.cc | 44 class FailTestRMA : public CollectiveRemoteAccessLocal {
48 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
74 CollectiveRemoteAccessLocal::RecvFromPeer( in RecvFromPeer()
88 CollectiveRemoteAccessLocal::PostToPeer( in PostToPeer()
545 CollectiveRemoteAccessLocal* rma_;
|
| D | hierarchical_tree_broadcaster_test.cc | 136 class FailTestRMA : public CollectiveRemoteAccessLocal {
140 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
169 CollectiveRemoteAccessLocal::RecvFromPeer( in RecvFromPeer()
182 CollectiveRemoteAccessLocal::PostToPeer( in PostToPeer()
714 CollectiveRemoteAccessLocal* rma_;
|
| D | ring_gatherer.cc | 102 CollectiveRemoteAccessLocal::MemCpyAsync( in Run()
|
| D | ring_reducer.cc | 90 CollectiveRemoteAccessLocal::MemCpyAsync( in Run()
|
| D | hierarchical_tree_broadcaster.cc | 370 CollectiveRemoteAccessLocal::MemCpyAsync( in RunTree()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 65 class CollectiveRemoteAccessDistributed : public CollectiveRemoteAccessLocal {
72 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in CollectiveRemoteAccessDistributed()
86 CollectiveRemoteAccessLocal::RecvFromPeer( in RecvFromPeer()
138 CollectiveRemoteAccessLocal::StartAbort(s); in StartAbort()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.h | 24 class CollectiveRemoteAccessDistributed : public CollectiveRemoteAccessLocal {
30 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in CollectiveRemoteAccessDistributed()
|
| D | collective_rma_distributed.cc | 82 CollectiveRemoteAccessLocal::RecvFromPeer( in RecvFromPeer()
182 CollectiveRemoteAccessLocal::StartAbort(s); in StartAbort()
|