Package | Description |
---|---|
org.bytedeco.tensorrt.nvinfer |
Modifier and Type | Method and Description |
---|---|
IGpuAllocator |
IGpuAllocator.getPointer(long i) |
IGpuAllocator |
VExecutionContext.getTemporaryStorageAllocator() |
IGpuAllocator |
IExecutionContext.getTemporaryStorageAllocator()
\brief Get allocator set by setTemporaryStorageAllocator.
|
IGpuAllocator |
IGpuAllocator.position(long position) |
Modifier and Type | Method and Description |
---|---|
void |
IPluginV2Ext.attachToContext(cudnnContext arg0,
cublasContext arg1,
IGpuAllocator arg2)
Deprecated.
\brief Attach the plugin object to an execution context and grant the plugin the access to some context
resource.
|
void |
VRuntime.setGpuAllocator(IGpuAllocator allocator) |
void |
IBuilder.setGpuAllocator(IGpuAllocator allocator)
\brief Set the GPU allocator.
|
void |
VBuilder.setGpuAllocator(IGpuAllocator allocator) |
void |
IRuntime.setGpuAllocator(IGpuAllocator allocator)
\brief Set the GPU allocator.
|
boolean |
VExecutionContext.setTemporaryStorageAllocator(IGpuAllocator allocator) |
boolean |
IExecutionContext.setTemporaryStorageAllocator(IGpuAllocator allocator)
\brief Specify allocator to use for internal temporary storage.
|
Copyright © 2024. All rights reserved.