| Package | Description |
|---|---|
| org.bytedeco.tensorrt.global | |
| org.bytedeco.tensorrt.nvinfer |
| Modifier and Type | Method and Description |
|---|---|
nvinfer.ReduceOperation |
nvinfer.ReduceOperation.intern() |
static nvinfer.ReduceOperation |
nvinfer.ReduceOperation.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static nvinfer.ReduceOperation[] |
nvinfer.ReduceOperation.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
| Modifier and Type | Method and Description |
|---|---|
nvinfer.ReduceOperation |
IReduceLayer.getOperation()
\brief Get the reduce operation for the layer.
|
nvinfer.ReduceOperation |
VReduceLayer.getOperation() |
| Modifier and Type | Method and Description |
|---|---|
IReduceLayer |
INetworkDefinition.addReduce(ITensor input,
nvinfer.ReduceOperation operation,
int reduceAxes,
boolean keepDimensions)
\brief Add a reduce layer to the network.
|
IReduceLayer |
VNetworkDefinition.addReduce(ITensor input,
nvinfer.ReduceOperation operation,
int reduceAxes,
boolean keepDimensions) |
void |
IReduceLayer.setOperation(nvinfer.ReduceOperation op)
\brief Set the reduce operation for the layer.
|
void |
VReduceLayer.setOperation(nvinfer.ReduceOperation op) |
Copyright © 2024. All rights reserved.