Searched refs:CollectiveRemoteAccessLocal (Results 1 – 12 of 12) 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()
97 void CollectiveRemoteAccessLocal::PostToPeer( in PostToPeer()
109 void CollectiveRemoteAccessLocal::CheckPeerHealth(const string& peer_task, in CheckPeerHealth()
118 void CollectiveRemoteAccessLocal::MemCpyAsync( in MemCpyAsync()
|
| D | collective_rma_local.h | 26 class CollectiveRemoteAccessLocal : public CollectiveRemoteAccess {
28 CollectiveRemoteAccessLocal(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessLocal() function
36 ~CollectiveRemoteAccessLocal() override = default;
|
| D | collective_executor_mgr.cc | 66 CollectiveRemoteAccessLocal* rma = in Create()
67 new CollectiveRemoteAccessLocal(dev_mgr_, dev_resolver_.get(), step_id); in Create()
|
| D | collective_test_util.cc | 35 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in FailTestRMA()
65 CollectiveRemoteAccessLocal::RecvFromPeer( in RecvFromPeer()
80 CollectiveRemoteAccessLocal::PostToPeer( in PostToPeer()
|
| D | collective_test_util.h | 38 class FailTestRMA : public CollectiveRemoteAccessLocal {
|
| D | collective_rma_local_test.cc | 54 rma_ = std::make_unique<CollectiveRemoteAccessLocal>(device_mgr_.get(), in CollectiveRemoteAccessLocalTest()
65 std::unique_ptr<CollectiveRemoteAccessLocal> rma_;
|
| D | all_to_all.cc | 68 CollectiveRemoteAccessLocal::MemCpyAsync( in CheckCounterAndCallDone()
|
| D | ring_gatherer.cc | 106 CollectiveRemoteAccessLocal::MemCpyAsync( in Run()
|
| D | ring_reducer.cc | 97 CollectiveRemoteAccessLocal::MemCpyAsync( in Run()
|
| D | hierarchical_tree_broadcaster.cc | 378 CollectiveRemoteAccessLocal::MemCpyAsync( in RunTree()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.h | 27 class CollectiveRemoteAccessDistributed : public CollectiveRemoteAccessLocal {
33 : CollectiveRemoteAccessLocal(dev_mgr, dev_resolver, step_id), in CollectiveRemoteAccessDistributed()
|
| D | collective_rma_distributed.cc | 115 CollectiveRemoteAccessLocal::RecvFromPeer( in RecvFromPeer()
295 CollectiveRemoteAccessLocal::StartAbort(s); in StartAbort()
|