| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add.cc | 47 int input2_shift; member
116 real_input2_multiplier, &data->input2_multiplier, &data->input2_shift); in Prepare()
159 data->input2_shift = input2_scale_log2_rounded - output_scale_log2_rounded; in Prepare()
163 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in Prepare()
165 TF_LITE_ENSURE(context, data->input2_shift <= 0); in Prepare()
236 op_params.input2_shift = data->input2_shift; in EvalAddQuantized()
278 op_params.input2_shift = data->input2_shift; \ in EvalAddQuantized()
|
| D | sub.cc | 48 int input2_shift; member
127 &op_params->input2_shift); in Prepare8BitSubOp()
176 data->input2_shift = input2_scale_log2_rounded - output_scale_log2_rounded; in PrepareInt16SubOp()
180 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in PrepareInt16SubOp()
182 TF_LITE_ENSURE(context, data->input2_shift <= 0); in PrepareInt16SubOp()
284 op_params.input2_shift = data->input2_shift; in EvalQuantized()
|
| D | comparisons.cc | 77 int input2_shift; \
79 &input2_multiplier, &input2_shift); \
88 op_params.input2_shift = input2_shift; \
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 904 int input2_shift, bool* output_data, in Comparison() argument
915 op_params.input2_shift = kReverseShift * input2_shift; in Comparison()
942 int32 input2_multiplier, int input2_shift, in BroadcastComparison() argument
954 op_params.input2_shift = kReverseShift * input2_shift; in BroadcastComparison()
976 int32 input2_multiplier, int input2_shift, bool* output_data, \
982 input2_multiplier, input2_shift, output_data, \
999 int32 input2_multiplier, int input2_shift, bool* output_data, \
1006 input2_shift, output_data, output_dims); \
1169 int32 input2_offset, int32 input2_multiplier, int input2_shift, in Add() argument
1192 op_params.input2_shift = kReverseShift * input2_shift; in Add()
[all …]
|
| D | reference_ops.h | 586 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
620 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddScalarBroadcast()
662 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
664 TFLITE_DCHECK_LE(params.input2_shift, 0); in Add()
668 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Add()
823 params.input2_shift); in BroadcastAdd4DSlow()
850 switched_params.input1_shift = unswitched_params.input2_shift; in BroadcastAddFivefold()
853 switched_params.input2_shift = unswitched_params.input1_shift; in BroadcastAddFivefold()
1384 params.input2_shift); in BroadcastSub4DSlow()
1563 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Sub16()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | add.h | 47 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
124 params.input2_shift); in BroadcastAdd4DSlow()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 949 int32 input2_offset, int32 input2_multiplier, int input2_shift, in Add() argument
972 op_params.input2_shift = kReverseShift * input2_shift; in Add()
1017 int input2_shift, int32 output_offset, in BroadcastAdd() argument
1041 op_params.input2_shift = kReverseShift * input2_shift; in BroadcastAdd()
1058 int input2_shift, int32 output_offset, int32 output_multiplier, in BroadcastAddFivefold() argument
1081 op_params.input2_shift = kReverseShift * input2_shift; in BroadcastAddFivefold()
1113 const Dims<4>& input2_dims, int input2_shift, in Add() argument
1130 op_params.input2_shift = kReverseShift * input2_shift; in Add()
|
| D | optimized_ops.h | 2722 const int32x4_t input2_shift_dup = vdupq_n_s32(params.input2_shift); in AddElementwise()
2755 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
2823 const int32x4_t input2_shift_dup = vdupq_n_s32(params.input2_shift); in AddScalarBroadcast()
2857 params.input2_shift); in AddScalarBroadcast()
2902 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
2904 TFLITE_DCHECK_LE(params.input2_shift, 0); in Add()
2908 input1_shift == 0 ? -params.input2_shift : -input1_shift; in Add()
2962 switched_params.input1_shift = unswitched_params.input2_shift; in BroadcastAddFivefold()
2965 switched_params.input2_shift = unswitched_params.input1_shift; in BroadcastAddFivefold()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 711 int input2_shift; member
748 int input2_shift; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 735 int input2_shift; member
772 int input2_shift; member
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 2938 const auto input2_shift_dup = vdupq_n_s32(params.input2_shift); in AddElementwise()
2971 shifted_input2_val, params.input2_multiplier, params.input2_shift); in AddElementwise()
3015 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
3017 TFLITE_DCHECK_LE(params.input2_shift, 0); 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()
|