org.bytedeco.tensorrt.nvinfer

Class IIteratorLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorrt.nvinfer.INoCopy

org.bytedeco.tensorrt.nvinfer.ILayer

org.bytedeco.tensorrt.nvinfer.ILoopBoundaryLayer

org.bytedeco.tensorrt.nvinfer.IIteratorLayer

All Implemented Interfaces:

AutoCloseable

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

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

Method Summary

Modifier and Type	Method and Description
int	getAxis() Get axis being iterated over.
boolean	getReverse() True if and only if reversing input.
void	setAxis(int axis) Set axis to iterate over.
void	setReverse(boolean reverse) For reverse=false, the layer is equivalent to addGather(tensor, I, 0) where I is a scalar tensor containing the loop iteration number.

Methods inherited from class org.bytedeco.tensorrt.nvinfer.ILoopBoundaryLayer

getLoop

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

IIteratorLayer

public IIteratorLayer(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

setAxis

@NoException(value=true)
public void setAxis(int axis)

Set axis to iterate over.

getAxis

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

Get axis being iterated over.

setReverse

@NoException(value=true)
public void setReverse(@Cast(value="bool")
                                                boolean reverse)

For reverse=false, the layer is equivalent to addGather(tensor, I, 0) where I is a scalar tensor containing the loop iteration number. For reverse=true, the layer is equivalent to addGather(tensor, M-1-I, 0) where M is the trip count computed from TripLimits of kind kCOUNT. The default is reverse=false.

getReverse

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

True if and only if reversing input.