Searched refs:RingReducer (Results 1 – 4 of 4) sorted by relevance
44 RingReducer::~RingReducer() { group_size_tensor_ready_.WaitForNotification(); } in ~RingReducer()46 Status RingReducer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()53 void RingReducer::Run(StatusCallback done) { in Run()111 void RingReducer::ContinueAfterInputCopy() { in ContinueAfterInputCopy()145 void RingReducer::InitRingField(RingField* rf, int chunk_idx, int subdiv_idx, in InitRingField()155 bool RingReducer::RunAsyncParts() { in RunAsyncParts()326 REGISTER_COLLECTIVE(RingReduce, RingReducer);
31 class RingReducer : public RingAlg {33 RingReducer() : RingAlg(REDUCTION_COLLECTIVE, "Reduce") {} in RingReducer() function34 ~RingReducer() override;
392 RingReducer reducer; in RunSubdivPermsTest()512 RingReducer reducer; in DoReduce()
155 // This ring order is used by the RingReducer implementation of