| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | add.h | 69 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddElementwise()
100 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddScalarBroadcast()
142 const int input1_shift = params.input1_shift; in Add() local
148 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
149 TFLITE_DCHECK_LE(input1_shift, 0); in Add()
151 const int16* not_shift_input = input1_shift == 0 ? input1_data : input2_data; in Add()
152 const int16* shift_input = input1_shift == 0 ? input2_data : input1_data; in Add()
154 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Add()
302 params.input1_shift); in BroadcastAdd4DSlow()
333 switched_params.input1_shift = unswitched_params.input2_shift; in BroadcastAddFivefold()
[all …]
|
| D | comparisons.h | 86 int input1_shift = op_params.input1_shift; in ComparisonWithScaling() local
100 shifted_input1_val, input1_multiplier, input1_shift); in ComparisonWithScaling()
170 int input1_shift = op_params.input1_shift; in BroadcastComparison4DSlowWithScaling() local
187 shifted_input1_val, input1_multiplier, input1_shift); in BroadcastComparison4DSlowWithScaling()
|
| D | legacy_reference_ops.h | 1022 int32 input1_multiplier, int input1_shift, in Comparison() argument
1032 op_params.input1_shift = kReverseShift * input1_shift; in Comparison()
1060 int32 input1_multiplier, int input1_shift, in BroadcastComparison() argument
1071 op_params.input1_shift = kReverseShift * input1_shift; in BroadcastComparison()
1095 int32 input1_offset, int32 input1_multiplier, int input1_shift, \
1101 input1_offset, input1_multiplier, input1_shift, \
1118 int32 input1_offset, int32 input1_multiplier, int input1_shift, \
1125 input1_shift, input2_data, input2_dims, \
1288 int32 input1_multiplier, int input1_shift, in Add() argument
1310 op_params.input1_shift = kReverseShift * input1_shift; in Add()
[all …]
|
| D | reference_ops.h | 777 params.input1_shift); in BroadcastSub4DSlow()
954 const int input1_shift = params.input1_shift; in Sub16() local
960 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Sub16()
961 TFLITE_DCHECK_LE(input1_shift, 0); in Sub16()
963 const int16* not_shift_input = input1_shift == 0 ? input1_data : input2_data; in Sub16()
964 const int16* shift_input = input1_shift == 0 ? input2_data : input1_data; in Sub16()
966 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Sub16()
968 if (input1_shift == 0) { in Sub16()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add.cc | 47 int input1_shift; member
113 real_input1_multiplier, &data->input1_multiplier, &data->input1_shift); in Prepare()
152 data->input1_shift = input1_scale_log2_rounded - output_scale_log2_rounded; in Prepare()
157 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in Prepare()
158 TF_LITE_ENSURE(context, data->input1_shift <= 0); in Prepare()
229 op_params.input1_shift = data->input1_shift; in EvalAddQuantized()
278 op_params.input1_shift = data->input1_shift; \ in EvalAddQuantized()
|
| D | sub.cc | 49 int input1_shift; member
126 &op_params->input1_shift); in Prepare8BitSubOp()
172 data->input1_shift = input1_scale_log2_rounded - output_scale_log2_rounded; in PrepareInt16SubOp()
177 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in PrepareInt16SubOp()
178 TF_LITE_ENSURE(context, data->input1_shift <= 0); in PrepareInt16SubOp()
278 op_params.input1_shift = data->input1_shift; in EvalQuantized()
|
| D | comparisons.cc | 66 int input1_shift; in ComparisonQuantized() local
68 &input1_multiplier, &input1_shift); in ComparisonQuantized()
78 op_params.input1_shift = input1_shift; in ComparisonQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis-nn/ |
| D | add.cc | 41 int input1_shift; member
80 real_input1_multiplier, &data->input1_multiplier, &data->input1_shift); in CalculateOpData()
127 op_params.input1_shift = data->input1_shift; in EvalAddQuantized()
150 op_params.input1_shift, op_params.input2_offset, in EvalAddQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | add.cc | 41 int input1_shift; member
91 real_input1_multiplier, &data->input1_multiplier, &data->input1_shift); in CalculateOpData()
138 op_params.input1_shift = data->input1_shift; in EvalAddQuantized()
|
| D | comparisons.cc | 45 int input1_shift; \
48 &input1_shift); \
59 op_params.input1_shift = input1_shift; \
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | add.h | 69 const int32x4_t input1_shift_dup = vdupq_n_s32(params.input1_shift); in AddElementwise()
100 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddElementwise()
152 const int32x4_t input1_shift_dup = vdupq_n_s32(params.input1_shift); in AddScalarBroadcast()
195 shifted_input1_val, params.input1_multiplier, params.input1_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 | 44 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddElementwise()
119 params.input1_shift); in BroadcastAdd4DSlow()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 765 int input1_shift; member
801 int input1_shift; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 3497 int32 input1_multiplier, int input1_shift, in Add() argument
3519 op_params.input1_shift = kReverseShift * input1_shift; in Add()
3564 int32 input1_multiplier, int input1_shift, in BroadcastAdd() argument
3588 op_params.input1_shift = kReverseShift * input1_shift; in BroadcastAdd()
3606 int32 input1_multiplier, int input1_shift, const uint8* input2_data, in BroadcastAddFivefold() argument
3628 op_params.input1_shift = kReverseShift * input1_shift; in BroadcastAddFivefold()
3662 int input1_shift, const int16* input2_data, in Add() argument
3679 op_params.input1_shift = kReverseShift * input1_shift; in Add()
|
| D | optimized_ops.h | 1769 const int32x4_t input1_shift_dup = vdupq_n_s32(params.input1_shift); in AddElementwise()
1800 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddElementwise()
1853 const int32x4_t input1_shift_dup = vdupq_n_s32(params.input1_shift); in AddScalarBroadcast()
1897 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddScalarBroadcast()
1977 const int input1_shift = params.input1_shift; in Add() local
1983 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
1984 TFLITE_DCHECK_LE(input1_shift, 0); in Add()
1986 const int16* not_shift_input = input1_shift == 0 ? input1_data : input2_data; in Add()
1987 const int16* shift_input = input1_shift == 0 ? input2_data : input1_data; in Add()
1989 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Add()
[all …]
|