@Namespace(value="nvinfer1") @Properties(inherit=nvinfer.class) public class IOutputAllocator extends Pointer
IExecutionContext::enqueueV3()
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
IOutputAllocator(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
int |
getInterfaceVersion()
\brief Return the API version of this IOutputAllocator.
|
void |
notifyShape(BytePointer tensorName,
Dims32 dims) |
void |
notifyShape(String tensorName,
Dims32 dims)
\brief Called by TensorRT when the shape of the output tensor is known.
|
Pointer |
reallocateOutput(BytePointer tensorName,
Pointer currentMemory,
long size,
long alignment) |
Pointer |
reallocateOutput(String tensorName,
Pointer currentMemory,
long size,
long alignment)
\brief Return a pointer to memory for an output tensor, or nullptr if memory cannot be allocated.
|
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 IOutputAllocator(Pointer p)
Pointer(Pointer)
.@NoException(value=true) public int getInterfaceVersion()
@NoException(value=true) public Pointer reallocateOutput(String tensorName, Pointer currentMemory, @Cast(value="uint64_t") long size, @Cast(value="uint64_t") long alignment)
tensorName
- name of the output tensor.currentMemory
- points to the address set by IExectionContext::setTensorAddress.size
- number of bytes required. Always positive, even for an empty tensor.alignment
- required alignment of the allocation.@NoException(value=true) public Pointer reallocateOutput(@Cast(value="const char*") BytePointer tensorName, Pointer currentMemory, @Cast(value="uint64_t") long size, @Cast(value="uint64_t") long alignment)
@NoException(value=true) public void notifyShape(String tensorName, @Cast(value="const nvinfer1::Dims*") @ByRef Dims32 dims)
dims
- dimensions of the outputtensorName
- name of the tensor@NoException(value=true) public void notifyShape(@Cast(value="const char*") BytePointer tensorName, @Cast(value="const nvinfer1::Dims*") @ByRef Dims32 dims)
Copyright © 2024. All rights reserved.