Home
last modified time | relevance | path

Searched refs:input2_data (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/lite/micro/kernels/
Dcomparisons_test.cc55 int* input2_dims_data, float* input2_data, in TestComparisonFloat() argument
64 CreateTensor(input2_data, input2_dims), in TestComparisonFloat()
73 int* input2_dims_data, bool* input2_data, in TestComparisonBool() argument
82 CreateTensor(input2_data, input2_dims), in TestComparisonBool()
91 int* input2_dims_data, int32_t* input2_data, in TestComparisonInt() argument
100 CreateTensor(input2_data, input2_dims), in TestComparisonInt()
111 float* input2_data, uint8_t* input2_quantized, in TestComparisonQuantizedUInt8() argument
122 CreateQuantizedTensor(input2_data, input2_quantized, input2_dims, in TestComparisonQuantizedUInt8()
134 float* input2_data, int8_t* input2_quantized, in TestComparisonQuantizedInt8() argument
145 CreateQuantizedTensor(input2_data, input2_quantized, input2_dims, in TestComparisonQuantizedInt8()
[all …]
Dpack_test.cc51 const float* input2_data, int axis, in TestPackTwoInputsFloat() argument
64 CreateTensor(input2_data, input2_dims), in TestPackTwoInputsFloat()
83 const int* input2_dims_data, const float* input2_data, in TestPackThreeInputsFloat() argument
97 CreateTensor(input2_data, input2_dims), in TestPackThreeInputsFloat()
118 const uint8_t* input2_data, int axis, in TestPackTwoInputsQuantized() argument
135 CreateQuantizedTensor(input2_data, input2_dims, 1.0, 128), in TestPackTwoInputsQuantized()
155 const int32_t* input2_data, int axis, in TestPackTwoInputsQuantized32() argument
168 CreateTensor(input2_data, input2_dims), in TestPackTwoInputsQuantized32()
Dmaximum_minimum_test.cc29 const int* input2_dims_data, const float* input2_data, in TestMaxMinFloat() argument
42 CreateTensor(input2_data, input2_dims), in TestMaxMinFloat()
68 const uint8_t* input2_data, const float input2_scale, in TestMaxMinQuantized() argument
84 CreateQuantizedTensor(input2_data, input2_dims, input2_scale, in TestMaxMinQuantized()
110 const int32_t* input2_data, const int32_t* expected_output_data, in TestMaxMinQuantizedInt32() argument
122 CreateTensor(input2_data, input2_dims), in TestMaxMinQuantizedInt32()
Dconcatenation_test.cc30 const float* input2_data, int axis, in TestConcatenateTwoInputs() argument
42 CreateTensor(input2_data, input2_dims), in TestConcatenateTwoInputs()
72 const int* input2_dims_data, const uint8_t* input2_data, in TestConcatenateQuantizedTwoInputs() argument
87 CreateQuantizedTensor(input2_data, input2_dims, input_scale, in TestConcatenateQuantizedTwoInputs()
Darg_min_max.cc36 const T1* input1_data, const T3* input2_data, in ArgMinMaxHelper() argument
40 reference_ops::ArgMinMax(input1_shape, input1_data, input2_data, in ArgMinMaxHelper()
43 reference_ops::ArgMinMax(input1_shape, input1_data, input2_data, in ArgMinMaxHelper()
Dmul_test.cc71 const int* input2_dims_data, const float* input2_data, in TestMulFloat() argument
84 CreateTensor(input2_data, input2_dims), in TestMulFloat()
95 const float* input2_data, T* input2_quantized, in TestMulQuantized() argument
112 CreateQuantizedTensor(input2_data, input2_quantized, input2_dims, in TestMulQuantized()
Dlogical_test.cc29 const int* input2_dims_data, const bool* input2_data, in TestLogicalOp() argument
42 CreateTensor(input2_data, input2_dims), in TestLogicalOp()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dcomparisons.h60 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonImpl() argument
64 output_data[i] = F(input1_data[i], input2_data[i]); in ComparisonImpl()
73 const float* input2_data, in Comparison() argument
76 input2_data, output_shape, output_data); in Comparison()
83 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonWithScaling() argument
96 const int32_t input2_val = input2_offset + input2_data[i]; in ComparisonWithScaling()
134 const RuntimeShape& unextended_input2_shape, const T* input2_data, in BroadcastComparison4DSlowImpl() argument
147 input2_data[SubscriptToIndex(dims.desc2, b, y, x, c)]); in BroadcastComparison4DSlowImpl()
159 const float* input2_data, in BroadcastComparison4DSlow() argument
163 input2_shape, input2_data, in BroadcastComparison4DSlow()
[all …]
Dsub.h36 const float* input2_data, in SubNonBroadcast() argument
43 input1_data[i] - input2_data[i], params.float_activation_min, in SubNonBroadcast()
52 const int32_t* input2_data, in SubNonBroadcast() argument
59 input1_data[i] - input2_data[i], params.quantized_activation_min, in SubNonBroadcast()
73 const float* input2_data, in BroadcastSubSlow() argument
102 input2_data[SubscriptToIndex(desc2, indexes)], in BroadcastSubSlow()
113 const uint8_t* input2_data, in BroadcastSubSlow() argument
142 params.input2_offset + input2_data[SubscriptToIndex(desc2, indexes)]; in BroadcastSubSlow()
170 const int32_t* input2_data, in BroadcastSubSlow() argument
199 input2_data[SubscriptToIndex(desc2, indexes)], in BroadcastSubSlow()
[all …]
Dadd.h28 const RuntimeShape& input2_shape, const T* input2_data, in Add() argument
34 input1_data[i] + input2_data[i], params.quantized_activation_min, in Add()
41 const RuntimeShape& input2_shape, const float* input2_data, in Add() argument
46 auto x = input1_data[i] + input2_data[i]; in Add()
60 const T* input1_data, const T* input2_data, in AddElementwise() argument
69 const int32_t input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
94 uint8_t input1_data, const uint8_t* input2_data, in AddScalarBroadcast() argument
107 const int32_t input2_val = params.input2_offset + input2_data[i]; in AddScalarBroadcast()
126 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Add() argument
137 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
[all …]
Dmul.h28 const uint8_t* input2_data, uint8_t* output_data) { in MulElementwise() argument
31 const int32_t input2_val = params.input2_offset + input2_data[i]; in MulElementwise()
47 const RuntimeShape& input2_shape, const T* input2_data, in Mul() argument
57 input1_data[i] * input2_data[i], output_activation_min, in Mul()
64 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Mul() argument
71 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
78 const uint8_t* input2_data, in BroadcastMul4DSlow() argument
97 input2_data[SubscriptToIndex(desc2, b, y, x, c)]; in BroadcastMul4DSlow()
119 const T* input2_data, in BroadcastMul4DSlow() argument
155 input2_data[SubscriptToIndex(desc2, b, y, x, c)], in BroadcastMul4DSlow()
Ddiv.h29 const uint8* input1_data, const uint8* input2_data, in DivElementwise() argument
40 const int32 input2_val = params.input2_offset + input2_data[i]; in DivElementwise()
63 const RuntimeShape& input2_shape, const uint8* input2_data, in Div() argument
70 DivElementwise(flat_size, params, input1_data, input2_data, output_data); in Div()
78 const uint8* input2_data, in BroadcastDivSlow() argument
104 params.input2_offset + input2_data[SubscriptToIndex(desc2, indexes)]; in BroadcastDivSlow()
136 const T* input2_data, in BroadcastDivSlow() argument
167 input2_data[SubscriptToIndex(desc2, indexes)], in BroadcastDivSlow()
176 const RuntimeShape& input2_shape, const T* input2_data, in Div() argument
186 input1_data[i] / input2_data[i], output_activation_min, in Div()
Dlegacy_reference_ops.h623 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv() argument
630 DimsToShape(input2_dims), input2_data, in BroadcastDiv()
636 const T* input2_data, const Dims<4>& input2_dims, in Div() argument
643 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Div()
980 const T* input2_data, const Dims<4>& input2_dims, in Comparison() argument
985 DimsToShape(input2_dims), input2_data, in Comparison()
993 const T* input2_data, const Dims<4>& input2_dims, in Comparison() argument
1009 DimsToShape(input2_dims), input2_data, in Comparison()
1016 const T* input2_data, in BroadcastComparison() argument
1023 input2_data, DimsToShape(output_dims), in BroadcastComparison()
[all …]
Dbinary_function.h32 const RuntimeShape& unextended_input2_shape, const T2* input2_data, in BroadcastBinaryFunction4DSlow() argument
54 auto in2_val = input2_data[in2_idx]; in BroadcastBinaryFunction4DSlow()
67 const T2* input2_data, in BinaryFunction() argument
73 output_data[i] = func(input1_data[i], input2_data[i]); in BinaryFunction()
Dmaximum_minimum.h28 const T* input2_data, in MaximumMinimumBroadcastSlow() argument
37 output_data[i] = op(input1_data[i], input2_data[i]); in MaximumMinimumBroadcastSlow()
55 input2_data[SubscriptToIndex(desc2, indexes)]); in MaximumMinimumBroadcastSlow()
Darg_min_max.h26 const T3* input2_data, const RuntimeShape& output_shape, in ArgMinMax() argument
31 int axis = input2_data[0]; in ArgMinMax()
Dreference_ops.h187 const uint8* input2_data = in BroadcastMulFivefold() local
196 const uint8* input2_data_reset = input2_data; in BroadcastMulFivefold()
222 const RuntimeShape& input2_shape, const int16* input2_data, in Mul() argument
234 F0::FromRaw(input1_data[i]) * F0::FromRaw(input2_data[i]); in Mul()
241 const RuntimeShape& input2_shape, const int16* input2_data, in Mul() argument
257 F0::FromRaw(input1_data[i]) * F0::FromRaw(input2_data[i]); in Mul()
270 const RuntimeShape& input2_shape, const int16_t* input2_data, in Sub16() argument
282 const int16* not_shift_input = input1_shift == 0 ? input1_data : input2_data; in Sub16()
283 const int16* shift_input = input1_shift == 0 ? input2_data : input1_data; in Sub16()
1423 const T* input2_data, const RuntimeShape& output_shape, in Minimum() argument
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dceva_tflm_lib.h91 const float* input2_data, float* output_data,
96 const float* input2_data,
101 const void* params_inp, const float* input1_data, const float* input2_data,
106 const void* params_inp, const float* input1_data, const float* input2_data,
112 const float* input2_data,
117 const float* input1_data, const float* input2_data, float* output_data,
121 const float* input1_data, const float* input2_data, float* output_data,
126 const float* input2_data, float* output_data,
129 const float* input2_data, float* output_data,
132 const float* input2_data, float* output_data,
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dadd.h39 const int8_t* input2_data, int8_t* output_data, in ElementWise() argument
44 output_data[i] = binary_func(input1_data[i], input2_data[i], params); in ElementWise()
51 const int8_t* input2_data, const RuntimeShape& output_shape, in BroadcastBinaryFunction4DSlow() argument
79 input2_data[SubscriptToIndex(desc2, b, y, x, c)], params); in BroadcastBinaryFunction4DSlow()
111 const int8_t* input1_data, const int8_t* input2_data, in AddElementwise() argument
113 ElementWise(size, params, input1_data, input2_data, output_data, in AddElementwise()
119 const RuntimeShape& input2_shape, const int8_t* input2_data, in Add() argument
126 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
133 const int8_t* input2_data, in BroadcastAdd4DSlow() argument
137 input2_data, output_shape, output_data, in BroadcastAdd4DSlow()
Dmul.h27 const T* input1_data, const T* input2_data, in MulElementwise() argument
31 const int32_t input2_val = params.input2_offset + input2_data[i]; in MulElementwise()
47 const RuntimeShape& input2_shape, const T* input2_data, in Mul() argument
55 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
61 const RuntimeShape& input2_shape, const int16_t* input2_data, in Mul() argument
77 F0::FromRaw(input1_data[i]) * F0::FromRaw(input2_data[i]); in Mul()
92 const T* input2_data, const RuntimeShape& output_shape, T* output_data) { in BroadcastMul4DSlow() argument
112 input2_data[SubscriptToIndex(desc2, b, y, x, c)]; in BroadcastMul4DSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmul.h36 const int8* input1_data, const int8* input2_data, in MulElementwise() argument
60 const int8x16_t input2_val_original = vld1q_s8(input2_data + i); in MulElementwise()
127 const int32 input2_val = params.input2_offset + input2_data[i]; in MulElementwise()
143 const int8* input2_data, int8* output_data) { in MulSimpleBroadcast() argument
166 const auto input2_val_original = vld1q_s8(input2_data + i); in MulSimpleBroadcast()
219 const int32 input2_val = params.input2_offset + input2_data[i]; in MulSimpleBroadcast()
234 const RuntimeShape& input2_shape, const int8* input2_data, in Mul() argument
242 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
249 const int8* input2_data, in BroadcastMulDispatch() argument
254 params, input1_shape, input1_data, input2_shape, input2_data, in BroadcastMulDispatch()
[all …]
Dadd.h36 const int8* input1_data, const int8* input2_data, in AddElementwise() argument
61 const int8x16_t input2_val_original = vld1q_s8(input2_data + i); in AddElementwise()
145 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
170 int8 input1_data, const int8* input2_data, in AddScalarBroadcast() argument
207 const int8x8_t input2_val_original = vld1_s8(input2_data + i); in AddScalarBroadcast()
248 const int32 input2_val = params.input2_offset + input2_data[i]; in AddScalarBroadcast()
269 const RuntimeShape& input2_shape, const int8* input2_data, in Add() argument
281 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
288 const int8* input2_data, in BroadcastAddDispatch() argument
293 params, input1_shape, input1_data, input2_shape, input2_data, in BroadcastAddDispatch()
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Darithmetic_test.cc35 const std::initializer_list<uint8_t>& input2_data) { in ArithmeticOpBaseModel() argument
40 if (input2_data.size() > 0) in ArithmeticOpBaseModel()
41 input2_ = AddConstInput(input2, input2_data); in ArithmeticOpBaseModel()
84 const std::initializer_list<uint8_t>& input2_data, in AddOpModel() argument
86 : ArithmeticOpBaseModel(input1, input2, output, input1_data, input2_data), in AddOpModel()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dbinary_elementwise_tester.cc234 std::vector<uint16_t> input2_data(ComputeSize(Input2Shape())); in CreateTfLiteModel() local
235 std::generate(input2_data.begin(), input2_data.end(), in CreateTfLiteModel()
240 reinterpret_cast<const uint8_t*>(input2_data.data()), in CreateTfLiteModel()
241 sizeof(uint16_t) * input2_data.size()))); in CreateTfLiteModel()
243 std::vector<float> input2_data(ComputeSize(Input2Shape())); in CreateTfLiteModel() local
244 std::generate(input2_data.begin(), input2_data.end(), input2_rng); in CreateTfLiteModel()
251 reinterpret_cast<const uint8_t*>(input2_data.data()), in CreateTfLiteModel()
252 sizeof(float) * input2_data.size()))); in CreateTfLiteModel()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h229 const T* input2_data = in BinaryBroadcastFiveFold() local
238 const T* input2_data_reset = input2_data; in BinaryBroadcastFiveFold()
1798 const float* input1_data, const float* input2_data, in AddElementwise() argument
1810 auto a20 = vld1q_f32(input2_data + i); in AddElementwise()
1811 auto a21 = vld1q_f32(input2_data + i + 4); in AddElementwise()
1812 auto a22 = vld1q_f32(input2_data + i + 8); in AddElementwise()
1813 auto a23 = vld1q_f32(input2_data + i + 12); in AddElementwise()
1833 auto a2 = vld1q_f32(input2_data + i); in AddElementwise()
1842 auto x = input1_data[i] + input2_data[i]; in AddElementwise()
1850 const RuntimeShape& input2_shape, const float* input2_data, in Add() argument
[all …]

12