Caffe2 - C++ API
A deep learning, cross platform ML framework
Data Structures | Public Member Functions
caffe2::dataset_ops::TreeIterator Class Reference

Provides functionality to iterate across a list of tensors where some of those tensors represent lengths in a hierarchical structure. More...

#include <dataset_ops.h>

Data Structures

struct  FieldDesc
 

Public Member Functions

 TreeIterator (const std::vector< std::string > &fields)
 
void advance (const std::vector< const TLength * > &lengths, std::vector< TOffset > &offsets, std::vector< TOffset > &sizes, std::vector< TOffset > &limits, TOffset num)
 
int numLengthFields () const
 
int numOffsetFields () const
 
const FieldDesclengthFieldFor (const FieldDesc &desc)
 
const FieldDesclengthField (int lengthFieldId)
 
int offsetFieldIdFor (const FieldDesc &fieldDesc)
 
const std::vector< FieldDesc > & fields ()
 
const std::vector< int > & lengthFieldIds () const
 

Detailed Description

Provides functionality to iterate across a list of tensors where some of those tensors represent lengths in a hierarchical structure.

Definition at line 24 of file dataset_ops.h.


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