| 210 | } |
| 211 | |
| 212 | void VoxelGeneratorPlugin::configurePlugin(nvinfer1::DynamicPluginTensorDesc const* in, int32_t nbInputs, |
| 213 | nvinfer1::DynamicPluginTensorDesc const* out, int32_t nbOutputs) noexcept |
| 214 | { |
| 215 | try |
| 216 | { |
| 217 | PLUGIN_VALIDATE(in != nullptr); |
| 218 | PLUGIN_VALIDATE(out != nullptr); |
| 219 | PLUGIN_VALIDATE(nbInputs == 2); |
| 220 | PLUGIN_VALIDATE(nbOutputs == 3); |
| 221 | |
| 222 | mPointFeatureNum = in[0].desc.dims.d[2]; |
| 223 | mGridXSize = npRound((mMaxXRange - mMinXRange) / mPillarXSize); |
| 224 | mGridYSize = npRound((mMaxYRange - mMinYRange) / mPillarYSize); |
| 225 | mGridZSize = npRound((mMaxZRange - mMinZRange) / mPillarZSize); |
| 226 | } |
| 227 | catch (std::exception const& e) |
| 228 | { |
| 229 | caughtError(e); |
| 230 | } |
| 231 | } |
| 232 | |
| 233 | size_t VoxelGeneratorPlugin::getWorkspaceSize(nvinfer1::PluginTensorDesc const* inputs, int32_t nbInputs, |
| 234 | nvinfer1::PluginTensorDesc const* outputs, int32_t nbOutputs) const noexcept |
nothing calls this directly
no test coverage detected