/external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
D | common.h | 57 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/ |
D | reference_ops.h | 162 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 …]
|
D | depthwiseconv_uint8.h | 37 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()
|
D | depthwiseconv_float.h | 31 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/ |
D | optimized_ops.h | 234 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 …]
|
D | cblas_conv.h | 39 float output_activation_max, float* output_data, in Conv() argument 86 output_activation_max); in Conv()
|
D | depthwiseconv_float.h | 916 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,
|
D | multithreaded_conv.h | 169 float output_activation_min, float output_activation_max, in Conv() argument 189 output_activation_max); in Conv()
|
D | depthwiseconv_uint8.h | 1679 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/ |
D | mul.cc | 85 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()
|
D | add.cc | 85 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()
|
D | conv.cc | 68 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()
|
D | depthwise_conv.cc | 61 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()
|
D | fully_connected.cc | 57 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()
|
D | div.cc | 65 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()
|
D | sub.cc | 65 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()
|