| Package | Description |
|---|---|
| org.bytedeco.tensorrt.global | |
| org.bytedeco.tensorrt.nvinfer |
| Modifier and Type | Method and Description |
|---|---|
nvinfer.ScatterMode |
nvinfer.ScatterMode.intern() |
static nvinfer.ScatterMode |
nvinfer.ScatterMode.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static nvinfer.ScatterMode[] |
nvinfer.ScatterMode.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
| Modifier and Type | Method and Description |
|---|---|
nvinfer.ScatterMode |
IScatterLayer.getMode()
\brief Get the scatter mode.
|
nvinfer.ScatterMode |
VScatterLayer.getMode() |
| Modifier and Type | Method and Description |
|---|---|
IScatterLayer |
INetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
nvinfer.ScatterMode mode)
\brief Add a Scatter layer to the network with specified mode and axis=0.
|
IScatterLayer |
VNetworkDefinition.addScatter(ITensor data,
ITensor indices,
ITensor updates,
nvinfer.ScatterMode mode) |
void |
IScatterLayer.setMode(nvinfer.ScatterMode mode)
\brief Set the scatter mode.
|
void |
VScatterLayer.setMode(nvinfer.ScatterMode mode) |
Copyright © 2024. All rights reserved.