| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 47 int input_left_shift = 0; member
133 &data->input_left_shift); in TanhPrepare()
135 CalculateInputRadius(kInputIntegerBits, data->input_left_shift); in TanhPrepare()
162 data->input_left_shift = in TanhPrepare()
166 TF_LITE_ENSURE(context, data->input_left_shift >= 0); in TanhPrepare()
167 TF_LITE_ENSURE(context, data->input_left_shift <= 1); in TanhPrepare()
202 &data->input_left_shift); in SigmoidPrepare()
204 CalculateInputRadius(kInputIntegerBits, data->input_left_shift); in SigmoidPrepare()
225 data->input_left_shift = in SigmoidPrepare()
228 TF_LITE_ENSURE_EQ(context, data->input_left_shift, 0); in SigmoidPrepare()
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | softmax.cc | 33 int input_left_shift = 0; member
51 &data->input_multiplier, &data->input_left_shift); in CalculateSoftmaxOpData()
53 kScaledDiffIntegerBits, data->input_left_shift); in CalculateSoftmaxOpData()
98 op_params.input_left_shift = data->input_left_shift; in Softmax1DQuantized()
117 op_params.input_left_shift = data->input_left_shift; in Softmax2DQuantized()
138 op_params.input_left_shift = data->input_left_shift; in Softmax4DQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| 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()
|
| D | softmax.h | 28 const int32 input_beta_left_shift = params.input_left_shift; in Softmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | quantization_util.cc | 302 int CalculateInputRadius(int input_integer_bits, int input_left_shift) { in CalculateInputRadius() argument
306 result >>= input_left_shift; in CalculateInputRadius()
311 (1ll << input_left_shift); in CalculateInputRadius()
|
| D | quantization_util.h | 255 int CalculateInputRadius(int input_integer_bits, int input_left_shift);
|
| D | types.h | 875 int input_left_shift; member
957 int32 input_left_shift; member
1004 int input_left_shift; member
|
| D | softmax_quantized_test.cc | 125 params.input_left_shift = input_beta_left_shift; in RunOneSoftmaxTest()
|
| D | logsoftmax_quantized_test.cc | 175 params.input_left_shift = input_beta_left_shift; in RunOneLogSoftmaxTest()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1448 params.input_left_shift = input_beta_left_shift; in Softmax()
1475 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
1481 params.input_left_shift = input_left_shift; in LogSoftmax()
1489 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
1494 input_left_shift, reverse_scaling_divisor, in LogSoftmax()
1501 int32 input_multiplier, int input_left_shift, in Logistic() argument
1507 params.input_left_shift = input_left_shift; in Logistic()
1519 int32 input_multiplier, int input_left_shift, in Logistic() argument
1522 input_range_radius, input_multiplier, input_left_shift, output_data, in Logistic()
1554 int32 input_multiplier, int input_left_shift, in Tanh() argument
[all …]
|
| D | optimized_ops.h | 4455 const int32 input_beta_left_shift = params.input_left_shift; in Softmax()
4705 const int32 input_left_shift = params.input_left_shift; in LogSoftmax() local
4741 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
4772 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
4814 const int input_left_shift = params.input_left_shift; in Logistic() local
4857 vdupq_n_s32(input_left_shift)); in Logistic()
4860 vdupq_n_s32(input_left_shift)); in Logistic()
4863 vdupq_n_s32(input_left_shift)); in Logistic()
4866 vdupq_n_s32(input_left_shift)); in Logistic()
4929 input_val_centered, input_multiplier, input_left_shift); in Logistic()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 712 params.input_left_shift = input_beta_left_shift; in Softmax()
725 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
731 params.input_left_shift = input_left_shift; in LogSoftmax()
740 int32 input_multiplier, int input_left_shift, in Logistic() argument
746 params.input_left_shift = input_left_shift; in Logistic()
759 int32 input_multiplier, int input_left_shift, in Tanh() argument
765 params.input_left_shift = input_left_shift; in Tanh()
770 int input_left_shift, int16* output_data, in Tanh() argument
773 params.input_left_shift = input_left_shift; in Tanh()
1700 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
[all …]
|
| D | softmax.h | 64 const int32 input_beta_left_shift = params.input_left_shift; in Softmax()
|
| D | reference_ops.h | 2604 const int32 input_left_shift = params.input_left_shift; in LogSoftmax() local
2639 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
2671 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
2714 const int input_left_shift = params.input_left_shift; in Logistic() local
2729 input_val_centered, input_multiplier, input_left_shift); in Logistic()
2794 const int input_left_shift = params.input_left_shift; in Tanh() local
2810 input_val_centered, input_multiplier, input_left_shift); in Tanh()
2834 const int input_left_shift = params.input_left_shift; in Tanh() local
2837 TFLITE_DCHECK_GE(input_left_shift, 0); in Tanh()
2838 TFLITE_DCHECK_LE(input_left_shift, 1); in Tanh()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4580 const int32 input_beta_left_shift = params.input_left_shift; in Softmax()
4795 params.input_left_shift = input_beta_left_shift; in Softmax()
4965 const int32 input_left_shift = params.input_left_shift; in LogSoftmax() local
5002 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
5033 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
5053 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
5059 params.input_left_shift = input_left_shift; in LogSoftmax()
5091 const int input_left_shift = params.input_left_shift; in Logistic() local
5134 vdupq_n_s32(input_left_shift)); in Logistic()
5137 vdupq_n_s32(input_left_shift)); in Logistic()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 852 int input_left_shift; member
912 int32 input_left_shift; member
959 int input_left_shift; member
|