org.bytedeco.tensorrt.nvinfer

Class IShuffleLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorrt.nvinfer.INoCopy

org.bytedeco.tensorrt.nvinfer.ILayer

org.bytedeco.tensorrt.nvinfer.IShuffleLayer

All Implemented Interfaces:

AutoCloseable

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

\class IShuffleLayer \brief Layer type for shuffling data. This layer shuffles data by applying in sequence: a transpose operation, a reshape operation and a second transpose operation. The dimension types of the output are those of the reshape dimension. The layer has an optional second input. If present, it must be a 1D tensor of type Int32 or Int64, and the reshape dimensions are taken from it. \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
IShuffleLayer(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
Permutation	getFirstTranspose() \brief Get the permutation applied by the first transpose operation.
Dims64	getReshapeDimensions() \brief Get the reshaped dimensions.
Permutation	getSecondTranspose() \brief Get the permutation applied by the second transpose operation.
boolean	getZeroIsPlaceholder() \brief Get meaning of 0 in reshape dimensions.
void	setFirstTranspose(Permutation permutation) \brief Set the permutation applied by the first transpose operation.
void	setReshapeDimensions(Dims64 dimensions) \brief Set the reshaped dimensions.
void	setSecondTranspose(Permutation permutation) \brief Set the permutation applied by the second transpose operation.
void	setZeroIsPlaceholder(boolean zeroIsPlaceholder) \brief Set meaning of 0 in reshape dimensions.

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

IShuffleLayer

public IShuffleLayer(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

setFirstTranspose

@NoException(value=true)
public void setFirstTranspose(@ByVal
                                                        Permutation permutation)

\brief Set the permutation applied by the first transpose operation.

Parameters:

permutation - The dimension permutation applied before the reshape. The default is the identity permutation.

See Also:

getFirstTranspose

getFirstTranspose

@ByVal
 @NoException(value=true)
public Permutation getFirstTranspose()

\brief Get the permutation applied by the first transpose operation.

Returns:

The dimension permutation applied before the reshape.

See Also:

setFirstTranspose

setReshapeDimensions

@NoException(value=true)
public void setReshapeDimensions(@Cast(value="const nvinfer1::Dims*") @ByRef
                                                           Dims64 dimensions)

\brief Set the reshaped dimensions.

Parameters:

dimensions - The reshaped dimensions. Two special values can be used as dimensions. Value 0 copies the corresponding dimension from input. This special value can be used more than once in the dimensions. If number of reshape dimensions is less than input, 0s are resolved by aligning the most significant dimensions of input. Value -1 infers that particular dimension by looking at input and rest of the reshape dimensions. Note that only a maximum of one dimension is permitted to be specified as -1. Avoid using -1 if the input can have zero volume and any of the other reshape dimensions can be zero (after resolving special treatment of 0), because the solution for the -1 becomes indeterminate and TensorRT will report an error. The product of the new dimensions must be equal to the product of the old. If a second input had been used to create this layer, that input is reset to null by this method.

getReshapeDimensions

@ByVal
 @Cast(value="nvinfer1::Dims*")
 @NoException(value=true)
public Dims64 getReshapeDimensions()

\brief Get the reshaped dimensions.

Returns:

The reshaped dimensions. If a second input is present and non-null, or setReshapeDimensions has not yet been called, this function returns Dims with nbDims == -1.

setSecondTranspose

@NoException(value=true)
public void setSecondTranspose(@ByVal
                                                         Permutation permutation)

\brief Set the permutation applied by the second transpose operation.

Parameters:

permutation - The dimension permutation applied after the reshape. The default is the identity permutation. The permutation is applied as outputDimensionIndex = permutation.order[inputDimensionIndex], so to permute from CHW order to HWC order, the required permutation is [1, 2, 0].

See Also:

getSecondTranspose

getSecondTranspose

@ByVal
 @NoException(value=true)
public Permutation getSecondTranspose()

\brief Get the permutation applied by the second transpose operation.

Returns:

The dimension permutation applied after the reshape.

See Also:

setSecondTranspose

setZeroIsPlaceholder

@NoException(value=true)
public void setZeroIsPlaceholder(@Cast(value="bool")
                                                           boolean zeroIsPlaceholder)

\brief Set meaning of 0 in reshape dimensions. If true, then a 0 in the reshape dimensions denotes copying the corresponding dimension from the first input tensor. If false, then a 0 in the reshape dimensions denotes a zero-length dimension. Default: true

See Also:

getZeroIsPlaceholder();

getZeroIsPlaceholder

@Cast(value="bool")
 @NoException(value=true)
public boolean getZeroIsPlaceholder()

\brief Get meaning of 0 in reshape dimensions.

Returns:

true if 0 is placeholder for corresponding input dimension, false if 0 denotes a zero-length dimension.

See Also:

setZeroIsPlaceholder