Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dcommon.h57 float* output_activation_max) { in GetActivationMinMax() argument
61 *output_activation_max = std::numeric_limits<float>::max(); in GetActivationMinMax()
65 *output_activation_max = std::numeric_limits<float>::max(); in GetActivationMinMax()
69 *output_activation_max = 1.f; in GetActivationMinMax()
73 *output_activation_max = 6.f; in GetActivationMinMax()
79 float output_activation_max) { 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()
89 output_activation_max); in ActivationFunction()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
Dreference_ops.h162 float output_activation_max, float* output_data, in Conv() argument
212 output_activation_max); 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()
255 int32 output_activation_max, uint8* output_data, in Conv() argument
262 TFLITE_DCHECK_LE(output_activation_min, output_activation_max); in Conv()
307 acc = std::min(acc, output_activation_max); in Conv()
325 int32 output_activation_max, uint8* output_data, in Conv() argument
336 TFLITE_DCHECK_EQ(output_activation_max, 255); in Conv()
[all …]
Ddepthwiseconv_uint8.h37 int32 output_activation_max, uint8* output_data, in DepthwiseConv() argument
83 acc = std::min(acc, output_activation_max); in DepthwiseConv()
102 int32 output_activation_min, int32 output_activation_max, in DepthwiseConv() argument
106 TFLITE_DCHECK_EQ(output_activation_max, 255); in DepthwiseConv()
112 output_activation_min, output_activation_max, output_data, in DepthwiseConv()
125 int32 output_activation_max, uint8* output_data, in DepthwiseConv() argument
131 output_activation_min, output_activation_max, output_data, in DepthwiseConv()
Ddepthwiseconv_float.h31 float output_activation_max, float* output_data, in DepthwiseConv() argument
76 output_activation_max); 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.h234 float output_activation_max) { in AddBiasAndEvalActivationFunction() argument
243 const auto activation_max = vdupq_n_f32(output_activation_max); in AddBiasAndEvalActivationFunction()
283 output_activation_max); in AddBiasAndEvalActivationFunction()
296 output_activation_max); in AddBiasAndEvalActivationFunction()
308 float output_activation_min, output_activation_max; in AddBiasAndEvalActivationFunction() local
309 GetActivationMinMax(Ac, &output_activation_min, &output_activation_max); in AddBiasAndEvalActivationFunction()
312 output_activation_max); in AddBiasAndEvalActivationFunction()
332 float output_activation_max, float* output_data, in FullyConnected() argument
354 output_activation_max); in FullyConnected()
363 float output_activation_min, output_activation_max; in FullyConnected() local
[all …]
Dcblas_conv.h39 float output_activation_max, float* output_data, in Conv() argument
86 output_activation_max); in Conv()
Ddepthwiseconv_float.h916 float output_activation_max, float* output_data,
1039 vminq_f32(vdupq_n_f32(output_activation_max), acc[k]));
1051 vminq_f32(vdupq_n_f32(output_activation_max), acc));
1061 std::min(output_activation_max, acc));
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
189 output_activation_max); in Conv()
Ddepthwiseconv_uint8.h1679 int32 output_activation_max, uint8* output_data,
1682 TFLITE_DCHECK_LE(output_activation_min, output_activation_max);
1808 vdupq_n_s32(output_activation_max);
1913 acc = std::min(acc, output_activation_max);
1930 int32 output_activation_min, int32 output_activation_max,
1934 TFLITE_DCHECK_EQ(output_activation_max, 255);
1940 output_activation_min, output_activation_max, output_data,
1953 int32 output_activation_max, uint8* output_data,
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
87 &output_activation_max); 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()
135 output_activation_max, GetTensorData<uint8_t>(output), \ in EvalQuantized()
Dadd.cc85 float output_activation_min, output_activation_max; in EvalAddFloat() local
87 &output_activation_max); 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.cc68 int32_t output_activation_max; member
190 &data->output_activation_max); in Prepare()
311 data->output_activation_max, GetTensorData<uint8_t>(output), in EvalQuantized()
326 data->output_activation_max, GetTensorData<uint8_t>(output), in EvalQuantized()
338 float output_activation_min, output_activation_max; in EvalFloat() local
340 &output_activation_max); 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.cc61 int32_t output_activation_max; member
158 &data->output_activation_max); in Prepare()
174 float output_activation_min, output_activation_max; in EvalFloat() local
176 &output_activation_max); in EvalFloat()
192 params->depth_multiplier, output_activation_min, output_activation_max, in EvalFloat()
222 data->output_activation_max, GetTensorData<uint8_t>(output), in EvalQuantized()
Dfully_connected.cc57 int32_t output_activation_max; member
121 &data->output_activation_max); in Prepare()
193 data->output_activation_min, data->output_activation_max, \ in EvalQuantized()
214 float output_activation_min, output_activation_max; in EvalFloat() local
216 &output_activation_max); in EvalFloat()
221 output_activation_min, output_activation_max, \ in EvalFloat()
Ddiv.cc65 float output_activation_min, output_activation_max; in EvalDivFloat() local
67 &output_activation_max); in EvalDivFloat()
71 output_activation_min, output_activation_max, \ in EvalDivFloat()
Dsub.cc65 float output_activation_min, output_activation_max; in EvalSubFloat() local
67 &output_activation_max); in EvalSubFloat()
71 output_activation_min, output_activation_max, \ in EvalSubFloat()