Struct IIteratorLayer 

pub struct IIteratorLayer { /* private fields */ }

IIteratorLayer

A layer to do iterations.

The iterator layer iterates over a tensor along the given axis and in the given direction. It enables each loop iteration to inspect a different slice of the tensor.

See ILoop::addIterator()

Implementations

impl IIteratorLayer

pub fn setAxis(self: Pin<&mut IIteratorLayer>, axis: i32)

Set axis to iterate over.

pub fn getAxis(self: &IIteratorLayer) -> i32

Get axis being iterated over.

pub fn setReverse(self: Pin<&mut IIteratorLayer>, reverse: bool)

Set iteration order to be reverse.

For reverse=false, the layer is equivalent to addGather(tensor, I, 0) where I is a scalar tensor containing the loop iteration number. For reverse=true, the layer is equivalent to addGather(tensor, M-1-I, 0) where M is the trip count computed from TripLimits of kind kCOUNT. The default is reverse=false.

pub fn getReverse(self: &IIteratorLayer) -> bool

Check if the iteration order is reverse.

True if and only if reversing input.

Trait Implementations

impl AsLayer for IIteratorLayer

fn as_layer(&self) -> &ILayer

fn as_layer_pin_mut(&mut self) -> Pin<&mut ILayer>

impl AsLayerTyped for IIteratorLayer

const TYPE: LayerType = LayerType::kITERATOR

impl AsRef<ILoopBoundaryLayer> for IIteratorLayer

fn as_ref(self: &IIteratorLayer) -> &ILoopBoundaryLayer

Converts this type into a shared reference of the (usually inferred) input type.

impl ExternType for IIteratorLayer

type Id = (n, v, i, n, f, e, r, _1, (), I, I, t, e, r, a, t, o, r, L, a, y, e, r)

A type-level representation of the type’s C++ namespace and type name.

type Kind = Opaque

Either cxx::kind::Opaque or cxx::kind::Trivial.

impl MakeCppStorage for IIteratorLayer

unsafe fn allocate_uninitialized_cpp_storage() -> *mut IIteratorLayer

Allocates heap space for this type in C++ and return a pointer to that space, but do not initialize that space (i.e. do not yet call a constructor).

unsafe fn free_uninitialized_cpp_storage(arg0: *mut IIteratorLayer)

Frees a C++ allocation which has not yet had a constructor called.