@Namespace(value="nvinfer1") @NoOffset @Properties(inherit=nvinfer.class) public class INormalizationLayer extends ILayer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
INormalizationLayer(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
int |
getAxes()
\brief Get the axes value used for the normalization calculation.
|
nvinfer.DataType |
getComputePrecision()
\brief Get the compute precision of this layer.
|
float |
getEpsilon()
\brief Get the epsilon value used for the normalization calculation.
|
int |
getNbGroups()
\brief Get the number of groups used to split the channels for the normalization calculation.
|
void |
setAxes(int axesMask)
\brief Set the reduction axes for the normalization calculation.
|
void |
setComputePrecision(int type) |
void |
setComputePrecision(nvinfer.DataType type)
\brief Set the compute precision of this layer.
|
void |
setEpsilon(float eps)
\brief Set the epsilon value used for the normalization calculation.
|
void |
setNbGroups(int nbGroups)
\brief Set the number of groups used to split the channels in the normalization calculation.
|
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 INormalizationLayer(Pointer p)
Pointer(Pointer)
.@NoException(value=true) public void setEpsilon(float eps)
eps
- The epsilon value used for the normalization calculation.@NoException(value=true) public float getEpsilon()
@NoException(value=true) public void setAxes(@Cast(value="uint32_t") int axesMask)
axesMask
- The axes used for the normalization calculation.@Cast(value="uint32_t") @NoException(value=true) public int getAxes()
@NoException(value=true) public void setNbGroups(int nbGroups)
nbGroups
- The number of groups to split the channels into for the normalization calculation.@NoException(value=true) public int getNbGroups()
@NoException(value=true) public void setComputePrecision(nvinfer.DataType type)
type
- The datatype used for the compute precision of this layer.
By default TensorRT will run the normalization computation in DataType::kFLOAT32 even in mixed precision
mode regardless of any set builder flags to avoid overflow errors. To override this default,
use this function to set the desired compute precision.
setPrecision() and setOutputPrecision() functions can still be called to control the input and output data types
to this layer.
Only DataType::kFLOAT32 and DataType::kHALF are valid types for \p type.@NoException(value=true) public void setComputePrecision(@Cast(value="nvinfer1::DataType") int type)
@NoException(value=true) public nvinfer.DataType getComputePrecision()
Copyright © 2024. All rights reserved.