org.bytedeco.tensorrt.nvinfer

Class IScaleLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorrt.nvinfer.INoCopy

org.bytedeco.tensorrt.nvinfer.ILayer

org.bytedeco.tensorrt.nvinfer.IScaleLayer

All Implemented Interfaces:

AutoCloseable

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

\class IScaleLayer \brief A Scale layer in a network definition. This layer applies a per-element computation to its input: \p output = (\p input* \p scale + \p shift)^ \p power The coefficients can be applied on a per-tensor, per-channel, or per-element basis. \note If the number of weights is 0, then a default value is used for shift, power, and scale. The default shift is 0, the default power is 1, and the default scale is 1. The output size is the same as the input size. \note The input tensor is required to have at least 4 dimensions. A scale layer may be used as an INT8 quantization node in a graph, if the output is constrained to INT8 and the input to FP32. Quantization rounds ties to even, and clamps to [-128, 127].

See Also:

\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
IScaleLayer(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
int	getChannelAxis() \brief Get the channel axis.
nvinfer.ScaleMode	getMode() \brief Get the scale mode.
Weights	getPower() \brief Get the power value.
Weights	getScale() \brief Get the scale value.
Weights	getShift() \brief Get the shift value.
void	setChannelAxis(int channelAxis) \brief Set the channel axis.
void	setMode(int mode)
void	setMode(nvinfer.ScaleMode mode) \brief Set the scale mode.
void	setPower(Weights power) \brief Set the power value.
void	setScale(Weights scale) \brief Set the scale value.
void	setShift(Weights shift) \brief Set the shift value.

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

IScaleLayer

public IScaleLayer(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

setMode

@NoException(value=true)
public void setMode(nvinfer.ScaleMode mode)

\brief Set the scale mode.

See Also:

getMode()

setMode

@NoException(value=true)
public void setMode(@Cast(value="nvinfer1::ScaleMode")
                                             int mode)

getMode

@NoException(value=true)
public nvinfer.ScaleMode getMode()

\brief Get the scale mode.

See Also:

setMode()

setShift

@NoException(value=true)
public void setShift(@ByVal
                                              Weights shift)

\brief Set the shift value.

See Also:

getShift()

getShift

@ByVal
 @NoException(value=true)
public Weights getShift()

\brief Get the shift value.

See Also:

setShift()

setScale

@NoException(value=true)
public void setScale(@ByVal
                                              Weights scale)

\brief Set the scale value.

See Also:

getScale()

getScale

@ByVal
 @NoException(value=true)
public Weights getScale()

\brief Get the scale value.

See Also:

setScale()

setPower

@NoException(value=true)
public void setPower(@ByVal
                                              Weights power)

\brief Set the power value.

See Also:

getPower()

getPower

@ByVal
 @NoException(value=true)
public Weights getPower()

\brief Get the power value.

See Also:

setPower()

getChannelAxis

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

\brief Get the channel axis.

Returns:

channelAxis parameter passed to addScaleNd() or set by setChannelAxis() The value is the index of the channel axis in the input tensor's dimensions. Scaling happens along the channel axis when ScaleMode::kCHANNEL is enabled.

See Also:

addScaleNd()

setChannelAxis

@NoException(value=true)
public void setChannelAxis(int channelAxis)

\brief Set the channel axis. The value is the index of the channel axis in the input tensor's dimensions. For ScaleMode::kCHANNEL, there can be distinct scale, shift, and power weights for each channel coordinate. For ScaleMode::kELEMENTWISE, there can be distinct scale, shift, and power weights for each combination of coordinates from the channel axis and axes after it. For example, suppose the input tensor has dimensions [10,20,30,40] and the channel axis is 1. Let [n,c,h,w] denote an input coordinate. For ScaleMode::kCHANNEL, the scale, shift, and power weights are indexed by c. For ScaleMode::kELEMENTWISE, the scale, shift, and power weights are indexed by [c,h,w].

See Also:

addScaleNd()