Searched refs:RingAlg (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.cc | 76 void RingAlg::PCQueue::Enqueue(RingField* rf) { in Enqueue()
84 RingAlg::RingField* RingAlg::PCQueue::Dequeue() { in Dequeue()
98 RingAlg::RingAlg(CollectiveType type, const string& name) in RingAlg() function in tensorflow::RingAlg
164 Status RingAlg::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
243 Status RingAlg::InitializeCollectiveContext(CollectiveContext* col_ctx) { in InitializeCollectiveContext()
252 string RingAlg::TensorDebugString(const Tensor& tensor) { in TensorDebugString()
271 void RingAlg::StartAbort(const Status& s) { in StartAbort()
293 void RingAlg::Finish(bool ok) { in Finish()
309 void RingAlg::InitRingField(RingField* rf, int chunk_idx, int subdiv_idx, in InitRingField()
352 void RingAlg::AdvanceToSecondPass(RingField* rf) { in AdvanceToSecondPass()
[all …]
|
| D | ring_alg.h | 31 class RingAlg : public CollectiveImplementationInterface {
33 explicit RingAlg(CollectiveType type, const string& name);
34 ~RingAlg() override {} in ~RingAlg()
|
| D | ring_gatherer.h | 31 class RingGatherer : public RingAlg {
33 RingGatherer() : RingAlg(GATHER_COLLECTIVE, "Gather") {} in RingGatherer()
|
| D | ring_reducer.h | 31 class RingReducer : public RingAlg {
33 RingReducer() : RingAlg(REDUCTION_COLLECTIVE, "Reduce") {} in RingReducer()
|
| D | ring_reducer.cc | 50 return RingAlg::InitializeCollectiveParams(col_params); in InitializeCollectiveParams()
147 RingAlg::InitRingField(rf, chunk_idx, subdiv_idx, field_idx); in InitRingField()
|
| D | ring_gatherer.cc | 58 return RingAlg::InitializeCollectiveParams(col_params); in InitializeCollectiveParams()
|