Home
last modified time | relevance | path

Searched defs:input2_data (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/lite/micro/kernels/
Dcomparisons_test.cc55 int* input2_dims_data, float* input2_data, in TestComparisonFloat()
73 int* input2_dims_data, bool* input2_data, in TestComparisonBool()
91 int* input2_dims_data, int32_t* input2_data, in TestComparisonInt()
111 float* input2_data, uint8_t* input2_quantized, in TestComparisonQuantizedUInt8()
134 float* input2_data, int8_t* input2_quantized, in TestComparisonQuantizedInt8()
164 bool input2_data[] = {true, true, false, false}; in TF_LITE_MICRO_TEST() local
180 float input2_data[] = {0.1, 0.2, 0.6, 0.5}; in TF_LITE_MICRO_TEST() local
196 int32_t input2_data[] = {1, 2, 7, 5}; in TF_LITE_MICRO_TEST() local
211 int32_t input2_data[] = {7}; in TF_LITE_MICRO_TEST() local
227 int32_t input2_data[] = {7, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
[all …]
Dmaximum_minimum_test.cc29 const int* input2_dims_data, const float* input2_data, in TestMaxMinFloat()
68 const uint8_t* input2_data, const float input2_scale, in TestMaxMinQuantized()
110 const int32_t* input2_data, const int32_t* expected_output_data, in TestMaxMinQuantizedInt32()
Dpack_test.cc51 const float* input2_data, int axis, in TestPackTwoInputsFloat()
83 const int* input2_dims_data, const float* input2_data, in TestPackThreeInputsFloat()
118 const uint8_t* input2_data, int axis, in TestPackTwoInputsQuantized()
155 const int32_t* input2_data, int axis, in TestPackTwoInputsQuantized32()
Dconcatenation_test.cc30 const float* input2_data, int axis, in TestConcatenateTwoInputs()
72 const int* input2_dims_data, const uint8_t* input2_data, in TestConcatenateQuantizedTwoInputs()
Darg_min_max.cc36 const T1* input1_data, const T3* input2_data, in ArgMinMaxHelper()
Dmul_test.cc71 const int* input2_dims_data, const float* input2_data, in TestMulFloat()
95 const float* input2_data, T* input2_quantized, in TestMulQuantized()
Dprelu.cc59 const RuntimeShape& unextended_input2_shape, const float* input2_data, in BroadcastPrelu4DSlowFloat()
Dlogical_test.cc29 const int* input2_dims_data, const bool* input2_data, in TestLogicalOp()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsub.h36 const float* input2_data, in SubNonBroadcast()
52 const int32_t* input2_data, in SubNonBroadcast()
73 const float* input2_data, in BroadcastSubSlow()
113 const uint8_t* input2_data, in BroadcastSubSlow()
170 const int32_t* input2_data, in BroadcastSubSlow()
210 const int8_t* input2_data, in BroadcastSubSlow()
264 const int64_t* input2_data, in BroadcastSubSlow()
301 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastSubSlow()
340 const int16_t* input2_data, in BroadcastSub16POTSlow()
383 const uint8_t* input2_data, uint8_t* output_data) { in SubElementwise()
[all …]
Dadd.h28 const RuntimeShape& input2_shape, const T* input2_data, in Add()
41 const RuntimeShape& input2_shape, const float* input2_data, in Add()
60 const T* input1_data, const T* input2_data, in AddElementwise()
94 uint8_t input1_data, const uint8_t* input2_data, in AddScalarBroadcast()
126 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Add()
144 const int16_t* input2_data, in AddGeneralParamScale()
209 const float* input2_data, in BroadcastAdd4DSlow()
249 const int32_t* input2_data, in BroadcastAdd4DSlow()
293 const T* input2_data, const RuntimeShape& output_shape, T* output_data) { in BroadcastAdd4DSlow()
373 const uint8_t* input2_data = in BroadcastAddFivefold() local
Dmul.h28 const uint8_t* input2_data, uint8_t* output_data) { in MulElementwise()
47 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
64 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Mul()
78 const uint8_t* input2_data, in BroadcastMul4DSlow()
119 const T* input2_data, in BroadcastMul4DSlow()
Ddiv.h29 const uint8* input1_data, const uint8* input2_data, in DivElementwise()
63 const RuntimeShape& input2_shape, const uint8* input2_data, in Div()
78 const uint8* input2_data, in BroadcastDivSlow()
136 const T* input2_data, in BroadcastDivSlow()
176 const RuntimeShape& input2_shape, const T* input2_data, in Div()
Dcomparisons.h60 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonImpl()
73 const float* input2_data, in Comparison()
83 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonWithScaling()
134 const RuntimeShape& unextended_input2_shape, const T* input2_data, in BroadcastComparison4DSlowImpl()
159 const float* input2_data, in BroadcastComparison4DSlow()
171 const RuntimeShape& unextended_input2_shape, const T* input2_data, in BroadcastComparison4DSlowWithScaling()
Dbinary_function.h32 const RuntimeShape& unextended_input2_shape, const T2* input2_data, in BroadcastBinaryFunction4DSlow()
67 const T2* input2_data, in BinaryFunction()
Dmaximum_minimum.h28 const T* input2_data, in MaximumMinimumBroadcastSlow()
Dlegacy_reference_ops.h623 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv()
636 const T* input2_data, const Dims<4>& input2_dims, in Div()
980 const T* input2_data, const Dims<4>& input2_dims, in Comparison()
993 const T* input2_data, const Dims<4>& input2_dims, in Comparison()
1016 const T* input2_data, in BroadcastComparison()
1031 const T* input2_data, in BroadcastComparison()
1259 const uint8* input2_data, const Dims<4>& input2_dims, in Add()
1296 const int32* input2_data, const Dims<4>& input2_dims, in Add()
1313 const uint8* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
1352 const float* input2_data, const Dims<4>& input2_dims, in Add()
[all …]
Darg_min_max.h26 const T3* input2_data, const RuntimeShape& output_shape, in ArgMinMax()
Dreference_ops.h187 const uint8* input2_data = in BroadcastMulFivefold() local
222 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
241 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
270 const RuntimeShape& input2_shape, const int16_t* input2_data, in Sub16()
1423 const T* input2_data, const RuntimeShape& output_shape, in Minimum()
1437 const RuntimeShape&, const T* input2_data, in Minimum()
1445 const T* input2_data, const RuntimeShape& output_shape, in Maximum()
1459 const RuntimeShape&, const T* input2_data, in Maximum()
1467 const T3* input2_data, const RuntimeShape& output_shape, in ArgMax()
1477 const RuntimeShape& input2_shape, const T3* input2_data, in ArgMax()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dadd.h39 const int8_t* input2_data, int8_t* output_data, in ElementWise()
51 const int8_t* input2_data, const RuntimeShape& output_shape, in BroadcastBinaryFunction4DSlow()
111 const int8_t* input1_data, const int8_t* input2_data, in AddElementwise()
119 const RuntimeShape& input2_shape, const int8_t* input2_data, in Add()
133 const int8_t* input2_data, in BroadcastAdd4DSlow()
Dmul.h27 const T* input1_data, const T* input2_data, in MulElementwise()
47 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
61 const RuntimeShape& input2_shape, const int16_t* input2_data, in Mul()
92 const T* input2_data, const RuntimeShape& output_shape, T* output_data) { in BroadcastMul4DSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmul.h36 const int8* input1_data, const int8* input2_data, in MulElementwise()
143 const int8* input2_data, int8* output_data) { in MulSimpleBroadcast()
234 const RuntimeShape& input2_shape, const int8* input2_data, in Mul()
249 const int8* input2_data, in BroadcastMulDispatch()
Dadd.h36 const int8* input1_data, const int8* input2_data, in AddElementwise()
170 int8 input1_data, const int8* input2_data, in AddScalarBroadcast()
269 const RuntimeShape& input2_shape, const int8* input2_data, in Add()
288 const int8* input2_data, in BroadcastAddDispatch()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dbinary_elementwise_tester.cc234 std::vector<uint16_t> input2_data(ComputeSize(Input2Shape())); in CreateTfLiteModel() local
243 std::vector<float> input2_data(ComputeSize(Input2Shape())); in CreateTfLiteModel() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Darithmetic_test.cc35 const std::initializer_list<uint8_t>& input2_data) { in ArithmeticOpBaseModel()
84 const std::initializer_list<uint8_t>& input2_data, in AddOpModel()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h229 const T* input2_data = in BinaryBroadcastFiveFold() local
1798 const float* input1_data, const float* input2_data, in AddElementwise()
1850 const RuntimeShape& input2_shape, const float* input2_data, in Add()
1861 const uint8* input1_data, const uint8* input2_data, in AddElementwise()
1953 uint8 input1_data, const uint8* input2_data, in AddScalarBroadcast()
2056 float broadcast_value, const float* input2_data, in AddScalarBroadcast()
2087 const RuntimeShape& input2_shape, const uint8* input2_data, in Add()
2104 const RuntimeShape& input2_shape, const int16* input2_data, in Add()
2141 const RuntimeShape& input2_shape, const int32* input2_data, in Add()
2173 const T* input2_data, const RuntimeShape& output_shape, T* output_data) { in BroadcastAddDispatch()
[all …]

12