Searched refs:GroupIterable (Results 1 – 6 of 6) sorted by relevance
22 void GroupIterable::IteratorStep::UpdateEndOfGroup() { in UpdateEndOfGroup()31 bool GroupIterable::IteratorStep::operator!=(const IteratorStep& rhs) const { in operator !=()36 bool GroupIterable::IteratorStep::operator==(const IteratorStep& rhs) const { in operator ==()41 GroupIterable::IteratorStep& GroupIterable::IteratorStep::48 GroupIterable::IteratorStep GroupIterable::IteratorStep::operator++( in operator ++()
29 class GroupIterable; // Predeclare GroupIterable for Group. variable36 Group(GroupIterable* iter, int64 loc, int64 next_loc) in Group()45 GroupIterable* iter_;76 class GroupIterable {80 GroupIterable(Tensor ix, Tensor vals, int dims, const VarDimArray& group_dims) in GroupIterable() function110 IteratorStep(GroupIterable* iter, int64 loc) in IteratorStep()124 GroupIterable* iter_;
197 GroupIterable group(const VarDimArray& group_ix) const { in group()205 return GroupIterable(ix_, vals_, dims_, group_ix); in group()
203 sparse::GroupIterable group_iterable_ GUARDED_BY(mu_);204 sparse::GroupIterable::IteratorStep iter_ GUARDED_BY(mu_);
227 sparse::GroupIterable minibatch = input_st.group({0}); in Compute()
281 sparse::GroupIterable minibatch = input_st.group({0}); in Compute()