Searched refs:reducers_ (Results 1 – 3 of 3) sorted by relevance
32 reducers_(zone), in GraphReducer()44 reducers_.push_back(reducer); in AddReducer()67 for (Reducer* const reducer : reducers_) reducer->Finalize(); in ReduceNode()82 auto skip = reducers_.end(); in Reduce()83 for (auto i = reducers_.begin(); i != reducers_.end();) { in Reduce()97 i = reducers_.begin(); in Reduce()111 if (skip == reducers_.end()) { in Reduce()
181 ZoneVector<Reducer*> reducers_; in NON_EXPORTED_BASE() local
2562 InlinedVector<HloComputation*, 1>* GetMutableReducers() { return &reducers_; } in GetMutableReducers()2564 return reducers_; in GetReducers()2566 int GetNumberOfReduces() const { return reducers_.size(); } in GetNumberOfReduces()2609 InlinedVector<HloComputation*, 1> reducers_; member in xla::gpu::ReductionCodegenInfo