Searched refs:TensorShapeIter (Results 1 – 2 of 2) sorted by relevance
35 class TensorShapeIter; variable254 TensorShapeIter<Shape> begin() const;255 TensorShapeIter<Shape> end() const;332 class TensorShapeIter {334 TensorShapeIter(const Shape* shape, int d) : shape_(shape), d_(d) {} in TensorShapeIter() function335 bool operator==(const TensorShapeIter& rhs) {339 bool operator!=(const TensorShapeIter& rhs) {
500 TensorShapeIter<Shape> TensorShapeBase<Shape>::begin() const { in begin()501 return TensorShapeIter<Shape>(static_cast<const Shape*>(this), 0); in begin()505 TensorShapeIter<Shape> TensorShapeBase<Shape>::end() const { in end()507 return TensorShapeIter<Shape>(static_cast<const Shape*>(this), dims()); in end()