| Package | Description |
|---|---|
| org.bytedeco.tensorrt.nvinfer |
| Modifier and Type | Method and Description |
|---|---|
PluginTensorDesc |
DynamicPluginTensorDesc.desc()
Information required to interpret a pointer to tensor data, except that desc.dims has -1 in place of any runtime dimension.
|
PluginTensorDesc |
PluginTensorDesc.dims(Dims32 setter) |
PluginTensorDesc |
PluginTensorDesc.format(nvinfer.TensorFormat setter) |
PluginTensorDesc |
PluginTensorDesc.getPointer(long i) |
PluginTensorDesc |
PluginTensorDesc.position(long position) |
PluginTensorDesc |
PluginTensorDesc.scale(float setter) |
PluginTensorDesc |
PluginTensorDesc.type(nvinfer.DataType setter) |
| Modifier and Type | Method and Description |
|---|---|
void |
IPluginV2IOExt.configurePlugin(PluginTensorDesc in,
int nbInput,
PluginTensorDesc out,
int nbOutput)
\brief Configure the layer.
|
DynamicPluginTensorDesc |
DynamicPluginTensorDesc.desc(PluginTensorDesc setter) |
int |
IPluginV2DynamicExt.enqueue(PluginTensorDesc inputDesc,
PluginTensorDesc outputDesc,
Pointer inputs,
Pointer outputs,
Pointer workspace,
org.bytedeco.cuda.cudart.CUstream_st stream) |
int |
IPluginV2DynamicExt.enqueue(PluginTensorDesc inputDesc,
PluginTensorDesc outputDesc,
PointerPointer inputs,
PointerPointer outputs,
Pointer workspace,
org.bytedeco.cuda.cudart.CUstream_st stream)
\brief Execute the layer.
|
long |
IPluginV2DynamicExt.getWorkspaceSize(PluginTensorDesc inputs,
int nbInputs,
PluginTensorDesc outputs,
int nbOutputs)
\brief Find the workspace size required by the layer.
|
boolean |
IPluginV2IOExt.supportsFormatCombination(int pos,
PluginTensorDesc inOut,
int nbInputs,
int nbOutputs)
\brief Return true if plugin supports the format and datatype for the input/output indexed by pos.
|
boolean |
IPluginV2DynamicExt.supportsFormatCombination(int pos,
PluginTensorDesc inOut,
int nbInputs,
int nbOutputs)
\brief Return true if plugin supports the format and datatype for the input/output indexed by pos.
|
Copyright © 2024. All rights reserved.