Searched refs:activation_min (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | pooling.cc | 126 float activation_min, activation_max; in AverageEvalFloat() local 127 CalculateActivationRange(params->activation, &activation_min, in AverageEvalFloat() 137 op_params.float_activation_min = activation_min; \ in AverageEvalFloat() 155 int32_t activation_min; in AverageEvalQuantizedUint8() local 157 CalculateActivationRangeUint8(params->activation, output, &activation_min, in AverageEvalQuantizedUint8() 167 op_params.quantized_activation_min = activation_min; \ in AverageEvalQuantizedUint8() 183 int32_t activation_min; in AverageEvalQuantizedInt8() local 185 CalculateActivationRangeInt8(params->activation, output, &activation_min, in AverageEvalQuantizedInt8() 194 op_params.quantized_activation_min = activation_min; in AverageEvalQuantizedInt8() 205 float activation_min, activation_max; in MaxEvalFloat() local [all …]
|
D | kernel_util.h | 126 T* activation_min, T* activation_max) { in CalculateActivationRange() argument 128 *activation_min = 0; in CalculateActivationRange() 131 *activation_min = 0; in CalculateActivationRange() 134 *activation_min = -1; in CalculateActivationRange() 137 *activation_min = std::numeric_limits<T>::lowest(); in CalculateActivationRange()
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | optimized_ops.h | 200 const auto activation_min = vdupq_n_f32(output_activation_min); in AddBiasAndEvalActivationFunction() local 217 x0 = vmaxq_f32(activation_min, x0); in AddBiasAndEvalActivationFunction() 218 x1 = vmaxq_f32(activation_min, x1); in AddBiasAndEvalActivationFunction() 219 x2 = vmaxq_f32(activation_min, x2); in AddBiasAndEvalActivationFunction() 220 x3 = vmaxq_f32(activation_min, x3); in AddBiasAndEvalActivationFunction() 234 x = vmaxq_f32(activation_min, x); in AddBiasAndEvalActivationFunction() 2632 const auto activation_min = vdupq_n_f32(params.float_activation_min); in Add() local 2647 x0 = vmaxq_f32(activation_min, x0); in Add() 2648 x1 = vmaxq_f32(activation_min, x1); in Add() 2649 x2 = vmaxq_f32(activation_min, x2); in Add() [all …]
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
D | optimized_ops.h | 295 const auto activation_min = vdupq_n_f32(output_activation_min); in AddBiasAndEvalActivationFunction() local 312 x0 = vmaxq_f32(activation_min, x0); in AddBiasAndEvalActivationFunction() 313 x1 = vmaxq_f32(activation_min, x1); in AddBiasAndEvalActivationFunction() 314 x2 = vmaxq_f32(activation_min, x2); in AddBiasAndEvalActivationFunction() 315 x3 = vmaxq_f32(activation_min, x3); in AddBiasAndEvalActivationFunction() 329 x = vmaxq_f32(activation_min, x); in AddBiasAndEvalActivationFunction() 2849 const auto activation_min = vdupq_n_f32(params.float_activation_min); in Add() local 2864 x0 = vmaxq_f32(activation_min, x0); in Add() 2865 x1 = vmaxq_f32(activation_min, x1); in Add() 2866 x2 = vmaxq_f32(activation_min, x2); in Add() [all …]
|