| 88 | } |
| 89 | |
| 90 | Dims Normalize::getOutputDimensions(int32_t index, Dims const* inputs, int32_t nbInputDims) noexcept |
| 91 | { |
| 92 | PLUGIN_ASSERT(nbInputDims == 1); |
| 93 | PLUGIN_ASSERT(index == 0); |
| 94 | PLUGIN_ASSERT(inputs[0].nbDims == 3); |
| 95 | return Dims3(inputs[0].d[0], inputs[0].d[1], inputs[0].d[2]); |
| 96 | } |
| 97 | |
| 98 | int32_t Normalize::initialize() noexcept |
| 99 | { |