/external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
D | common.h | 56 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/ |
D | reference_ops.h | 161 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 …]
|
D | depthwiseconv_uint8.h | 36 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()
|
D | depthwiseconv_float.h | 30 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/ |
D | optimized_ops.h | 233 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 …]
|
D | cblas_conv.h | 38 int pad_height, float output_activation_min, in Conv() argument 85 bias_data, bias_dims, output_data, output_dims, output_activation_min, in Conv()
|
D | depthwiseconv_float.h | 915 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,
|
D | multithreaded_conv.h | 169 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()
|
D | depthwiseconv_uint8.h | 1678 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/ |
D | mul.cc | 85 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()
|
D | add.cc | 85 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()
|
D | conv.cc | 67 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()
|
D | depthwise_conv.cc | 60 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()
|
D | fully_connected.cc | 56 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()
|
D | div.cc | 65 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()
|
D | sub.cc | 65 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()
|