| 36 | std::vector<PluginField> NormalizePluginCreator::mPluginAttributes; |
| 37 | |
| 38 | Normalize::Normalize(Weights const* weights, int32_t nbWeights, bool acrossSpatial, bool channelShared, float eps) |
| 39 | : acrossSpatial(acrossSpatial) |
| 40 | , channelShared(channelShared) |
| 41 | , eps(eps) |
| 42 | { |
| 43 | mNbWeights = nbWeights; |
| 44 | PLUGIN_VALIDATE(nbWeights == 1); |
| 45 | PLUGIN_VALIDATE(weights[0].count >= 1); |
| 46 | mWeights = copyToDevice(weights[0].values, weights[0].count); |
| 47 | mScalarScale = static_cast<float const*>(weights[0].values)[0]; |
| 48 | } |
| 49 | |
| 50 | Normalize::Normalize(Weights const* weights, int32_t nbWeights, float scalarScale, bool acrossSpatial, |
| 51 | bool channelShared, float eps, int32_t C, int32_t H, int32_t W) |
no test coverage detected