Trait IterGetSet

pub trait IterGetSet {
    type Item;

    // Required methods
    fn set_end_index(&mut self, end_index: usize);
    fn set_intervals(&mut self, intervals: Arc<Vec<(usize, usize)>>);
    fn set_strides(&mut self, strides: Strides);
    fn set_shape(&mut self, shape: Shape);
    fn set_prg(&mut self, prg: Vec<i64>);
    fn intervals(&self) -> &Arc<Vec<(usize, usize)>>;
    fn strides(&self) -> &Strides;
    fn shape(&self) -> &Shape;
    fn layout(&self) -> &Layout;
    fn broadcast_set_strides(&mut self, shape: &Shape);
    fn outer_loop_size(&self) -> usize;
    fn inner_loop_size(&self) -> usize;
    fn next(&mut self);
    fn inner_loop_next(&mut self, index: usize) -> Self::Item;
}

A trait for getting and setting values from an iterator.

Required Associated Types

type Item

The type of the iterator’s elements.

Required Methods

fn set_end_index(&mut self, end_index: usize)

set the end index of the iterator, this is used when rayon perform data splitting

fn set_intervals(&mut self, intervals: Arc<Vec<(usize, usize)>>)

set the chunk intervals of the iterator, we chunk the outer loop

fn set_strides(&mut self, strides: Strides)

set the strides for the iterator, we call this method normally when we do broadcasting

fn set_shape(&mut self, shape: Shape)

set the shape for the iterator, we call this method normally when we do broadcasting

fn set_prg(&mut self, prg: Vec<i64>)

set the loop progress for the iterator

fn intervals(&self) -> &Arc<Vec<(usize, usize)>>

get the intervals of the iterator

fn strides(&self) -> &Strides

get the strides of the iterator

fn shape(&self) -> &Shape

get the shape of the iterator

fn layout(&self) -> &Layout

get the layout of the iterator

fn broadcast_set_strides(&mut self, shape: &Shape)

set the strides for all the iterators

fn outer_loop_size(&self) -> usize

get the outer loop size

fn inner_loop_size(&self) -> usize

get the inner loop size

fn next(&mut self)

update the loop progress

fn inner_loop_next(&mut self, index: usize) -> Self::Item

get the next element of the inner loop

Implementors

impl<'a, A, B> IterGetSet for ParStridedZip<'a, A, B>

where A: IterGetSet, B: IterGetSet,

type Item = (<A as IterGetSet>::Item, <B as IterGetSet>::Item)

impl<'a, A, B> IterGetSet for StridedZip<'a, A, B>

where A: IterGetSet, B: IterGetSet,

type Item = (<A as IterGetSet>::Item, <B as IterGetSet>::Item)

impl<'a, T> IterGetSet for ParStridedMut<'a, T>

where T: CommonBounds + 'a,

type Item = &'a mut T

impl<'a, T> IterGetSet for StridedMut<'a, T>

where T: CommonBounds + 'a,

type Item = &'a mut T

impl<'a, T: 'a + CommonBounds> IterGetSet for ParStridedMapMut<'a, T>

type Item = &'a mut T

impl<'a, T: 'a + CommonBounds> IterGetSet for StridedMapMut<'a, T>

where T::Vec: Send,

type Item = &'a mut T

impl<T: CommonBounds> IterGetSet for ParStrided<T>

type Item = T

impl<T: CommonBounds> IterGetSet for Strided<T>

type Item = T