org.bytedeco.tensorrt.nvinfer

Class ISliceLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorrt.nvinfer.INoCopy

org.bytedeco.tensorrt.nvinfer.ILayer

org.bytedeco.tensorrt.nvinfer.ISliceLayer

All Implemented Interfaces:

AutoCloseable

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

\brief Slices an input tensor into an output tensor based on the offset and strides. The slice layer has two variants, static and dynamic. Static slice specifies the start, size, and stride dimensions at layer creation time via Dims and can use the get/set accessor functions of the ISliceLayer. Dynamic slice specifies one or more of start, size or stride as ITensors, by using ILayer::setInput to add a second, third, or fourth input respectively. The corresponding Dims are used if an input is missing or null. An application can determine if the ISliceLayer has a dynamic output shape based on whether the size input (third input) is present and non-null. The slice layer selects for each dimension a start location from within the input tensor, and copies elements to the output tensor using the specified stride across the input tensor. Start, size, and stride tensors must be 1D Int32 shape tensors if not specified via Dims. An example of using slice on a tensor: input = {{0, 2, 4}, {1, 3, 5}} start = {1, 0} size = {1, 2} stride = {1, 2} output = {{1, 5}} When the sliceMode is kCLAMP or kREFLECT, for each input dimension, if its size is 0 then the corresponding output dimension must be 0 too. A slice layer can produce a shape tensor if the following conditions are met: * start, size, and stride are build time constants, either as static Dims or as constant input tensors. * The number of elements in the output tensor does not exceed 2 * Dims::MAX_DIMS. The input tensor is a shape tensor if the output is a shape tensor. The following constraints must be satisfied to execute this layer on DLA: * start, size, and stride are build time constants, either as static Dims or as constant input tensors. * sliceMode is kDEFAULT. * Strides are 1 for all dimensions. * Slicing is not performed on the first dimension * The input tensor has four dimensions \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
ISliceLayer(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
nvinfer.SampleMode	getMode() \brief Get the slice mode.
Dims32	getSize() \brief Get dimensions of the output slice.
Dims32	getStart() \brief Get the start offset for the slice layer.
Dims32	getStride() \brief Get the stride for the output slice.
void	setMode(int mode)
void	setMode(nvinfer.SampleMode mode) \brief Set the slice mode.
void	setSize(Dims32 size) \brief Set the dimensions of the output slice.
void	setStart(Dims32 start) \brief Set the start offset that the slice layer uses to create the output slice.
void	setStride(Dims32 stride) \brief Set the stride for computing the output slice data.

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

ISliceLayer

public ISliceLayer(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

setStart

@NoException(value=true)
public void setStart(@ByVal @Cast(value="nvinfer1::Dims*")
                                              Dims32 start)

\brief Set the start offset that the slice layer uses to create the output slice.

Parameters:

start - The start offset to read data from the input tensor. If a second input had been used to create this layer, that input is reset to null by this method.

See Also:

getStart

getStart

@ByVal
 @Cast(value="nvinfer1::Dims*")
 @NoException(value=true)
public Dims32 getStart()

\brief Get the start offset for the slice layer.

Returns:

The start offset, or an invalid Dims structure. If the second input is present and non-null, this function returns a Dims with nbDims = -1.

See Also:

setStart

setSize

@NoException(value=true)
public void setSize(@ByVal @Cast(value="nvinfer1::Dims*")
                                             Dims32 size)

\brief Set the dimensions of the output slice.

Parameters:

size - The dimensions of the output slice. If a third input had been used to create this layer, that input is reset to null by this method.

See Also:

getSize

getSize

@ByVal
 @Cast(value="nvinfer1::Dims*")
 @NoException(value=true)
public Dims32 getSize()

\brief Get dimensions of the output slice.

Returns:

The output dimension, or an invalid Dims structure. If the third input is present and non-null, this function returns a Dims with nbDims = -1.

See Also:

setSize

setStride

@NoException(value=true)
public void setStride(@ByVal @Cast(value="nvinfer1::Dims*")
                                               Dims32 stride)

\brief Set the stride for computing the output slice data.

Parameters:

stride - The dimensions of the stride to compute the values to store in the output slice. If a fourth input had been used to create this layer, that input is reset to null by this method.

See Also:

getStride

getStride

@ByVal
 @Cast(value="nvinfer1::Dims*")
 @NoException(value=true)
public Dims32 getStride()

\brief Get the stride for the output slice.

Returns:

The slicing stride, or an invalid Dims structure. If the fourth input is present and non-null, this function returns a Dims with nbDims = -1.

See Also:

setStride

setMode

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

\brief Set the slice mode.

See Also:

getMode()

setMode

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

getMode

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

\brief Get the slice mode.

See Also:

setMode()