Home
last modified time | relevance | path

Searched refs:_epsilon (Results 1 – 25 of 33) sorted by relevance

12

/external/ComputeLibrary/src/core/NEON/kernels/
DNEInstanceNormalizationLayerKernel.cpp131 : _input(nullptr), _output(nullptr), _gamma(1), _beta(0), _epsilon(1e-12) in NEInstanceNormalizationLayerKernel()
143 _epsilon = info.epsilon; in configure()
146 …MPUTE_ERROR_THROW_ON(validate_arguments(_input->info(), _output->info(), _gamma, _beta, _epsilon)); in configure()
171 uk->ukernel(_input, _output, _gamma, _beta, _epsilon, _use_mixed_precision, window); in run()
DNEMeanStdDevNormalizationKernel.cpp132 : _input(nullptr), _output(nullptr), _epsilon(1e-8f) in NEMeanStdDevNormalizationKernel()
144 _epsilon = epsilon; in configure()
168 uk->ukernel(_input, _output, _epsilon, window); in run()
DNEL2NormalizeLayerKernel.cpp150 : _input(nullptr), _sum(nullptr), _output(nullptr), _actual_axis(0), _epsilon(1e-12) in NEL2NormalizeLayerKernel()
163 _epsilon = epsilon; in configure()
195 uk->ukernel(_input, _sum, _output, _epsilon, window, _actual_axis); in run()
DNEBatchNormalizationLayerKernel.cpp188 const auto epsilon_vec = wrapper::vdup_n(static_cast<T>(_epsilon), ExactTagType{}); in batch_normalization_nchw()
310 …t(nullptr), _mean(nullptr), _var(nullptr), _gamma(nullptr), _beta(nullptr), _epsilon(), _act_info() in NEBatchNormalizationLayerKernel()
333 _epsilon = epsilon; in configure()
392 uk->ukernel(_input, _output, _mean, _var, _beta, _gamma, _epsilon, _act_info, window); in run()
DNEFuseBatchNormalizationKernel.cpp215 …, _bn_gamma(nullptr), _bn_beta(nullptr), _fused_weights(nullptr), _fused_bias(nullptr), _epsilon(), in NEFuseBatchNormalizationKernel()
235 _epsilon = epsilon; in configure()
287 …input_bias, _fused_weights, _fused_bias, _bn_mean, _bn_var, _bn_beta, _bn_gamma, _epsilon, window); in run()
DNEL2NormalizeLayerKernel.h87 float _epsilon; variable
DNEMeanStdDevNormalizationKernel.h91 float _epsilon; variable
DNEInstanceNormalizationLayerKernel.h91 float _epsilon; variable
/external/ComputeLibrary/src/core/CL/kernels/
DCLL2NormalizeLayerKernel.cpp75 : _input(nullptr), _sum(nullptr), _output(nullptr), _actual_axis(0), _epsilon(1e-12) in CLL2NormalizeLayerKernel()
95 _epsilon = epsilon; in configure()
131 _kernel.setArg<cl_float>(idx, _epsilon); in configure()
135 _kernel.setArg<cl_half>(idx, _epsilon); in configure()
DCLBatchNormalizationLayerKernel.cpp110 …llptr), _mean(nullptr), _var(nullptr), _beta(nullptr), _gamma(nullptr), _epsilon(0), _run_in_place… in CLBatchNormalizationLayerKernel()
134 _epsilon = epsilon; in configure()
170 _kernel.setArg<cl_float>(idx++, _epsilon); in configure()
DCLL2NormalizeLayerKernel.h97 float _epsilon; variable
DCLFuseBatchNormalizationKernel.cpp103 …, _bn_gamma(nullptr), _bn_beta(nullptr), _fused_weights(nullptr), _fused_bias(nullptr), _epsilon(), in CLFuseBatchNormalizationKernel()
134 _epsilon = epsilon; in configure()
/external/ComputeLibrary/src/graph/nodes/
DBatchNormalizationLayerNode.cpp35 : _epsilon(epsilon), _fused_activation(fused_activation) in BatchNormalizationLayerNode()
43 return _epsilon; in epsilon()
DL2NormalizeLayerNode.cpp44 : _axis(axis), _epsilon(epsilon) in L2NormalizeLayerNode()
82 return _epsilon; in epsilon()
DFusedConvolutionBatchNormalizationWithPostOpsNode.cpp39 …: _epsilon(epsilon), _info(std::move(info)), _num_groups(num_groups), _method(method), _fast_math_… in FusedConvolutionBatchNormalizationWithPostOpsNode()
52 return _epsilon; in epsilon()
DFusedDepthwiseConvolutionBatchNormalizationNode.cpp40 …: _epsilon(epsilon), _info(std::move(info)), _depth_multiplier(depth_multiplier), _method(method),… in FusedDepthwiseConvolutionBatchNormalizationNode()
58 return _epsilon; in epsilon()
DFusedConvolutionBatchNormalizationNode.cpp40 …: _epsilon(epsilon), _info(std::move(info)), _num_groups(num_groups), _method(method), _fast_math_… in FusedConvolutionBatchNormalizationNode()
53 return _epsilon; in epsilon()
/external/ComputeLibrary/tests/validation/fixtures/
DFuseBatchNormalizationFixture.h97 …&var, w_fused_to_use, b_fused_to_use, b_to_use, beta_to_use, gamma_to_use, _epsilon, fuse_bn_type); in compute_target()
184 …ence::fuse_batch_normalization_dwc_layer(w, mean, var, w_fused, b_fused, b, beta, gamma, _epsilon); in compute_reference()
188 …nce::fuse_batch_normalization_conv_layer(w, mean, var, w_fused, b_fused, b, beta, gamma, _epsilon); in compute_reference()
197 const float _epsilon{ 0.0001f };
/external/tensorflow/tensorflow/python/training/
Dadadelta.py125 self._epsilon = epsilon
140 epsilon = self._call_if_callable(self._epsilon)
/external/ComputeLibrary/src/cpu/kernels/meanstddevnorm/generic/neon/
Dimpl.h33 void mean_stddev_normalization(ITensor *_input, ITensor *_output, float _epsilon, const Window &win…
/external/ComputeLibrary/arm_compute/graph/nodes/
DL2NormalizeLayerNode.h75 float _epsilon; variable
DBatchNormalizationLayerNode.h69 float _epsilon;
DFusedConvolutionBatchNormalizationWithPostOpsNode.h117 float _epsilon;
DFusedDepthwiseConvolutionBatchNormalizationNode.h126 float _epsilon;
DFusedConvolutionBatchNormalizationNode.h131 float _epsilon;

12