org.bytedeco.tensorrt.nvinfer

Class ITopKLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorrt.nvinfer.INoCopy

org.bytedeco.tensorrt.nvinfer.ILayer

org.bytedeco.tensorrt.nvinfer.ITopKLayer

All Implemented Interfaces:

AutoCloseable

@Namespace(value="nvinfer1")
 @NoOffset
 @Properties(inherit=nvinfer.class)
public class ITopKLayer
extends ILayer

\class ITopKLayer \brief 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. \warning Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
ITopKLayer(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
int	getK() \brief Get the k value for the layer.
nvinfer.TopKOperation	getOperation() \brief Get the operation for the layer.
int	getReduceAxes() \brief Get the axes to reduce for the layer.
void	setK(int k) \brief Set the static k value for the layer.
void	setOperation(int op)
void	setOperation(nvinfer.TopKOperation op) \brief Set the operation for the layer.
void	setReduceAxes(int reduceAxes) \brief Set which axes to reduce for the layer.

Methods inherited from class org.bytedeco.tensorrt.nvinfer.ILayer

getInput, getMetadata, getName, getNbInputs, getNbOutputs, getOutput, getOutputType, getPrecision, getType, outputTypeIsSet, precisionIsSet, resetOutputType, resetPrecision, setInput, setMetadata, setMetadata, setName, setName, setOutputType, setOutputType, setPrecision, setPrecision

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

ITopKLayer

public ITopKLayer(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

setOperation

@NoException(value=true)
public void setOperation(nvinfer.TopKOperation op)

\brief Set the operation for the layer.

See Also:

getOperation(), TopKOperation

setOperation

@NoException(value=true)
public void setOperation(@Cast(value="nvinfer1::TopKOperation")
                                                  int op)

getOperation

@NoException(value=true)
public nvinfer.TopKOperation getOperation()

\brief Get the operation for the layer.

See Also:

setOperation(), TopKOperation

setK

@NoException(value=true)
public void setK(int k)

\brief 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 Also:

getK()

getK

@NoException(value=true)
public int getK()

\brief 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 Also:

setK()

setReduceAxes

@NoException(value=true)
public void setReduceAxes(@Cast(value="uint32_t")
                                                   int reduceAxes)

\brief Set which axes to reduce for the layer.

See Also:

getReduceAxes()

getReduceAxes

@Cast(value="uint32_t")
 @NoException(value=true)
public int getReduceAxes()

\brief Get the axes to reduce for the layer.

See Also:

setReduceAxes()