| 131 | } |
| 132 | |
| 133 | void ClipPlugin::configureWithFormat(Dims const* inputs, int32_t nbInputs, Dims const* outputs, int32_t nbOutputs, |
| 134 | DataType type, PluginFormat format, int32_t) noexcept |
| 135 | { |
| 136 | PLUGIN_ASSERT(nbOutputs == 1); |
| 137 | PLUGIN_API_CHECK_ENUM_RANGE(DataType, type); |
| 138 | PLUGIN_API_CHECK_ENUM_RANGE(PluginFormat, format); |
| 139 | mDataType = type; |
| 140 | |
| 141 | size_t volume = 1; |
| 142 | for (int32_t i = 0; i < inputs->nbDims; i++) |
| 143 | { |
| 144 | volume *= inputs->d[i]; |
| 145 | } |
| 146 | mInputVolume = volume; |
| 147 | } |
| 148 | |
| 149 | bool ClipPlugin::supportsFormat(DataType type, PluginFormat format) const noexcept |
| 150 | { |