Searched refs:_norm_info (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NENormalizationLayerKernel.cpp | 68 …: _func(nullptr), _input(nullptr), _input_squared(nullptr), _output(nullptr), _norm_info(NormType:… in NENormalizationLayerKernel()
86 _norm_info = norm_info; in configure()
187 const int radius = _norm_info.norm_size() / 2; in normalize_float()
195 … const auto coeff_vec = wrapper::vdup_n(static_cast<T>(_norm_info.scale_coeff()), ExactTagType{}); in normalize_float()
196 const auto beta_vec = wrapper::vdup_n(static_cast<T>(_norm_info.beta()), ExactTagType{}); in normalize_float()
197 const auto kappa_vec = wrapper::vdup_n(static_cast<T>(_norm_info.kappa()), ExactTagType{}); in normalize_float()
220 … = std::pow(accu * static_cast<T>(_norm_info.scale_coeff()) + static_cast<T>(_norm_info.kappa… in normalize_float()
|
| D | NENormalizationLayerKernel.h | 105 NormalizationLayerInfo _norm_info; variable
|
| /external/ComputeLibrary/arm_compute/graph/frontend/ |
| D | Layers.h | 876 : _norm_info(norm_info) in NormalizationLayer()
884 return GraphBuilder::add_normalization_node(s.graph(), common_params, input, _norm_info); in create_layer()
888 NormalizationLayerInfo _norm_info;
|