Home
last modified time | relevance | path

Searched refs:output_activation_min (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dcommon.h56 float* output_activation_min, in GetActivationMinMax() argument
60 *output_activation_min = std::numeric_limits<float>::lowest(); in GetActivationMinMax()
64 *output_activation_min = 0.f; in GetActivationMinMax()
68 *output_activation_min = -1.f; in GetActivationMinMax()
72 *output_activation_min = 0.f; in GetActivationMinMax()
78 inline float ActivationFunctionWithMinMax(float x, float output_activation_min, in ActivationFunctionWithMinMax() argument
80 return std::min(std::max(x, output_activation_min), output_activation_max); in ActivationFunctionWithMinMax()
86 float output_activation_min, output_activation_max; in ActivationFunction() local
87 GetActivationMinMax(Ac, &output_activation_min, &output_activation_max); in ActivationFunction()
88 return ActivationFunctionWithMinMax(x, output_activation_min, in ActivationFunction()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
Dreference_ops.h161 int pad_height, float output_activation_min, in Conv() argument
211 output_activation_min, in Conv()
227 float output_activation_min, output_activation_max; in Conv() local
228 GetActivationMinMax(Ac, &output_activation_min, &output_activation_max); in Conv()
231 output_activation_min, output_activation_max, output_data, output_dims, in Conv()
254 int output_shift, int32 output_activation_min, in Conv() argument
262 TFLITE_DCHECK_LE(output_activation_min, output_activation_max); in Conv()
306 acc = std::max(acc, output_activation_min); in Conv()
324 int output_shift, int32 output_activation_min, in Conv() argument
335 TFLITE_DCHECK_EQ(output_activation_min, 0); in Conv()
[all …]
Ddepthwiseconv_uint8.h36 int output_shift, int32 output_activation_min, in DepthwiseConv() argument
82 acc = std::max(acc, output_activation_min); in DepthwiseConv()
102 int32 output_activation_min, int32 output_activation_max, in DepthwiseConv() argument
105 TFLITE_DCHECK_EQ(output_activation_min, 0); in DepthwiseConv()
112 output_activation_min, output_activation_max, output_data, in DepthwiseConv()
124 int output_shift, int32 output_activation_min, in DepthwiseConv() argument
131 output_activation_min, output_activation_max, output_data, in DepthwiseConv()
Ddepthwiseconv_float.h30 float output_activation_min, in DepthwiseConv() argument
75 output_activation_min, in DepthwiseConv()
92 float output_activation_min, output_activation_max; in DepthwiseConv() local
93 GetActivationMinMax(Ac, &output_activation_min, &output_activation_max); in DepthwiseConv()
96 depth_multiplier, output_activation_min, output_activation_max, in DepthwiseConv()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h233 float output_activation_min, in AddBiasAndEvalActivationFunction() argument
242 const auto activation_min = vdupq_n_f32(output_activation_min); in AddBiasAndEvalActivationFunction()
282 output_activation_min, in AddBiasAndEvalActivationFunction()
295 array_data[array_offset + i] + bias_data[i], output_activation_min, in AddBiasAndEvalActivationFunction()
308 float output_activation_min, output_activation_max; in AddBiasAndEvalActivationFunction() local
309 GetActivationMinMax(Ac, &output_activation_min, &output_activation_max); in AddBiasAndEvalActivationFunction()
311 output_activation_min, in AddBiasAndEvalActivationFunction()
331 float output_activation_min, in FullyConnected() argument
353 output_dims, output_activation_min, in FullyConnected()
363 float output_activation_min, output_activation_max; in FullyConnected() local
[all …]
Dcblas_conv.h38 int pad_height, float output_activation_min, in Conv() argument
85 bias_data, bias_dims, output_data, output_dims, output_activation_min, in Conv()
Ddepthwiseconv_float.h915 float output_activation_min,
1038 vdupq_n_f32(output_activation_min),
1050 acc = vmaxq_f32(vdupq_n_f32(output_activation_min),
1060 acc = std::max(output_activation_min,
1078 float output_activation_min, output_activation_max;
1079 GetActivationMinMax(Ac, &output_activation_min, &output_activation_max);
1082 depth_multiplier, output_activation_min, output_activation_max,
Dmultithreaded_conv.h169 float output_activation_min, float output_activation_max, in Conv() argument
188 bias_data, bias_dims, output_data, output_dims, output_activation_min, in Conv()
Ddepthwiseconv_uint8.h1678 int output_shift, int32 output_activation_min,
1682 TFLITE_DCHECK_LE(output_activation_min, output_activation_max);
1806 vdupq_n_s32(output_activation_min);
1912 acc = std::max(acc, output_activation_min);
1930 int32 output_activation_min, int32 output_activation_max,
1933 TFLITE_DCHECK_EQ(output_activation_min, 0);
1940 output_activation_min, output_activation_max, output_data,
1952 int output_shift, int32 output_activation_min,
1959 output_activation_min, output_activation_max, output_data,
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dmul.cc85 float output_activation_min, output_activation_max; in EvalFloat() local
86 CalculateActivationRangeFloat(params->activation, &output_activation_min, in EvalFloat()
91 output_activation_min, output_activation_max, \ in EvalFloat()
126 int32 output_activation_min, output_activation_max; in EvalQuantized() local
128 &output_activation_min, &output_activation_max); in EvalQuantized()
134 output_multiplier, output_shift, output_activation_min, \ in EvalQuantized()
Dadd.cc85 float output_activation_min, output_activation_max; in EvalAddFloat() local
86 CalculateActivationRangeFloat(params->activation, &output_activation_min, in EvalAddFloat()
91 output_activation_min, output_activation_max, \ in EvalAddFloat()
140 int32 output_activation_min, output_activation_max; in EvalAddQuantized() local
142 &output_activation_min, &output_activation_max); in EvalAddQuantized()
150 output_activation_min, output_activation_max, \ in EvalAddQuantized()
Dconv.cc67 int32_t output_activation_min; member
189 &data->output_activation_min, in Prepare()
310 data->output_shift, data->output_activation_min, in EvalQuantized()
325 data->output_shift, data->output_activation_min, in EvalQuantized()
338 float output_activation_min, output_activation_max; in EvalFloat() local
339 CalculateActivationRangeFloat(params->activation, &output_activation_min, in EvalFloat()
349 output_activation_min, output_activation_max, in EvalFloat()
360 output_activation_min, output_activation_max, in EvalFloat()
377 output_activation_min, output_activation_max, in EvalFloat()
388 output_activation_min, output_activation_max, in EvalFloat()
Ddepthwise_conv.cc60 int32_t output_activation_min; member
157 &data->output_activation_min, in Prepare()
174 float output_activation_min, output_activation_max; in EvalFloat() local
175 CalculateActivationRangeFloat(params->activation, &output_activation_min, in EvalFloat()
192 params->depth_multiplier, output_activation_min, output_activation_max, in EvalFloat()
221 data->output_shift, data->output_activation_min, in EvalQuantized()
Dfully_connected.cc56 int32_t output_activation_min; member
120 &data->output_activation_min, in Prepare()
193 data->output_activation_min, data->output_activation_max, \ in EvalQuantized()
214 float output_activation_min, output_activation_max; in EvalFloat() local
215 CalculateActivationRangeFloat(params->activation, &output_activation_min, in EvalFloat()
221 output_activation_min, output_activation_max, \ in EvalFloat()
Ddiv.cc65 float output_activation_min, output_activation_max; in EvalDivFloat() local
66 CalculateActivationRangeFloat(params->activation, &output_activation_min, in EvalDivFloat()
71 output_activation_min, output_activation_max, \ in EvalDivFloat()
Dsub.cc65 float output_activation_min, output_activation_max; in EvalSubFloat() local
66 CalculateActivationRangeFloat(params->activation, &output_activation_min, in EvalSubFloat()
71 output_activation_min, output_activation_max, \ in EvalSubFloat()