Home
last modified time | relevance | path

Searched refs:input2_offset (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmul.h36 TFLITE_DCHECK_GT(params.input2_offset, -256); in MulElementwise()
37 TFLITE_DCHECK_LT(params.input2_offset, 256); in MulElementwise()
42 const auto input2_offset_vector = vdupq_n_s16(params.input2_offset); in MulElementwise()
89 const int32 input2_val = params.input2_offset + input2_data[i]; in MulElementwise()
112 TFLITE_DCHECK_GT(params.input2_offset, -256); in MulSimpleBroadcast()
113 TFLITE_DCHECK_LT(params.input2_offset, 256); in MulSimpleBroadcast()
117 const auto input2_offset_vector = vdupq_n_s16(params.input2_offset); in MulSimpleBroadcast()
158 const int32 input2_val = params.input2_offset + input2_data[i]; in MulSimpleBroadcast()
194 switched_params.input1_offset = unswitched_params.input2_offset; in BroadcastMulFivefold()
195 switched_params.input2_offset = unswitched_params.input1_offset; in BroadcastMulFivefold()
Dadd.h35 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddElementwise()
37 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddElementwise()
51 vaddq_s16(input2_val_s16, vdupq_n_s16(params.input2_offset)); in AddElementwise()
95 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
126 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddScalarBroadcast()
128 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddScalarBroadcast()
160 vaddq_s16(input2_val_s16, vdupq_n_s16(params.input2_offset)); in AddScalarBroadcast()
198 const int32 input2_val = params.input2_offset + input2_data[i]; in AddScalarBroadcast()
228 TFLITE_DCHECK_GT(params.input2_offset, -256); in Add()
230 TFLITE_DCHECK_LT(params.input2_offset, 256); in Add()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dadd.h33 TFLITE_DCHECK_GE(params.input2_offset, -1 * int8_max_value); in AddElementwise()
35 TFLITE_DCHECK_LE(params.input2_offset, int8_max_value); in AddElementwise()
39 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
71 TFLITE_DCHECK_GE(params.input2_offset, -1 * int8_max_value); in Add()
73 TFLITE_DCHECK_LE(params.input2_offset, int8_max_value); in Add()
110 params.input2_offset + in BroadcastAdd4DSlow()
Dmul.h30 const int32 input2_val = params.input2_offset + input2_data[i]; in MulElementwise()
110 params.input2_offset + in BroadcastMul4DSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dadd.h58 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddElementwise()
60 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddElementwise()
64 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
92 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddScalarBroadcast()
94 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddScalarBroadcast()
102 const int32 input2_val = params.input2_offset + input2_data[i]; in AddScalarBroadcast()
129 TFLITE_DCHECK_GT(params.input2_offset, -256); in Add()
131 TFLITE_DCHECK_LT(params.input2_offset, 256); in Add()
293 params.input2_offset + in BroadcastAdd4DSlow()
331 switched_params.input1_offset = unswitched_params.input2_offset; in BroadcastAddFivefold()
[all …]
Dcomparisons.h87 int32 input2_offset = op_params.input2_offset; in ComparisonWithScaling() local
95 const int32 input2_val = input2_offset + input2_data[i]; in ComparisonWithScaling()
171 int32 input2_offset = op_params.input2_offset; in BroadcastComparison4DSlowWithScaling() local
182 input2_offset + input2_data[SubscriptToIndex(desc2, b, y, x, c)]; in BroadcastComparison4DSlowWithScaling()
Dlegacy_reference_ops.h1024 int32 input2_offset, int32 input2_multiplier, in Comparison() argument
1033 op_params.input2_offset = input2_offset; in Comparison()
1062 const Dims<4>& input2_dims, int32 input2_offset, in BroadcastComparison() argument
1072 op_params.input2_offset = input2_offset; in BroadcastComparison()
1096 const T* input2_data, const Dims<4>& input2_dims, int32 input2_offset, \
1102 input2_data, input2_dims, input2_offset, \
1119 const T* input2_data, const Dims<4>& input2_dims, int32 input2_offset, \
1126 input2_offset, input2_multiplier, \
1290 int32 input2_offset, int32 input2_multiplier, int input2_shift, in Add() argument
1311 op_params.input2_offset = input2_offset; in Add()
[all …]
Dmul.h31 const int32 input2_val = params.input2_offset + input2_data[i]; in MulElementwise()
96 params.input2_offset + in BroadcastMul4DSlow()
Dreference_ops.h382 switched_params.input1_offset = unswitched_params.input2_offset; in BroadcastMulFivefold()
383 switched_params.input2_offset = unswitched_params.input1_offset; in BroadcastMulFivefold()
552 TFLITE_DCHECK_GT(params.input2_offset, -256); in DivElementwise()
553 TFLITE_DCHECK_LT(params.input2_offset, 256); in DivElementwise()
559 const int32 input2_val = params.input2_offset + input2_data[i]; in DivElementwise()
613 TFLITE_DCHECK_GT(params.input2_offset, -256); in BroadcastDiv4DSlow()
614 TFLITE_DCHECK_LT(params.input2_offset, 256); in BroadcastDiv4DSlow()
626 params.input2_offset + in BroadcastDiv4DSlow()
768 params.input2_offset + in BroadcastSub4DSlow()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis-nn/
Dadd.cc53 int32 input2_offset; member
66 data->input2_offset = -input2->params.zero_point; in CalculateOpData()
128 op_params.input2_offset = data->input2_offset; in EvalAddQuantized()
150 op_params.input1_shift, op_params.input2_offset, in EvalAddQuantized()
Dmul.cc74 op_params.input2_offset = -input2->params.zero_point; in EvalQuantized()
93 op_params.input1_offset, op_params.input2_offset, in EvalQuantized()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dadd.cc53 int32 input2_offset; member
76 data->input2_offset = -input2->params.zero_point; in CalculateOpData()
139 op_params.input2_offset = data->input2_offset; in EvalAddQuantized()
Dcomparisons.cc41 auto input2_offset = -input2->params.zero_point; \
60 op_params.input2_offset = input2_offset; \
Dmul.cc81 op_params.input2_offset = -input2->params.zero_point; in EvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/
Dadd.cc59 int32 input2_offset; member
99 data->input2_offset = -input2->params.zero_point; in Prepare()
230 op_params.input2_offset = data->input2_offset; in EvalAddQuantized()
Dsub.cc61 int32 input2_offset; member
110 op_params->input2_offset = -input2_quantization_params.zero_point; in Prepare8BitSubOp()
279 op_params.input2_offset = data->input2_offset; in EvalQuantized()
Dcomparisons.cc62 auto input2_offset = -input2->params.zero_point; in ComparisonQuantized() local
79 op_params.input2_offset = input2_offset; in ComparisonQuantized()
Ddiv.cc158 op_params.input2_offset = -input2->params.zero_point; in EvalQuantized()
Dmul.cc161 op_params.input2_offset = -input2->params.zero_point; in EvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1733 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddElementwise()
1735 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddElementwise()
1751 vaddq_s16(input2_val_s16, vdupq_n_s16(params.input2_offset)); in AddElementwise()
1795 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
1826 TFLITE_DCHECK_GT(params.input2_offset, -256); in AddScalarBroadcast()
1828 TFLITE_DCHECK_LT(params.input2_offset, 256); in AddScalarBroadcast()
1862 vaddq_s16(input2_val_s16, vdupq_n_s16(params.input2_offset)); in AddScalarBroadcast()
1900 const int32 input2_val = params.input2_offset + input2_data[i]; in AddScalarBroadcast()
1963 TFLITE_DCHECK_GT(params.input2_offset, -256); in Add()
1965 TFLITE_DCHECK_LT(params.input2_offset, 256); in Add()
[all …]
Dlegacy_optimized_ops.h3499 int32 input2_offset, int32 input2_multiplier, int input2_shift, in Add() argument
3520 op_params.input2_offset = input2_offset; in Add()
3566 int32 input2_offset, int32 input2_multiplier, in BroadcastAdd() argument
3589 op_params.input2_offset = input2_offset; in BroadcastAdd()
3607 const Dims<4>& input2_dims, int32 input2_offset, int32 input2_multiplier, in BroadcastAddFivefold() argument
3629 op_params.input2_offset = input2_offset; in BroadcastAddFivefold()
3719 const Dims<4>& input2_dims, int32 input2_offset, in BroadcastMul() argument
3727 op_params.input2_offset = input2_offset; in BroadcastMul()
3741 const Dims<4>& input2_dims, int32 input2_offset, in BroadcastMul() argument
3747 input2_dims, input2_offset, output_offset, output_multiplier, in BroadcastMul()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h758 int32 input2_offset; member
802 int32 input2_offset; member