Searched refs:_input_squared (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NENormalizationLayer.cpp | 40 : _memory_group(std::move(memory_manager)), _norm_kernel(), _multiply_f(), _input_squared() in NENormalizationLayer()
50 _input_squared.allocator()->init(tensor_info); in configure()
53 _memory_group.manage(&_input_squared); in configure()
57 _norm_kernel->configure(input, &_input_squared, output, norm_info); in configure()
58 …_multiply_f.configure(input, input, &_input_squared, 1.0f, ConvertPolicy::SATURATE, RoundingPolicy… in configure()
61 _input_squared.allocator()->allocate(); in configure()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NENormalizationLayerKernel.cpp | 68 …: _func(nullptr), _input(nullptr), _input_squared(nullptr), _output(nullptr), _norm_info(NormType:… in NENormalizationLayerKernel()
84 _input_squared = input_squared; in configure()
183 Iterator input_squared(_input_squared, win); in normalize_float()
188 const int input_squared_stride_x = _input_squared->info()->strides_in_bytes()[0]; in normalize_float()
189 const int input_squared_stride_slice = _input_squared->info()->strides_in_bytes()[dim]; in normalize_float()
190 const int input_squared_stride_row = _input_squared->info()->strides_in_bytes()[dim_y]; in normalize_float()
|
| D | NENormalizationLayerKernel.h | 103 const ITensor *_input_squared; variable
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NENormalizationLayer.h | 100 …Tensor _input_squared; /**< The intermediate buffer which sto… variable
|