@Namespace(value="nvinfer1") @NoOffset @Properties(inherit=nvinfer.class) public class ITopKLayer extends ILayer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
ITopKLayer(Pointer p)
Pointer cast constructor.
|
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.
|
getInput, getMetadata, getName, getNbInputs, getNbOutputs, getOutput, getOutputType, getPrecision, getType, outputTypeIsSet, precisionIsSet, resetOutputType, resetPrecision, setInput, setMetadata, setMetadata, setName, setName, setOutputType, setOutputType, setPrecision, setPrecision
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
public ITopKLayer(Pointer p)
Pointer(Pointer)
.@NoException(value=true) public void setOperation(nvinfer.TopKOperation op)
getOperation(), TopKOperation
@NoException(value=true) public void setOperation(@Cast(value="nvinfer1::TopKOperation") int op)
@NoException(value=true) public nvinfer.TopKOperation getOperation()
setOperation(), TopKOperation
@NoException(value=true) public void setK(int k)
getK()
@NoException(value=true) public int getK()
setK()
@NoException(value=true) public void setReduceAxes(@Cast(value="uint32_t") int reduceAxes)
getReduceAxes()
@Cast(value="uint32_t") @NoException(value=true) public int getReduceAxes()
setReduceAxes()
Copyright © 2024. All rights reserved.