@Namespace(value="nvinfer1") @Properties(inherit=nvinfer.class) public class IAlgorithmSelector extends Pointer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
IAlgorithmSelector(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
void |
reportAlgorithms(IAlgorithmContext algoContexts,
IAlgorithm algoChoices,
int nbAlgorithms) |
void |
reportAlgorithms(PointerPointer algoContexts,
PointerPointer algoChoices,
int nbAlgorithms)
\brief Called by TensorRT to report choices it made.
|
int |
selectAlgorithms(IAlgorithmContext context,
IAlgorithm choices,
int nbChoices,
int[] selection) |
int |
selectAlgorithms(IAlgorithmContext context,
IAlgorithm choices,
int nbChoices,
IntBuffer selection) |
int |
selectAlgorithms(IAlgorithmContext context,
IAlgorithm choices,
int nbChoices,
IntPointer selection) |
int |
selectAlgorithms(IAlgorithmContext context,
PointerPointer choices,
int nbChoices,
IntPointer selection)
\brief Select Algorithms for a layer from the given list of algorithm choices.
|
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 IAlgorithmSelector(Pointer p)
Pointer(Pointer)
.@NoException(value=true) public int selectAlgorithms(@Const @ByRef IAlgorithmContext context, @Cast(value="const nvinfer1::IAlgorithm*const*") PointerPointer choices, int nbChoices, IntPointer selection)
context
- The context for which the algorithm choices are valid.choices
- The list of algorithm choices to select for implementation of this layer.nbChoices
- Number of algorithm choices.selection
- The user writes indices of selected choices in to selection buffer which is of size nbChoices.
\note TensorRT uses its default algorithm selection to choose from the list provided.
If return value is 0, TensorRT's default algorithm selection is used unless
BuilderFlag::kREJECT_EMPTY_ALGORITHMS (or the deprecated BuilderFlag::kSTRICT_TYPES) is set.
The list of choices is valid only for this specific algorithm context.@NoException(value=true) public int selectAlgorithms(@Const @ByRef IAlgorithmContext context, @Const @ByPtrPtr IAlgorithm choices, int nbChoices, IntPointer selection)
@NoException(value=true) public int selectAlgorithms(@Const @ByRef IAlgorithmContext context, @Const @ByPtrPtr IAlgorithm choices, int nbChoices, IntBuffer selection)
@NoException(value=true) public int selectAlgorithms(@Const @ByRef IAlgorithmContext context, @Const @ByPtrPtr IAlgorithm choices, int nbChoices, int[] selection)
@NoException(value=true) public void reportAlgorithms(@Cast(value="const nvinfer1::IAlgorithmContext*const*") PointerPointer algoContexts, @Cast(value="const nvinfer1::IAlgorithm*const*") PointerPointer algoChoices, int nbAlgorithms)
algoContexts
- The list of all algorithm contexts.algoChoices
- The list of algorithm choices made by TensorRTnbAlgorithms
- The size of algoContexts as well as algoChoices.@NoException(value=true) public void reportAlgorithms(@Const @ByPtrPtr IAlgorithmContext algoContexts, @Const @ByPtrPtr IAlgorithm algoChoices, int nbAlgorithms)
Copyright © 2024. All rights reserved.