Searched refs:_actual_axis (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEL2NormalizeLayerKernel.cpp | 150 : _input(nullptr), _sum(nullptr), _output(nullptr), _actual_axis(0), _epsilon(1e-12) in NEL2NormalizeLayerKernel()
162 _actual_axis = wrap_around(axis, max_input_tensor_dim); in configure()
186 if(_actual_axis > 2) in run()
191 …mentation(L2NormalizeLayerSelectorData{ _output->info()->data_type(), _actual_axis, CPUInfo::get()… in run()
195 uk->ukernel(_input, _sum, _output, _epsilon, window, _actual_axis); in run()
|
| D | NEL2NormalizeLayerKernel.h | 86 unsigned int _actual_axis; variable
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLL2NormalizeLayerKernel.cpp | 75 : _input(nullptr), _sum(nullptr), _output(nullptr), _actual_axis(0), _epsilon(1e-12) in CLL2NormalizeLayerKernel()
94 _actual_axis = wrap_around(axis, max_input_tensor_dim); in configure()
109 switch(_actual_axis) in configure()
161 switch(_actual_axis) in run()
|
| D | CLL2NormalizeLayerKernel.h | 96 unsigned int _actual_axis; variable
|