Package | Description |
---|---|
org.bytedeco.tensorrt.nvinfer |
Modifier and Type | Method and Description |
---|---|
IScaleLayer |
INetworkDefinition.addScale(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power) |
IScaleLayer |
VNetworkDefinition.addScale(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power) |
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,
int mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis) |
IScaleLayer |
VNetworkDefinition.addScaleNd(ITensor input,
int mode,
Weights shift,
Weights scale,
Weights power,
int channelAxis) |
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) |
Copyright © 2024. All rights reserved.