Package | Description |
---|---|
org.bytedeco.tensorrt.global | |
org.bytedeco.tensorrt.nvinfer |
Modifier and Type | Method and Description |
---|---|
nvinfer.ScaleMode |
nvinfer.ScaleMode.intern() |
static nvinfer.ScaleMode |
nvinfer.ScaleMode.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static nvinfer.ScaleMode[] |
nvinfer.ScaleMode.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
nvinfer.ScaleMode |
VScaleLayer.getMode() |
nvinfer.ScaleMode |
IScaleLayer.getMode()
\brief Get the scale mode.
|
Modifier and Type | Method and Description |
---|---|
IScaleLayer |
INetworkDefinition.addScale(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power)
\brief Add a Scale layer to the network.
|
IScaleLayer |
VNetworkDefinition.addScale(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power) |
IScaleLayer |
INetworkDefinition.addScaleNd(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis)
\brief Add a multi-dimension scale layer to the network.
|
IScaleLayer |
VNetworkDefinition.addScaleNd(ITensor input,
nvinfer.ScaleMode mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis) |
void |
VScaleLayer.setMode(nvinfer.ScaleMode mode) |
void |
IScaleLayer.setMode(nvinfer.ScaleMode mode)
\brief Set the scale mode.
|
Copyright © 2024. All rights reserved.