| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add.cc | 58 int input2_shift; member
190 real_input2_multiplier, &data->input2_multiplier, &data->input2_shift); in Prepare()
216 data->input2_shift = input2_scale_log2_rounded - output_scale_log2_rounded; in Prepare()
220 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in Prepare()
222 TF_LITE_ENSURE(context, data->input2_shift <= 0); in Prepare()
296 op_params.input2_shift = data->input2_shift; in EvalAddQuantized()
351 op_params.input2_shift = data->input2_shift; in EvalAddQuantized()
|
| D | sub.cc | 61 int input2_shift; member
153 &op_params->input2_shift); in PrepareGeneralSubOp()
199 data->input2_shift = input2_scale_log2_rounded - output_scale_log2_rounded; in PrepareInt16SubOpPOT()
203 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in PrepareInt16SubOpPOT()
205 TF_LITE_ENSURE(context, data->input2_shift <= 0); in PrepareInt16SubOpPOT()
379 op_params.input2_shift = data->input2_shift; in EvalQuantized()
|
| D | comparisons.cc | 97 int input2_shift; in ComparisonQuantized() local
99 &input2_multiplier, &input2_shift); in ComparisonQuantized()
108 op_params.input2_shift = input2_shift; in ComparisonQuantized()
|
| D | squared_difference.cc | 122 &data->arithmetic_params.input2_shift); in Prepare()
156 shifted_input2_val, params.input2_multiplier, params.input2_shift); in SquaredDifference()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | add.h | 77 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
111 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddScalarBroadcast()
181 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0);
183 TFLITE_DCHECK_LE(params.input2_shift, 0);
188 input1_shift == 0 ? -params.input2_shift : -input1_shift;
333 params.input2_shift); in BroadcastAdd4DSlow()
360 switched_params.input1_shift = unswitched_params.input2_shift; in BroadcastAddFivefold()
363 switched_params.input2_shift = unswitched_params.input1_shift; in BroadcastAddFivefold()
|
| D | comparisons.h | 90 int input2_shift = op_params.input2_shift; in ComparisonWithScaling() local
104 shifted_input2_val, input2_multiplier, input2_shift); in ComparisonWithScaling()
184 int input2_shift = op_params.input2_shift; in BroadcastComparison4DSlowWithScaling() local
203 shifted_input2_val, input2_multiplier, input2_shift); in BroadcastComparison4DSlowWithScaling()
|
| D | legacy_reference_ops.h | 995 int input2_shift, bool* output_data, in Comparison() argument
1006 op_params.input2_shift = kReverseShift * input2_shift; in Comparison()
1033 int32 input2_multiplier, int input2_shift, in BroadcastComparison() argument
1045 op_params.input2_shift = kReverseShift * input2_shift; in BroadcastComparison()
1067 int32 input2_multiplier, int input2_shift, bool* output_data, \
1073 input2_multiplier, input2_shift, output_data, \
1090 int32 input2_multiplier, int input2_shift, bool* output_data, \
1097 input2_shift, output_data, output_dims); \
1260 int32 input2_offset, int32 input2_multiplier, int input2_shift, in Add() argument
1283 op_params.input2_shift = kReverseShift * input2_shift; in Add()
[all …]
|
| D | sub.h | 150 shifted_input2_val, params.input2_multiplier, params.input2_shift); in BroadcastSubSlow()
244 shifted_input2_val, params.input2_multiplier, params.input2_shift); in BroadcastSubSlow()
368 gemmlowp::RoundingDivideByPOT(input2_val, -params.input2_shift); in BroadcastSub16POTSlow()
399 shifted_input2_val, params.input2_multiplier, params.input2_shift); in SubElementwise()
433 shifted_input2_val, params.input2_multiplier, params.input2_shift); in SubElementwise()
|
| D | reference_ops.h | 279 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Sub16()
281 TFLITE_DCHECK_LE(params.input2_shift, 0); in Sub16()
285 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Sub16()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | add.cc | 44 int input2_shift; member
86 real_input2_multiplier, &data->input2_multiplier, &data->input2_shift); in CalculateOpData()
135 op_params.input2_shift = data->input2_shift; in EvalAddQuantized()
160 op_params.input2_multiplier, op_params.input2_shift, in EvalAddQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | sub.cc | 44 int input2_shift; member
85 real_input2_multiplier, &data->input2_multiplier, &data->input2_shift); in CalculateOpData()
162 op_params.input2_shift = data->input2_shift; in EvalSubQuantized()
|
| D | add.cc | 44 int input2_shift; member
90 real_input2_multiplier, &data->input2_multiplier, &data->input2_shift); in CalculateOpData()
144 op_params.input2_shift = data->input2_shift; in EvalAddQuantized()
|
| D | comparisons.cc | 637 int input2_shift; in Prepare() local
640 &input2_shift); in Prepare()
648 data->params.input2_shift = input2_shift; in Prepare()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | add.h | 52 const int input2_left_shift = params.left_shift + params.input2_shift; in AddElementwise()
153 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
219 const int32x4_t input2_shift_dup = vdupq_n_s32(params.input2_shift); in AddScalarBroadcast()
253 params.input2_shift); in AddScalarBroadcast()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | add.h | 96 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddFunc()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 66 int input2_shift; member
959 int input2_shift;
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 805 int input2_shift; member
847 int input2_shift; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3512 int32 input2_offset, int32 input2_multiplier, int input2_shift, in Add() argument
3535 op_params.input2_shift = kReverseShift * input2_shift; in Add()
3580 int input2_shift, int32 output_offset, in BroadcastAdd() argument
3604 op_params.input2_shift = kReverseShift * input2_shift; in BroadcastAdd()
3621 int input2_shift, int32 output_offset, int32 output_multiplier, in BroadcastAddFivefold() argument
3644 op_params.input2_shift = kReverseShift * input2_shift; in BroadcastAddFivefold()
3676 const Dims<4>& input2_dims, int input2_shift, in Add() argument
3693 op_params.input2_shift = kReverseShift * input2_shift; in Add()
|
| D | optimized_ops.h | 216 switched_params.input1_shift = unswitched_params.input2_shift; in BinaryBroadcastFiveFold()
219 switched_params.input2_shift = unswitched_params.input1_shift; in BinaryBroadcastFiveFold()
1903 const int32x4_t input2_shift_dup = vdupq_n_s32(params.input2_shift); in AddElementwise()
1936 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
2004 const int32x4_t input2_shift_dup = vdupq_n_s32(params.input2_shift); in AddScalarBroadcast()
2038 params.input2_shift); in AddScalarBroadcast()
2116 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
2118 TFLITE_DCHECK_LE(params.input2_shift, 0); in Add()
2122 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Add()
|