| Package | Description |
|---|---|
| org.bytedeco.tensorrt.global | |
| org.bytedeco.tensorrt.nvinfer |
| Modifier and Type | Method and Description |
|---|---|
nvinfer.GatherMode |
nvinfer.GatherMode.intern() |
static nvinfer.GatherMode |
nvinfer.GatherMode.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static nvinfer.GatherMode[] |
nvinfer.GatherMode.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
| Modifier and Type | Method and Description |
|---|---|
nvinfer.GatherMode |
IGatherLayer.getMode()
\brief Get the gather mode.
|
nvinfer.GatherMode |
VGatherLayer.getMode() |
| Modifier and Type | Method and Description |
|---|---|
IGatherLayer |
INetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
nvinfer.GatherMode mode)
\brief Add gather with specified mode, axis=0 and nbElementWiseDims=0.
|
IGatherLayer |
VNetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
nvinfer.GatherMode mode) |
void |
IGatherLayer.setMode(nvinfer.GatherMode mode)
\brief Set the gather mode.
|
void |
VGatherLayer.setMode(nvinfer.GatherMode mode) |
Copyright © 2024. All rights reserved.