| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add.cc | 48 int input2_shift; member
116 real_input2_multiplier, &data->input2_multiplier, &data->input2_shift); in Prepare()
153 data->input2_shift = input2_scale_log2_rounded - output_scale_log2_rounded; in Prepare()
157 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in Prepare()
159 TF_LITE_ENSURE(context, data->input2_shift <= 0); in Prepare()
232 op_params.input2_shift = data->input2_shift; in EvalAddQuantized()
279 op_params.input2_shift = data->input2_shift; \ in EvalAddQuantized()
|
| D | sub.cc | 50 int input2_shift; member
129 &op_params->input2_shift); in Prepare8BitSubOp()
173 data->input2_shift = input2_scale_log2_rounded - output_scale_log2_rounded; in PrepareInt16SubOp()
177 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in PrepareInt16SubOp()
179 TF_LITE_ENSURE(context, data->input2_shift <= 0); in PrepareInt16SubOp()
281 op_params.input2_shift = data->input2_shift; in EvalQuantized()
|
| D | comparisons.cc | 70 int input2_shift; in ComparisonQuantized() local
72 &input2_multiplier, &input2_shift); in ComparisonQuantized()
81 op_params.input2_shift = input2_shift; in ComparisonQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | add.h | 72 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
106 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddScalarBroadcast()
148 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
150 TFLITE_DCHECK_LE(params.input2_shift, 0); in Add()
154 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Add()
306 params.input2_shift); in BroadcastAdd4DSlow()
333 switched_params.input1_shift = unswitched_params.input2_shift; in BroadcastAddFivefold()
336 switched_params.input2_shift = unswitched_params.input1_shift; in BroadcastAddFivefold()
|
| D | comparisons.h | 89 int input2_shift = op_params.input2_shift; in ComparisonWithScaling() local
103 shifted_input2_val, input2_multiplier, input2_shift); in ComparisonWithScaling()
173 int input2_shift = op_params.input2_shift; in BroadcastComparison4DSlowWithScaling() local
190 shifted_input2_val, input2_multiplier, input2_shift); in BroadcastComparison4DSlowWithScaling()
|
| D | legacy_reference_ops.h | 1025 int input2_shift, bool* output_data, in Comparison() argument
1036 op_params.input2_shift = kReverseShift * input2_shift; in Comparison()
1063 int32 input2_multiplier, int input2_shift, in BroadcastComparison() argument
1075 op_params.input2_shift = kReverseShift * input2_shift; in BroadcastComparison()
1097 int32 input2_multiplier, int input2_shift, bool* output_data, \
1103 input2_multiplier, input2_shift, output_data, \
1120 int32 input2_multiplier, int input2_shift, bool* output_data, \
1127 input2_shift, output_data, output_dims); \
1290 int32 input2_offset, int32 input2_multiplier, int input2_shift, in Add() argument
1313 op_params.input2_shift = kReverseShift * input2_shift; in Add()
[all …]
|
| D | reference_ops.h | 781 params.input2_shift); in BroadcastSub4DSlow()
960 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Sub16()
962 TFLITE_DCHECK_LE(params.input2_shift, 0); in Sub16()
966 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Sub16()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis-nn/ |
| D | add.cc | 42 int input2_shift; member
83 real_input2_multiplier, &data->input2_multiplier, &data->input2_shift); in CalculateOpData()
130 op_params.input2_shift = data->input2_shift; in EvalAddQuantized()
151 op_params.input2_multiplier, op_params.input2_shift, in EvalAddQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | add.cc | 42 int input2_shift; member
94 real_input2_multiplier, &data->input2_multiplier, &data->input2_shift); in CalculateOpData()
141 op_params.input2_shift = data->input2_shift; in EvalAddQuantized()
|
| D | comparisons.cc | 50 int input2_shift; \
53 &input2_shift); \
62 op_params.input2_shift = input2_shift; \
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | add.h | 70 const int32x4_t input2_shift_dup = vdupq_n_s32(params.input2_shift); in AddElementwise()
103 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
169 const int32x4_t input2_shift_dup = vdupq_n_s32(params.input2_shift); in AddScalarBroadcast()
203 params.input2_shift); in AddScalarBroadcast()
246 switched_params.input1_shift = unswitched_params.input2_shift; in BroadcastAddFivefold()
249 switched_params.input2_shift = unswitched_params.input1_shift; in BroadcastAddFivefold()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | add.h | 47 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
123 params.input2_shift); in BroadcastAdd4DSlow()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 767 int input2_shift; member
804 int input2_shift; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3499 int32 input2_offset, int32 input2_multiplier, int input2_shift, in Add() argument
3522 op_params.input2_shift = kReverseShift * input2_shift; in Add()
3567 int input2_shift, int32 output_offset, in BroadcastAdd() argument
3591 op_params.input2_shift = kReverseShift * input2_shift; in BroadcastAdd()
3608 int input2_shift, int32 output_offset, int32 output_multiplier, in BroadcastAddFivefold() argument
3631 op_params.input2_shift = kReverseShift * input2_shift; in BroadcastAddFivefold()
3663 const Dims<4>& input2_dims, int input2_shift, in Add() argument
3680 op_params.input2_shift = kReverseShift * input2_shift; in Add()
|
| D | optimized_ops.h | 1770 const int32x4_t input2_shift_dup = vdupq_n_s32(params.input2_shift); in AddElementwise()
1803 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
1871 const int32x4_t input2_shift_dup = vdupq_n_s32(params.input2_shift); in AddScalarBroadcast()
1905 params.input2_shift); in AddScalarBroadcast()
1983 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
1985 TFLITE_DCHECK_LE(params.input2_shift, 0); in Add()
1989 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Add()
2045 switched_params.input1_shift = unswitched_params.input2_shift; in BroadcastAddFivefold()
2048 switched_params.input2_shift = unswitched_params.input1_shift; in BroadcastAddFivefold()
|