| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add.cc | 46 int input1_shift; member
113 real_input1_multiplier, &data->input1_multiplier, &data->input1_shift); in Prepare()
158 data->input1_shift = input1_scale_log2_rounded - output_scale_log2_rounded; in Prepare()
163 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in Prepare()
164 TF_LITE_ENSURE(context, data->input1_shift <= 0); in Prepare()
233 op_params.input1_shift = data->input1_shift; in EvalAddQuantized()
277 op_params.input1_shift = data->input1_shift; \ in EvalAddQuantized()
|
| D | sub.cc | 47 int input1_shift; member
124 &op_params->input1_shift); in Prepare8BitSubOp()
175 data->input1_shift = input1_scale_log2_rounded - output_scale_log2_rounded; in PrepareInt16SubOp()
180 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in PrepareInt16SubOp()
181 TF_LITE_ENSURE(context, data->input1_shift <= 0); in PrepareInt16SubOp()
281 op_params.input1_shift = data->input1_shift; in EvalQuantized()
|
| D | comparisons.cc | 73 int input1_shift; \
75 &input1_multiplier, &input1_shift); \
85 op_params.input1_shift = input1_shift; \
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 901 int32 input1_multiplier, int input1_shift, in Comparison() argument
911 op_params.input1_shift = kReverseShift * input1_shift; in Comparison()
939 int32 input1_multiplier, int input1_shift, in BroadcastComparison() argument
950 op_params.input1_shift = kReverseShift * input1_shift; in BroadcastComparison()
974 int32 input1_offset, int32 input1_multiplier, int input1_shift, \
980 input1_offset, input1_multiplier, input1_shift, \
997 int32 input1_offset, int32 input1_multiplier, int input1_shift, \
1004 input1_shift, input2_data, input2_dims, \
1167 int32 input1_multiplier, int input1_shift, in Add() argument
1189 op_params.input1_shift = kReverseShift * input1_shift; in Add()
[all …]
|
| D | reference_ops.h | 583 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddElementwise()
614 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddScalarBroadcast()
656 const int input1_shift = params.input1_shift; in Add() local
662 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
663 TFLITE_DCHECK_LE(input1_shift, 0); in Add()
665 const int16* not_shift_input = input1_shift == 0 ? input1_data : input2_data; in Add()
666 const int16* shift_input = input1_shift == 0 ? input2_data : input1_data; in Add()
668 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Add()
819 params.input1_shift); in BroadcastAdd4DSlow()
850 switched_params.input1_shift = unswitched_params.input2_shift; in BroadcastAddFivefold()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | add.h | 44 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddElementwise()
120 params.input1_shift); in BroadcastAdd4DSlow()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 947 int32 input1_multiplier, int input1_shift, in Add() argument
969 op_params.input1_shift = kReverseShift * input1_shift; in Add()
1014 int32 input1_multiplier, int input1_shift, in BroadcastAdd() argument
1038 op_params.input1_shift = kReverseShift * input1_shift; in BroadcastAdd()
1056 int32 input1_multiplier, int input1_shift, const uint8* input2_data, in BroadcastAddFivefold() argument
1078 op_params.input1_shift = kReverseShift * input1_shift; in BroadcastAddFivefold()
1112 int input1_shift, const int16* input2_data, in Add() argument
1129 op_params.input1_shift = kReverseShift * input1_shift; in Add()
|
| D | optimized_ops.h | 2721 const int32x4_t input1_shift_dup = vdupq_n_s32(params.input1_shift); in AddElementwise()
2752 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddElementwise()
2805 const int32x4_t input1_shift_dup = vdupq_n_s32(params.input1_shift); in AddScalarBroadcast()
2849 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddScalarBroadcast()
2896 const int input1_shift = params.input1_shift; in Add() local
2902 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
2903 TFLITE_DCHECK_LE(input1_shift, 0); in Add()
2905 const int16* not_shift_input = input1_shift == 0 ? input1_data : input2_data; in Add()
2906 const int16* shift_input = input1_shift == 0 ? input2_data : input1_data; in Add()
2908 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Add()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 709 int input1_shift; member
745 int input1_shift; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 733 int input1_shift; member
769 int input1_shift; member
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 2937 const auto input1_shift_dup = vdupq_n_s32(params.input1_shift); in AddElementwise()
2968 shifted_input1_val, params.input1_multiplier, params.input1_shift); in AddElementwise()
3009 const int input1_shift = params.input1_shift; in Add() local
3015 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
3016 TFLITE_DCHECK_LE(input1_shift, 0); in Add()
3018 const int16* not_shift_input = input1_shift == 0 ? input1_data : input2_data; in Add()
3019 const int16* shift_input = input1_shift == 0 ? input2_data : input1_data; in Add()
3021 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Add()
3075 switched_params.input1_shift = unswitched_params.input2_shift; in BroadcastAddFivefold()
3078 switched_params.input2_shift = unswitched_params.input1_shift; in BroadcastAddFivefold()
|