Package | Description |
---|---|
org.bytedeco.tensorrt.global | |
org.bytedeco.tensorrt.nvinfer |
Modifier and Type | Method and Description |
---|---|
nvinfer.TopKOperation |
nvinfer.TopKOperation.intern() |
static nvinfer.TopKOperation |
nvinfer.TopKOperation.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static nvinfer.TopKOperation[] |
nvinfer.TopKOperation.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
nvinfer.TopKOperation |
ITopKLayer.getOperation()
\brief Get the operation for the layer.
|
nvinfer.TopKOperation |
VTopKLayer.getOperation() |
Modifier and Type | Method and Description |
---|---|
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) |
void |
ITopKLayer.setOperation(nvinfer.TopKOperation op)
\brief Set the operation for the layer.
|
void |
VTopKLayer.setOperation(nvinfer.TopKOperation op) |
Copyright © 2024. All rights reserved.