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)

SplitUntil32Bit.

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: