| 126 | } |
| 127 | |
| 128 | void NERangeKernel::configure(ITensor *output, float start, float end, float step) |
| 129 | { |
| 130 | ARM_COMPUTE_ERROR_ON_NULLPTR(output); |
| 131 | |
| 132 | ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(*(output->info()), start, end, step)); |
| 133 | |
| 134 | // Auto initialize output if not initialized |
| 135 | auto_init_if_empty(*output->info(), TensorShape(num_of_elements_in_range(start, end, step)), 1, |
| 136 | output->info()->data_type(), output->info()->quantization_info()); |
| 137 | |
| 138 | // Configure kernel window |
| 139 | Window win = calculate_max_window(*output->info(), Steps()); |
| 140 | |
| 141 | _start = start; |
| 142 | _end = end; |
| 143 | _step = step; |
| 144 | _output = output; |
| 145 | |
| 146 | INEKernel::configure(win); |
| 147 | } |
| 148 | |
| 149 | Status NERangeKernel::validate(const ITensorInfo *output, float start, float end, float step) |
| 150 | { |
nothing calls this directly
no test coverage detected