Package | Description |
---|---|
org.bytedeco.tensorrt.nvinfer |
Modifier and Type | Method and Description |
---|---|
IGatherLayer |
INetworkDefinition.addGather(ITensor data,
ITensor indices,
int axis)
\brief Add gather with mode GatherMode::kDEFAULT and specified axis and nbElementWiseDims=0.
|
IGatherLayer |
VNetworkDefinition.addGather(ITensor data,
ITensor indices,
int axis) |
IGatherLayer |
INetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
int mode) |
IGatherLayer |
VNetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
int mode) |
IGatherLayer |
INetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
nvinfer.GatherMode mode)
\brief Add gather with specified mode, axis=0 and nbElementWiseDims=0.
|
IGatherLayer |
VNetworkDefinition.addGatherV2(ITensor data,
ITensor indices,
nvinfer.GatherMode mode) |
Copyright © 2024. All rights reserved.