Package | Description |
---|---|
org.bytedeco.tensorrt.nvinfer |
Modifier and Type | Method and Description |
---|---|
ITopKLayer |
INetworkDefinition.addTopK(ITensor input,
int op,
int k,
int reduceAxes) |
ITopKLayer |
VNetworkDefinition.addTopK(ITensor input,
int op,
int k,
int reduceAxes) |
ITopKLayer |
INetworkDefinition.addTopK(ITensor input,
nvinfer.TopKOperation op,
int k,
int reduceAxes)
\brief Add a TopK layer to the network.
|
ITopKLayer |
VNetworkDefinition.addTopK(ITensor input,
nvinfer.TopKOperation op,
int k,
int reduceAxes) |
Copyright © 2024. All rights reserved.