| 163 | } |
| 164 | |
| 165 | void NEL2NormalizeLayerKernel::configure( |
| 166 | const ITensor *input, const ITensor *sum, ITensor *output, int axis, float epsilon) |
| 167 | { |
| 168 | ARM_COMPUTE_ERROR_ON_NULLPTR(input, sum, output); |
| 169 | ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(input->info(), sum->info(), output->info(), axis, epsilon)); |
| 170 | |
| 171 | _input = input; |
| 172 | _sum = sum; |
| 173 | _output = output; |
| 174 | _actual_axis = wrap_around(axis, max_input_tensor_dim); |
| 175 | _epsilon = epsilon; |
| 176 | |
| 177 | // Configure kernel window |
| 178 | auto win_config = validate_and_configure_window(_input->info(), _output->info()); |
| 179 | ARM_COMPUTE_ERROR_THROW_ON(std::get<0>(win_config)); |
| 180 | |
| 181 | INEKernel::configure(std::get<1>(win_config)); |
| 182 | } |
| 183 | |
| 184 | Status NEL2NormalizeLayerKernel::validate( |
| 185 | const ITensorInfo *input, const ITensorInfo *sum, const ITensorInfo *output, int axis, float epsilon) |
nothing calls this directly
no test coverage detected