pub struct IReduceLayer { /* private fields */ }Expand description
! ! \class IReduceLayer ! ! \brief Layer that represents a reduction across a non-bool tensor. ! ! \warning Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI. !
Implementations§
Source§impl IReduceLayer
impl IReduceLayer
Sourcepub fn setOperation(self: Pin<&mut IReduceLayer>, op: ReduceOperation)
pub fn setOperation(self: Pin<&mut IReduceLayer>, op: ReduceOperation)
! ! \brief Set the reduce operation for the layer. ! ! \see getOperation(), ReduceOperation !
Sourcepub fn getOperation(self: &IReduceLayer) -> ReduceOperation
pub fn getOperation(self: &IReduceLayer) -> ReduceOperation
! ! \brief Get the reduce operation for the layer. ! ! \see setOperation(), ReduceOperation !
Sourcepub fn setReduceAxes(self: Pin<&mut IReduceLayer>, reduceAxes: u32)
pub fn setReduceAxes(self: Pin<&mut IReduceLayer>, reduceAxes: u32)
! ! \brief Set the axes over which to reduce. ! ! \see getReduceAxes !
Sourcepub fn getReduceAxes(self: &IReduceLayer) -> u32
pub fn getReduceAxes(self: &IReduceLayer) -> u32
! ! \brief Get the axes over which to reduce for the layer. ! ! \see setReduceAxes !
Sourcepub fn setKeepDimensions(self: Pin<&mut IReduceLayer>, keepDimensions: bool)
pub fn setKeepDimensions(self: Pin<&mut IReduceLayer>, keepDimensions: bool)
! ! \brief Set the boolean that specifies whether or not to keep the reduced dimensions for the layer. ! ! \see getKeepDimensions !
Sourcepub fn getKeepDimensions(self: &IReduceLayer) -> bool
pub fn getKeepDimensions(self: &IReduceLayer) -> bool
! ! \brief Get the boolean that specifies whether or not to keep the reduced dimensions for the layer. ! ! \see setKeepDimensions !