Searched refs:activation_max (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
128 &activation_max); in AverageEvalFloat()
138 op_params.float_activation_max = activation_max; \ in AverageEvalFloat()
156 int32_t activation_max; in AverageEvalQuantizedUint8() local
158 &activation_max); in AverageEvalQuantizedUint8()
168 op_params.quantized_activation_max = activation_max; \ in AverageEvalQuantizedUint8()
184 int32_t activation_max; in AverageEvalQuantizedInt8() local
186 &activation_max); in AverageEvalQuantizedInt8()
195 op_params.quantized_activation_max = activation_max; 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
129 *activation_max = std::numeric_limits<T>::max(); in CalculateActivationRange()
132 *activation_max = 6; in CalculateActivationRange()
135 *activation_max = 1; in CalculateActivationRange()
138 *activation_max = std::numeric_limits<T>::max(); in CalculateActivationRange()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 201 const auto activation_max = vdupq_n_f32(output_activation_max); in AddBiasAndEvalActivationFunction() local
221 x0 = vminq_f32(activation_max, x0); in AddBiasAndEvalActivationFunction()
222 x1 = vminq_f32(activation_max, x1); in AddBiasAndEvalActivationFunction()
223 x2 = vminq_f32(activation_max, x2); in AddBiasAndEvalActivationFunction()
224 x3 = vminq_f32(activation_max, x3); in AddBiasAndEvalActivationFunction()
235 x = vminq_f32(activation_max, x); in AddBiasAndEvalActivationFunction()
2633 const auto activation_max = vdupq_n_f32(params.float_activation_max); in Add() local
2651 x0 = vminq_f32(activation_max, x0); in Add()
2652 x1 = vminq_f32(activation_max, x1); in Add()
2653 x2 = vminq_f32(activation_max, x2); in Add()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 296 const auto activation_max = vdupq_n_f32(output_activation_max); in AddBiasAndEvalActivationFunction() local
316 x0 = vminq_f32(activation_max, x0); in AddBiasAndEvalActivationFunction()
317 x1 = vminq_f32(activation_max, x1); in AddBiasAndEvalActivationFunction()
318 x2 = vminq_f32(activation_max, x2); in AddBiasAndEvalActivationFunction()
319 x3 = vminq_f32(activation_max, x3); in AddBiasAndEvalActivationFunction()
330 x = vminq_f32(activation_max, x); in AddBiasAndEvalActivationFunction()
2850 const auto activation_max = vdupq_n_f32(params.float_activation_max); in Add() local
2868 x0 = vminq_f32(activation_max, x0); in Add()
2869 x1 = vminq_f32(activation_max, x1); in Add()
2870 x2 = vminq_f32(activation_max, x2); in Add()
[all …]
|