Struct caffe2op_dataset::TreeWalker

pub struct TreeWalker<'a> { /* private fields */ }

| Simple wrapper class allowing an easy | traversal of the tensors representing | the hirerarchical structure. |

Implementations

impl<'a> TreeWalker<'a>

pub fn size(&self) -> TOffset

Returns the number of records in a dataset

pub fn input(&self, idx: i32) -> &TensorCPU

pub fn field(&self, idx: i32) -> &TreeIteratorFieldDesc

pub fn length_idx(&self, field_id: i32) -> i32

pub fn offset(&self, field_id: i32) -> TOffset

pub fn fields(&self) -> &Vec<TreeWalkerField<'_>>

| Notice that a reference is returned. | If advance() is called the fields will | be updated to represent the new state. |

pub fn new(inputs: &Vec<*const Blob>, cursor: &mut TreeCursor) -> Self

pub fn advance(&mut self)

pub fn field_dim(&self, field_id: i32) -> Vec<i64>

pub fn field_ptr(&self, field_id: i32) -> *mut c_void

pub fn gather_length_data(&mut self)

pub fn gather_size_limits(&mut self)