Struct ITopKLayer 

pub struct ITopKLayer { /* private fields */ }

ITopKLayer

Layer that represents a TopK reduction.

This layer can accept both static and dynamic k. Static k can be set through the addTopK() API function, or accessed using the getK() and setK() functions after layer creation. For dynamic k, use the setInput() method to pass in k as a tensor with index 1, which overrides the static k value in calculations.

Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Implementations

impl ITopKLayer

pub fn setOperation(self: Pin<&mut ITopKLayer>, op: TopKOperation)

Set the operation for the layer.

See [getOperation()], TopKOperation

pub fn getOperation(self: &ITopKLayer) -> TopKOperation

Get the operation for the layer.

See [setOperation()], TopKOperation

pub fn setK(self: Pin<&mut ITopKLayer>, k: i32)

Set the static k value for the layer.

Currently only values up to 3840 are supported.

If a second input to this layer has been set, it will be reset to null by this method.

See [getK()]

pub fn getK(self: &ITopKLayer) -> i32

Get the k value for the layer.

This function will return the static k value passed into addTopK(), or the value passed into setK().

If a second layer input is present and non-null, this function returns -1.

See [setK()]

pub fn setReduceAxes(self: Pin<&mut ITopKLayer>, reduceAxes: u32)

Set which axes to reduce for the layer.

See [getReduceAxes()]

pub fn getReduceAxes(self: &ITopKLayer) -> u32

Get the axes to reduce for the layer.

See [setReduceAxes()]

pub fn setIndicesType(self: Pin<&mut ITopKLayer>, type_: DataType) -> bool

Set the indices type for the layer.

• type The DataType of the indices tensor.

true if set successfully, false otherwise.

Set the indices (the second output) type of the TopK layer. Valid values are DataType::kINT32 and DataType::kINT64, otherwise an error occurs and the type is not updated.

pub fn getIndicesType(self: &ITopKLayer) -> DataType

Return the TopK layer indices type.

indices type set during layer creation or by setIndicesType(). The return value is the indices type of the TopK layer. The default value is DataType::kINT32.

Trait Implementations

impl AsLayer for ITopKLayer

fn as_layer(&self) -> &ILayer

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

impl AsLayerTyped for ITopKLayer

const TYPE: LayerType = LayerType::kTOPK

impl AsRef<ILayer> for ITopKLayer

fn as_ref(self: &ITopKLayer) -> &ILayer

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

impl ExternType for ITopKLayer

type Id = (n, v, i, n, f, e, r, _1, (), I, T, o, p, K, 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 ITopKLayer

unsafe fn allocate_uninitialized_cpp_storage() -> *mut ITopKLayer

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 ITopKLayer)

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