| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | add.h | 63 TFLITE_DCHECK_GT(params.input2_offset, -std::numeric_limits<T>::max()); in AddElementwise()
65 TFLITE_DCHECK_LT(params.input2_offset, std::numeric_limits<T>::max()); in AddElementwise()
69 const int32_t input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
97 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddScalarBroadcast()
99 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddScalarBroadcast()
107 const int32_t input2_val = params.input2_offset + input2_data[i]; in AddScalarBroadcast()
134 TFLITE_DCHECK_GT(params.input2_offset, -256); in Add()
136 TFLITE_DCHECK_LT(params.input2_offset, 256); in Add()
155 TFLITE_DCHECK_GT(params.input2_offset, -max_value); in AddGeneralParamScale()
157 TFLITE_DCHECK_LT(params.input2_offset, max_value); in AddGeneralParamScale()
[all …]
|
| D | div.h | 33 TFLITE_DCHECK_GT(params.input2_offset, -256); in DivElementwise()
34 TFLITE_DCHECK_LT(params.input2_offset, 256); in DivElementwise()
40 const int32 input2_val = params.input2_offset + input2_data[i]; in DivElementwise()
95 TFLITE_DCHECK_GT(params.input2_offset, -256); in BroadcastDivSlow()
96 TFLITE_DCHECK_LT(params.input2_offset, 256); in BroadcastDivSlow()
104 params.input2_offset + input2_data[SubscriptToIndex(desc2, indexes)]; in BroadcastDivSlow()
|
| D | sub.h | 142 params.input2_offset + input2_data[SubscriptToIndex(desc2, indexes)]; in BroadcastSubSlow()
236 params.input2_offset + input2_data[SubscriptToIndex(desc2, indexes)]; in BroadcastSubSlow()
385 TFLITE_DCHECK_GT(params.input2_offset, -256); in SubElementwise()
387 TFLITE_DCHECK_LT(params.input2_offset, 256); in SubElementwise()
391 const int32_t input2_val = params.input2_offset + input2_data[i]; in SubElementwise()
419 TFLITE_DCHECK_GE(params.input2_offset, -1 * int8_max_value); in SubElementwise()
421 TFLITE_DCHECK_LE(params.input2_offset, int8_max_value); in SubElementwise()
425 const int32_t input2_val = params.input2_offset + input2_data[i]; in SubElementwise()
456 TFLITE_DCHECK_GT(params.input2_offset, -256); in Sub()
458 TFLITE_DCHECK_LT(params.input2_offset, 256); in Sub()
[all …]
|
| D | comparisons.h | 88 int32_t input2_offset = op_params.input2_offset; in ComparisonWithScaling() local
96 const int32_t input2_val = input2_offset + input2_data[i]; in ComparisonWithScaling()
182 int32_t input2_offset = op_params.input2_offset; in BroadcastComparison4DSlowWithScaling() local
194 input2_offset + in BroadcastComparison4DSlowWithScaling()
|
| D | legacy_reference_ops.h | 994 int32 input2_offset, int32 input2_multiplier, in Comparison() argument
1003 op_params.input2_offset = input2_offset; in Comparison()
1032 const Dims<4>& input2_dims, int32 input2_offset, in BroadcastComparison() argument
1042 op_params.input2_offset = input2_offset; in BroadcastComparison()
1066 const T* input2_data, const Dims<4>& input2_dims, int32 input2_offset, \
1072 input2_data, input2_dims, input2_offset, \
1089 const T* input2_data, const Dims<4>& input2_dims, int32 input2_offset, \
1096 input2_offset, input2_multiplier, \
1260 int32 input2_offset, int32 input2_multiplier, int input2_shift, in Add() argument
1281 op_params.input2_offset = input2_offset; in Add()
[all …]
|
| D | mul.h | 31 const int32_t input2_val = params.input2_offset + input2_data[i]; in MulElementwise()
96 params.input2_offset + in BroadcastMul4DSlow()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | add.h | 41 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddElementwise()
43 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddElementwise()
57 const int16x8_t input2_offset_dup = vdupq_n_s16(params.input2_offset); in AddElementwise()
145 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
176 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddScalarBroadcast()
178 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddScalarBroadcast()
210 vaddq_s16(input2_val_s16, vdupq_n_s16(params.input2_offset)); in AddScalarBroadcast()
248 const int32 input2_val = params.input2_offset + input2_data[i]; in AddScalarBroadcast()
278 TFLITE_DCHECK_GT(params.input2_offset, -256); in Add()
280 TFLITE_DCHECK_LT(params.input2_offset, 256); in Add()
|
| D | mul.h | 42 TFLITE_DCHECK_GT(params.input2_offset, -256); in MulElementwise()
43 TFLITE_DCHECK_LT(params.input2_offset, 256); in MulElementwise()
48 const int16x8_t input2_offset_vector = vdupq_n_s16(params.input2_offset); in MulElementwise()
127 const int32 input2_val = params.input2_offset + input2_data[i]; in MulElementwise()
150 TFLITE_DCHECK_GT(params.input2_offset, -256); in MulSimpleBroadcast()
151 TFLITE_DCHECK_LT(params.input2_offset, 256); in MulSimpleBroadcast()
155 const auto input2_offset_vector = vdupq_n_s16(params.input2_offset); in MulSimpleBroadcast()
219 const int32 input2_val = params.input2_offset + input2_data[i]; in MulSimpleBroadcast()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | add.cc | 55 int32_t input2_offset; member
68 data->input2_offset = -input2->params.zero_point; in CalculateOpData()
133 op_params.input2_offset = data->input2_offset; in EvalAddQuantized()
159 op_params.input1_shift, op_params.input2_offset, in EvalAddQuantized()
|
| D | mul.cc | 111 op_params.input2_offset = -data.input2_zero_point; in EvalQuantized()
134 op_params.input1_offset, op_params.input2_offset, in EvalQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | add.h | 32 TFLITE_DCHECK_GE(-params.input2_offset, std::numeric_limits<int8_t>::min()); in CheckArithmeticParams()
34 TFLITE_DCHECK_LE(-params.input2_offset, std::numeric_limits<int8_t>::max()); in CheckArithmeticParams()
88 const int32_t input2_val = params.input2_offset + y; in AddFunc()
|
| D | mul.h | 31 const int32_t input2_val = params.input2_offset + input2_data[i]; in MulElementwise()
111 params.input2_offset + in BroadcastMul4DSlow()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | sub.cc | 55 int32_t input2_offset; member
68 data->input2_offset = -input2->params.zero_point; in CalculateOpData()
160 op_params.input2_offset = data->input2_offset; in EvalSubQuantized()
|
| D | add.cc | 55 int32_t input2_offset; member
72 data->input2_offset = -input2->params.zero_point; in CalculateOpData()
142 op_params.input2_offset = data->input2_offset; in EvalAddQuantized()
|
| D | comparisons.cc | 628 auto input2_offset = -input2->params.zero_point; in Prepare() local
646 data->params.input2_offset = input2_offset; in Prepare()
|
| D | div.cc | 171 op_params.input2_offset = -input2->params.zero_point; in EvalQuantized()
|
| D | mul.cc | 98 op_params.input2_offset = -data->input2_zero_point; in EvalQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add.cc | 69 int32 input2_offset; member
169 data->input2_offset = -input2->params.zero_point; in Prepare()
294 op_params.input2_offset = data->input2_offset; in EvalAddQuantized()
|
| D | sub.cc | 72 int32 input2_offset; member
130 op_params->input2_offset = -input2_quantization_params.zero_point; in PrepareGeneralSubOp()
377 op_params.input2_offset = data->input2_offset; in EvalQuantized()
|
| D | squared_difference.cc | 99 data->arithmetic_params.input2_offset = in Prepare()
148 const int32_t input2_val = params.input2_offset + y; in SquaredDifference()
|
| D | comparisons.cc | 89 auto input2_offset = -input2->params.zero_point; in ComparisonQuantized() local
106 op_params.input2_offset = input2_offset; in ComparisonQuantized()
|
| D | mul.cc | 176 op_params.input2_offset = -input2->params.zero_point; in EvalQuantized()
209 TF_LITE_ENSURE_EQ(context, op_params.input2_offset, 0.0); in EvalQuantized()
|
| D | div.cc | 171 op_params.input2_offset = -input2->params.zero_point; in EvalQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 214 switched_params.input1_offset = unswitched_params.input2_offset; in BinaryBroadcastFiveFold()
217 switched_params.input2_offset = unswitched_params.input1_offset; in BinaryBroadcastFiveFold()
1866 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddElementwise()
1868 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddElementwise()
1884 vaddq_s16(input2_val_s16, vdupq_n_s16(params.input2_offset)); in AddElementwise()
1928 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
1959 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddScalarBroadcast()
1961 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddScalarBroadcast()
1995 vaddq_s16(input2_val_s16, vdupq_n_s16(params.input2_offset)); in AddScalarBroadcast()
2033 const int32 input2_val = params.input2_offset + input2_data[i]; in AddScalarBroadcast()
[all …]
|
| D | legacy_optimized_ops.h | 3512 int32 input2_offset, int32 input2_multiplier, int input2_shift, in Add() argument
3533 op_params.input2_offset = input2_offset; in Add()
3579 int32 input2_offset, int32 input2_multiplier, in BroadcastAdd() argument
3602 op_params.input2_offset = input2_offset; in BroadcastAdd()
3620 const Dims<4>& input2_dims, int32 input2_offset, int32 input2_multiplier, in BroadcastAddFivefold() argument
3642 op_params.input2_offset = input2_offset; in BroadcastAddFivefold()
3732 const Dims<4>& input2_dims, int32 input2_offset, in BroadcastMul() argument
3740 op_params.input2_offset = input2_offset; in BroadcastMul()
3754 const Dims<4>& input2_dims, int32 input2_offset, in BroadcastMul() argument
3760 input2_dims, input2_offset, output_offset, output_multiplier, in BroadcastMul()
|