Searched refs:TensorShapeIter (Results 1 – 2 of 2) sorted by relevance
37 class TensorShapeIter; variable315 TensorShapeIter<Shape> begin() const;316 TensorShapeIter<Shape> end() const;435 class TensorShapeIter {437 TensorShapeIter(const Shape* shape, int d) : shape_(shape), d_(d) {} in TensorShapeIter() function438 bool operator==(const TensorShapeIter& rhs) {442 bool operator!=(const TensorShapeIter& rhs) {
715 TensorShapeIter<Shape> TensorShapeBase<Shape>::begin() const { in begin()716 return TensorShapeIter<Shape>(static_cast<const Shape*>(this), 0); in begin()720 TensorShapeIter<Shape> TensorShapeBase<Shape>::end() const { in end()722 return TensorShapeIter<Shape>(static_cast<const Shape*>(this), max_dim); in end()