Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
at::SplitUntil32Bit::iterator Struct Reference

Public Member Functions

 iterator (const TensorIterator &iter)
 SplitUntil32Bit. More...
 iterator (iterator &&)=default
TensorIteratoroperator* () const
iteratoroperator++ ()
bool operator== (const iterator &other) const
bool operator!= (const iterator &other) const

Data Fields

std::vector< std::unique_ptr< TensorIterator > > vec
 stack of TensorIterators to be split

Detailed Description

Definition at line 290 of file TensorIterator.h.

Constructor & Destructor Documentation

at::SplitUntil32Bit::iterator::iterator ( const TensorIterator iter)


Recursively splits an iterator into sub-iterators that can use 32-bit indexing.

Definition at line 654 of file TensorIterator.cpp.

The documentation for this struct was generated from the following files: