MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / configure

Method configure

src/core/NEON/kernels/NERangeKernel.cpp:128–147  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

126}
127
128void 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
149Status NERangeKernel::validate(const ITensorInfo *output, float start, float end, float step)
150{

Callers

nothing calls this directly

Calls 9

auto_init_if_emptyFunction · 0.85
StepsClass · 0.85
validate_argumentsFunction · 0.70
TensorShapeClass · 0.50
num_of_elements_in_rangeFunction · 0.50
calculate_max_windowFunction · 0.50
infoMethod · 0.45
data_typeMethod · 0.45
quantization_infoMethod · 0.45

Tested by

no test coverage detected