Searched refs:CollectiveRemoteAccessDistributed (Results 1 – 5 of 5) sorted by relevance
65 class CollectiveRemoteAccessDistributed : public CollectiveRemoteAccessLocal { class67 CollectiveRemoteAccessDistributed(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessDistributed() function in tensorflow::__anon0f845a970111::CollectiveRemoteAccessDistributed76 ~CollectiveRemoteAccessDistributed() override {} in ~CollectiveRemoteAccessDistributed()151 CollectiveRemoteAccessDistributed* rma = in Create()152 new CollectiveRemoteAccessDistributed(dev_mgr_, dev_resolver_.get(), in Create()
24 class CollectiveRemoteAccessDistributed : public CollectiveRemoteAccessLocal {26 CollectiveRemoteAccessDistributed(const DeviceMgr* dev_mgr, in CollectiveRemoteAccessDistributed() function33 ~CollectiveRemoteAccessDistributed() override {} in ~CollectiveRemoteAccessDistributed()
49 CollectiveRemoteAccessDistributed* rma = in Create()50 new CollectiveRemoteAccessDistributed(dev_mgr_, dev_resolver_.get(), in Create()
75 void CollectiveRemoteAccessDistributed::RecvFromPeer( in RecvFromPeer()181 void CollectiveRemoteAccessDistributed::StartAbort(const Status& s) { in StartAbort()
198 rma_.reset(new CollectiveRemoteAccessDistributed( in SetUp()248 std::unique_ptr<CollectiveRemoteAccessDistributed> rma_;