@Namespace(value="nvinfer1") @NoOffset @Properties(inherit=nvinfer.class) public class IIteratorLayer extends ILoopBoundaryLayer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
IIteratorLayer(Pointer p)
Pointer cast constructor.
|
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.
|
getLoop
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 IIteratorLayer(Pointer p)
Pointer(Pointer)
.@NoException(value=true) public void setAxis(int axis)
@NoException(value=true) public int getAxis()
@NoException(value=true) public void setReverse(@Cast(value="bool") boolean reverse)
@Cast(value="bool") @NoException(value=true) public boolean getReverse()
Copyright © 2024. All rights reserved.