| Package | Description |
|---|---|
| org.bytedeco.tensorrt.nvinfer |
| Modifier and Type | Method and Description |
|---|---|
IOutputAllocator |
VExecutionContext.getOutputAllocator(BytePointer name) |
IOutputAllocator |
IExecutionContext.getOutputAllocator(BytePointer tensorName) |
IOutputAllocator |
VExecutionContext.getOutputAllocator(String name) |
IOutputAllocator |
IExecutionContext.getOutputAllocator(String tensorName)
\brief Get output allocator associated with output tensor of given name, or nullptr if the provided name does
not map to an output tensor.
|
| Modifier and Type | Method and Description |
|---|---|
boolean |
VExecutionContext.setOutputAllocator(BytePointer tensorName,
IOutputAllocator outputAllocator) |
boolean |
IExecutionContext.setOutputAllocator(BytePointer tensorName,
IOutputAllocator outputAllocator) |
boolean |
VExecutionContext.setOutputAllocator(String tensorName,
IOutputAllocator outputAllocator) |
boolean |
IExecutionContext.setOutputAllocator(String tensorName,
IOutputAllocator outputAllocator)
\brief Set output allocator to use for output tensor of given name.
|
Copyright © 2024. All rights reserved.