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.