| 150 | } |
| 151 | |
| 152 | void BatchTilePlugin::configurePlugin(Dims const* inputDims, int32_t nbInputs, Dims const* outputDims, |
| 153 | int32_t nbOutputs, DataType const* inputTypes, DataType const* outputTypes, bool const* inputIsBroadcast, |
| 154 | bool const* outputIsBroadcast, PluginFormat floatFormat, int32_t maxBatchSize) noexcept |
| 155 | { |
| 156 | try |
| 157 | { |
| 158 | PLUGIN_VALIDATE(inputDims != nullptr); |
| 159 | PLUGIN_VALIDATE(outputDims != nullptr); |
| 160 | PLUGIN_VALIDATE(nbOutputs == 1); |
| 161 | PLUGIN_VALIDATE(inputDims[1].nbDims == 4); |
| 162 | PLUGIN_VALIDATE(inputDims[1].d[0] == 1); |
| 163 | mCopySize = pluginInternal::volume(inputDims[1]) * sizeof(float); |
| 164 | } |
| 165 | catch (std::exception const& e) |
| 166 | { |
| 167 | caughtError(e); |
| 168 | } |
| 169 | } |
| 170 | |
| 171 | bool BatchTilePlugin::supportsFormat(DataType type, PluginFormat format) const noexcept |
| 172 | { |
nothing calls this directly
no test coverage detected