Home
last modified time | relevance | path

Searched refs:input1_data (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/lite/micro/kernels/
Dcomparisons_test.cc54 int* input1_dims_data, float* input1_data, in TestComparisonFloat() argument
63 CreateTensor(input1_data, input1_dims), in TestComparisonFloat()
72 int* input1_dims_data, bool* input1_data, in TestComparisonBool() argument
81 CreateTensor(input1_data, input1_dims), in TestComparisonBool()
90 int* input1_dims_data, int32_t* input1_data, in TestComparisonInt() argument
99 CreateTensor(input1_data, input1_dims), in TestComparisonInt()
108 int* input1_dims_data, float* input1_data, in TestComparisonQuantizedUInt8() argument
120 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestComparisonQuantizedUInt8()
131 int* input1_dims_data, float* input1_data, in TestComparisonQuantizedInt8() argument
143 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestComparisonQuantizedInt8()
[all …]
Dpack_test.cc49 const float* input1_data, in TestPackTwoInputsFloat() argument
63 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsFloat()
82 const int* input1_dims_data, const float* input1_data, in TestPackThreeInputsFloat() argument
96 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackThreeInputsFloat()
116 const uint8_t* input1_data, in TestPackTwoInputsQuantized() argument
134 CreateQuantizedTensor(input1_data, input1_dims, 1.0, 128), in TestPackTwoInputsQuantized()
153 const int32_t* input1_data, in TestPackTwoInputsQuantized32() argument
167 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsQuantized32()
Dmaximum_minimum_test.cc28 const int* input1_dims_data, const float* input1_data, in TestMaxMinFloat() argument
41 CreateTensor(input1_data, input1_dims), in TestMaxMinFloat()
65 const uint8_t* input1_data, float const input1_scale, in TestMaxMinQuantized() argument
82 CreateQuantizedTensor(input1_data, input1_dims, input1_scale, in TestMaxMinQuantized()
109 const int32_t* input1_data, const int* input2_dims_data, in TestMaxMinQuantizedInt32() argument
121 CreateTensor(input1_data, input1_dims), in TestMaxMinQuantizedInt32()
Dconcatenation_test.cc28 const float* input1_data, in TestConcatenateTwoInputs() argument
41 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestConcatenateTwoInputs()
71 const int* input1_dims_data, const uint8_t* input1_data, in TestConcatenateQuantizedTwoInputs() argument
85 CreateQuantizedTensor(input1_data, input1_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.cc70 void TestMulFloat(const int* input1_dims_data, const float* input1_data, in TestMulFloat() argument
83 CreateTensor(input1_data, input1_dims), in TestMulFloat()
93 void TestMulQuantized(const int* input1_dims_data, const float* input1_data, in TestMulQuantized() argument
110 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestMulQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dcomparisons.h59 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonImpl() argument
64 output_data[i] = F(input1_data[i], input2_data[i]); in ComparisonImpl()
71 const float* input1_data, in Comparison() argument
75 ComparisonImpl<float, F>(op_params, input1_shape, input1_data, input2_shape, in Comparison()
82 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonWithScaling() argument
95 const int32_t input1_val = input1_offset + input1_data[i]; in ComparisonWithScaling()
133 const RuntimeShape& unextended_input1_shape, const T* input1_data, in BroadcastComparison4DSlowImpl() argument
146 F(input1_data[SubscriptToIndex(dims.desc1, b, y, x, c)], in BroadcastComparison4DSlowImpl()
157 const float* input1_data, in BroadcastComparison4DSlow() argument
162 BroadcastComparison4DSlowImpl<float, F>(op_params, input1_shape, input1_data, in BroadcastComparison4DSlow()
[all …]
Dsub.h34 const float* input1_data, in SubNonBroadcast() argument
43 input1_data[i] - input2_data[i], params.float_activation_min, in SubNonBroadcast()
50 const int32_t* input1_data, in SubNonBroadcast() argument
59 input1_data[i] - input2_data[i], params.quantized_activation_min, in SubNonBroadcast()
71 const float* input1_data, in BroadcastSubSlow() argument
101 input1_data[SubscriptToIndex(desc1, indexes)] - in BroadcastSubSlow()
111 const uint8_t* input1_data, in BroadcastSubSlow() argument
140 params.input1_offset + input1_data[SubscriptToIndex(desc1, indexes)]; in BroadcastSubSlow()
168 const int32_t* input1_data, in BroadcastSubSlow() argument
198 input1_data[SubscriptToIndex(desc1, indexes)] - in BroadcastSubSlow()
[all …]
Dadd.h27 const RuntimeShape& input1_shape, const T* input1_data, in Add() argument
34 input1_data[i] + input2_data[i], params.quantized_activation_min, in Add()
40 const RuntimeShape& input1_shape, const float* input1_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
68 const int32_t input1_val = params.input1_offset + input1_data[i]; in AddElementwise()
94 uint8_t input1_data, const uint8_t* input2_data, in AddScalarBroadcast() argument
101 const int32_t input1_val = params.input1_offset + input1_data; in AddScalarBroadcast()
125 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Add() argument
137 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
[all …]
Dmul.h27 const uint8_t* input1_data, in MulElementwise() argument
30 const int32_t input1_val = params.input1_offset + input1_data[i]; in MulElementwise()
46 const RuntimeShape& input1_shape, const T* input1_data, in Mul() argument
57 input1_data[i] * input2_data[i], output_activation_min, in Mul()
63 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Mul() argument
71 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
76 const uint8_t* input1_data, in BroadcastMul4DSlow() argument
94 input1_data[SubscriptToIndex(desc1, b, y, x, c)]; in BroadcastMul4DSlow()
117 const T* input1_data, in BroadcastMul4DSlow() argument
154 input1_data[SubscriptToIndex(desc1, b, y, x, c)] * in BroadcastMul4DSlow()
Ddiv.h29 const uint8* input1_data, const uint8* input2_data, in DivElementwise() argument
39 const int32 input1_val = params.input1_offset + input1_data[i]; in DivElementwise()
62 const RuntimeShape& input1_shape, const uint8* input1_data, in Div() argument
70 DivElementwise(flat_size, params, input1_data, input2_data, output_data); in Div()
76 const uint8* input1_data, in BroadcastDivSlow() argument
102 params.input1_offset + input1_data[SubscriptToIndex(desc1, indexes)]; in BroadcastDivSlow()
134 const T* input1_data, in BroadcastDivSlow() argument
166 input1_data[SubscriptToIndex(desc1, indexes)] / in BroadcastDivSlow()
175 const RuntimeShape& input1_shape, const T* input1_data, in Div() argument
186 input1_data[i] / input2_data[i], output_activation_min, in Div()
Dlegacy_reference_ops.h622 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv() argument
629 BroadcastDivSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastDiv()
635 inline void Div(const T* input1_data, const Dims<4>& input1_dims, in Div() argument
642 Div(op_params, DimsToShape(input1_dims), input1_data, in Div()
979 inline void Comparison(const T* input1_data, const Dims<4>& input1_dims, in Comparison() argument
984 ComparisonImpl<T, F>(op_params, DimsToShape(input1_dims), input1_data, in Comparison()
990 inline void Comparison(int left_shift, const T* input1_data, in Comparison() argument
1008 ComparisonWithScaling<T, F>(op_params, DimsToShape(input1_dims), input1_data, in Comparison()
1014 inline void BroadcastComparison(const T* input1_data, in BroadcastComparison() argument
1022 input1_data, DimsToShape(input2_dims), in BroadcastComparison()
[all …]
Dbinary_function.h31 const RuntimeShape& unextended_input1_shape, const T1* input1_data, in BroadcastBinaryFunction4DSlow() argument
53 auto in1_val = input1_data[in1_idx]; in BroadcastBinaryFunction4DSlow()
65 const T1* input1_data, in BinaryFunction() argument
73 output_data[i] = func(input1_data[i], input2_data[i]); in BinaryFunction()
Dmaximum_minimum.h26 const T* input1_data, in MaximumMinimumBroadcastSlow() argument
37 output_data[i] = op(input1_data[i], input2_data[i]); in MaximumMinimumBroadcastSlow()
54 op(input1_data[SubscriptToIndex(desc1, indexes)], in MaximumMinimumBroadcastSlow()
Darg_min_max.h25 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax() argument
51 auto min_max_value = input1_data[outer * axis_size * inner_size + inner]; in ArgMinMax()
55 input1_data[(outer * axis_size + i) * inner_size + inner]; in ArgMinMax()
Dreference_ops.h185 const uint8* input1_data = in BroadcastMulFivefold() local
195 const uint8* input1_data_ptr = input1_data; in BroadcastMulFivefold()
221 const RuntimeShape& input1_shape, const int16* input1_data, in Mul() argument
234 F0::FromRaw(input1_data[i]) * F0::FromRaw(input2_data[i]); in Mul()
240 const RuntimeShape& input1_shape, const int16* input1_data, in Mul() argument
257 F0::FromRaw(input1_data[i]) * F0::FromRaw(input2_data[i]); in Mul()
269 const RuntimeShape& input1_shape, const int16_t* input1_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()
1422 void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum() argument
[all …]
Dbatch_to_space_nd.h40 const T* input1_data, in BatchToSpaceND() argument
91 input1_data + Offset(input1_shape, in_batch, in_h, in_w, 0); in BatchToSpaceND()
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dceva_tflm_lib.h90 void CEVA_TFLM_Add_Float32(const void* params_inp, const float* input1_data,
95 const float* input1_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,
111 const float* input1_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,
125 void CEVA_TFLM_Maximum_Float32(const float* input1_data,
128 void CEVA_TFLM_Minimum_Float32(const float* input1_data,
131 void CEVA_TFLM_Maximum_Float32_asm(const float* input1_data,
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dadd.h38 int size, const ArithmeticParams& params, const int8_t* input1_data, in ElementWise() argument
44 output_data[i] = binary_func(input1_data[i], input2_data[i], params); in ElementWise()
50 const int8_t* input1_data, const RuntimeShape& input2_shape, in BroadcastBinaryFunction4DSlow() argument
78 input1_data[SubscriptToIndex(desc1, b, y, x, c)], 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()
118 const RuntimeShape& input1_shape, const int8_t* input1_data, in Add() argument
126 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
131 const int8_t* input1_data, in BroadcastAdd4DSlow() argument
136 BroadcastBinaryFunction4DSlow(params, input1_shape, input1_data, input2_shape, in BroadcastAdd4DSlow()
Dmul.h27 const T* input1_data, const T* input2_data, in MulElementwise() argument
30 const int32_t input1_val = params.input1_offset + input1_data[i]; in MulElementwise()
46 const RuntimeShape& input1_shape, const T* input1_data, in Mul() argument
55 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
60 const RuntimeShape& input1_shape, const int16_t* input1_data, in Mul() argument
77 F0::FromRaw(input1_data[i]) * F0::FromRaw(input2_data[i]); in Mul()
91 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastMul4DSlow() argument
109 input1_data[SubscriptToIndex(desc1, b, y, x, c)]; in BroadcastMul4DSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dadd.h36 const int8* input1_data, const int8* input2_data, in AddElementwise() argument
60 const int8x16_t input1_val_original = vld1q_s8(input1_data + i); in AddElementwise()
144 const int32 input1_val = params.input1_offset + input1_data[i]; in AddElementwise()
170 int8 input1_data, const int8* input2_data, in AddScalarBroadcast() argument
190 const int8x8_t input1_val_original = vdup_n_s8(input1_data); in AddScalarBroadcast()
241 const int32 input1_val = params.input1_offset + input1_data; in AddScalarBroadcast()
268 const RuntimeShape& input1_shape, const int8* input1_data, in Add() argument
281 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
286 const int8* input1_data, in BroadcastAddDispatch() argument
293 params, input1_shape, input1_data, input2_shape, input2_data, in BroadcastAddDispatch()
[all …]
Dmul.h36 const int8* input1_data, const int8* input2_data, in MulElementwise() argument
59 const int8x16_t input1_val_original = vld1q_s8(input1_data + i); in MulElementwise()
126 const int32 input1_val = params.input1_offset + input1_data[i]; in MulElementwise()
233 const RuntimeShape& input1_shape, const int8* input1_data, in Mul() argument
242 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
247 const int8* input1_data, in BroadcastMulDispatch() argument
254 params, input1_shape, input1_data, input2_shape, input2_data, in BroadcastMulDispatch()
259 params, input1_shape, input1_data, input2_shape, input2_data, in BroadcastMulDispatch()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Darithmetic_test.cc34 const std::initializer_list<uint8_t>& input1_data, in ArithmeticOpBaseModel() argument
36 if (input1_data.size() > 0) in ArithmeticOpBaseModel()
37 input1_ = AddConstInput(input1, input1_data); in ArithmeticOpBaseModel()
83 const std::initializer_list<uint8_t>& input1_data, in AddOpModel() argument
86 : ArithmeticOpBaseModel(input1, input2, output, input1_data, input2_data), in AddOpModel()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dbinary_elementwise_tester.cc209 std::vector<uint16_t> input1_data(ComputeSize(Input1Shape())); in CreateTfLiteModel() local
210 std::generate(input1_data.begin(), input1_data.end(), in CreateTfLiteModel()
215 reinterpret_cast<const uint8_t*>(input1_data.data()), in CreateTfLiteModel()
216 sizeof(uint16_t) * input1_data.size()))); in CreateTfLiteModel()
218 std::vector<float> input1_data(ComputeSize(Input1Shape())); in CreateTfLiteModel() local
219 std::generate(input1_data.begin(), input1_data.end(), input1_rng); in CreateTfLiteModel()
226 reinterpret_cast<const uint8_t*>(input1_data.data()), in CreateTfLiteModel()
227 sizeof(float) * input1_data.size()))); in CreateTfLiteModel()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h227 const T* input1_data = in BinaryBroadcastFiveFold() local
237 const T* input1_data_ptr = input1_data; in BinaryBroadcastFiveFold()
1798 const float* input1_data, const float* input2_data, in AddElementwise() argument
1806 auto a10 = vld1q_f32(input1_data + i); in AddElementwise()
1807 auto a11 = vld1q_f32(input1_data + i + 4); in AddElementwise()
1808 auto a12 = vld1q_f32(input1_data + i + 8); in AddElementwise()
1809 auto a13 = vld1q_f32(input1_data + i + 12); in AddElementwise()
1832 auto a1 = vld1q_f32(input1_data + i); in AddElementwise()
1842 auto x = input1_data[i] + input2_data[i]; in AddElementwise()
1849 const RuntimeShape& input1_shape, const float* input1_data, in Add() argument
[all …]

12