| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | tanh.h | 27 int32_t input_multiplier, int32_t input_shift, in Tanh() argument
48 MultiplyByQuantizedMultiplier(input, input_multiplier, input_shift); in Tanh()
62 inline void Tanh(int32_t input_multiplier, int32_t input_left_shift, in Tanh() argument
73 if (input_multiplier == 0) { // power of two case in Tanh()
74 input_multiplier = 3 << input_left_shift; in Tanh()
84 ((*ptr_input_data) * input_multiplier + round) >> input_left_shift; in Tanh()
|
| D | logistic.h | 25 int32_t input_multiplier, int32_t input_left_shift, in Logistic() argument
44 input, input_multiplier, input_left_shift); in Logistic()
61 inline void Logistic(int32_t input_multiplier, int32_t input_left_shift, in Logistic() argument
73 if (input_multiplier == 0) { // power of two case in Logistic()
74 input_multiplier = 3 << input_left_shift; in Logistic()
82 ((*ptr_input_data) * input_multiplier + round) >> input_left_shift; in Logistic()
|
| D | log_softmax.h | 23 inline void LogSoftmax(int32_t input_multiplier, int32_t input_shift, in LogSoftmax() argument
58 input_diff, input_multiplier, input_shift); in LogSoftmax()
87 input_diff, input_multiplier, input_shift); in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | tanh.cc | 40 int32_t input_multiplier; member
67 data->input_multiplier = static_cast<int32_t>(TfLiteRound(q * (1ll << 31))); in CalculateArithmeticOpData()
118 params.input_multiplier = data.input_multiplier; in TanhEval()
129 data.input_zero_point, data.input_range_radius, data.input_multiplier, in TanhEval()
|
| D | logistic.cc | 39 int32_t input_multiplier; member
63 data->input_multiplier = static_cast<int32_t>(TfLiteRound(q * (1ll << 31))); in CalculateArithmeticOpData()
113 data->input_multiplier, data->input_left_shift, in LogisticEval()
|
| D | softmax_common.cc | 67 QuantizeMultiplier(input_scale_beta_rescale, &op_data->input_multiplier, in CalculateSoftmaxParams()
75 &op_data->input_multiplier, &input_left_shift); in CalculateSoftmaxParams()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | tanh.h | 88 const int32_t input_multiplier = params.input_multiplier; in Tanh() local
105 input_val_centered, input_multiplier, input_left_shift); in Tanh()
|
| D | legacy_reference_ops.h | 761 params.input_multiplier = input_beta_multiplier; in Softmax()
775 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
780 params.input_multiplier = input_multiplier; in LogSoftmax()
793 const int32 input_multiplier = params.input_multiplier; in Logistic() local
809 input_val_centered, input_multiplier, input_left_shift); in Logistic()
831 int32 input_multiplier, int input_left_shift, in Logistic() argument
836 params.input_multiplier = input_multiplier; in Logistic()
850 int32 input_multiplier, int input_left_shift, in Tanh() argument
855 params.input_multiplier = input_multiplier; in Tanh()
1791 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
[all …]
|
| D | softmax.h | 70 const int32_t input_beta_multiplier = params.input_multiplier; in Softmax()
155 input_diff, params.input_multiplier, params.input_left_shift); in SoftMaxCalculateExp()
|
| D | reference_ops.h | 941 const int32 input_multiplier = params.input_multiplier; in LogSoftmax() local
977 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
1009 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 64 int32_t input_multiplier = 0; member
406 data->input_multiplier = static_cast<int16_t>(q_fixed); in TanhPrepare()
464 data->input_multiplier = static_cast<int32_t>(multiplier); in TanhPrepare()
511 data->input_multiplier = static_cast<int16_t>(q_fixed); in SigmoidPrepare()
567 data->input_multiplier = static_cast<int32_t>(multiplier); in SigmoidPrepare()
645 QuantizeMultiplier(input_scale_beta_rescale, &data->params.input_multiplier, in SoftmaxPrepare()
921 if (kernel_type == kReference || (data->input_multiplier > 0)) { in TanhEval()
923 data->input_multiplier, data->input_left_shift, in TanhEval()
938 params.input_multiplier = data->input_multiplier; in TanhEval()
953 params.input_multiplier = data->input_multiplier; in TanhEval()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3993 const int32 input_beta_multiplier = params.input_multiplier; in Softmax()
4212 params.input_multiplier = input_beta_multiplier; in Softmax()
4240 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
4245 params.input_multiplier = input_multiplier; in LogSoftmax()
4255 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
4260 input_data, DimsToShape(input_dims), input_multiplier, input_left_shift, in LogSoftmax()
4271 const int32 input_multiplier = params.input_multiplier; in Logistic() local
4326 vqrdmulhq_n_s32(input_val_rescaled_0, input_multiplier); in Logistic()
4328 vqrdmulhq_n_s32(input_val_rescaled_1, input_multiplier); in Logistic()
4330 vqrdmulhq_n_s32(input_val_rescaled_2, input_multiplier); in Logistic()
[all …]
|
| D | optimized_ops.h | 7057 int16x8_t input_val_3, int input_left_shift, int input_multiplier) { in SaturatingRounding() argument
7068 result.val[0] = vqrdmulhq_n_s16(input_val_shifted_0, input_multiplier); in SaturatingRounding()
7069 result.val[1] = vqrdmulhq_n_s16(input_val_shifted_1, input_multiplier); in SaturatingRounding()
7070 result.val[2] = vqrdmulhq_n_s16(input_val_shifted_2, input_multiplier); in SaturatingRounding()
7071 result.val[3] = vqrdmulhq_n_s16(input_val_shifted_3, input_multiplier); in SaturatingRounding()
7205 const int16 input_multiplier = static_cast<int16>(params.input_multiplier); in Tanh16bitPrecision() local
7245 input_left_shift, input_multiplier); in Tanh16bitPrecision()
7284 static_cast<int16>(input_multiplier)); in Tanh16bitPrecision()
7312 const int16 input_multiplier = static_cast<int16>(params.input_multiplier); in Tanh16bitPrecision() local
7349 input_left_shift, input_multiplier); in Tanh16bitPrecision()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | softmax.cc | 56 depth, op_data.input_multiplier, op_data.input_left_shift, in SoftmaxQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 91 int32_t input_multiplier; member
1093 int32_t input_multiplier;
1182 int32_t input_multiplier;
1236 int32_t input_multiplier;
|
| D | ceva_tflm_lib.h | 305 int32_t input_multiplier, int32_t input_shift,
311 int32_t input_multiplier, int32_t input_left_shift,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | softmax_quantized_test.cc | 130 params.input_multiplier = input_beta_multiplier; in RunOneSoftmaxTest()
|
| D | types.h | 994 int32_t input_multiplier; member
1083 int32_t input_multiplier; member
1139 int32_t input_multiplier; member
|
| D | logsoftmax_quantized_test.cc | 175 params.input_multiplier = input_beta_multiplier; in RunOneLogSoftmaxTest()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | softmax.cc | 231 op_data->params.input_multiplier, depth, p_scratch); in EvalHifi4()
|