Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A 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 int32_t* input2_data, in BroadcastSubSlow()
153 const int64_t* input2_data, in BroadcastSubSlow()
190 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastSubSlow()
229 const int16_t* input2_data, in BroadcastSub16POTSlow()
273 const T* input2_data, in BroadcastQuantSubSlow()
328 const T* input1_data, const T* input2_data, in SubElementwise()
355 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Sub()
[all …]
H A Ddiv.h46 const T* input1_data, const T* input2_data, in DivElementwise()
80 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Div()
92 const RuntimeShape& input2_shape, const int8_t* input2_data, in Div()
106 const T* input2_data, const RuntimeShape& unextended_output_shape, in BroadcastDivSlowQuantized()
159 const uint8_t* input2_data, in BroadcastDivSlow()
172 const int8_t* input2_data, in BroadcastDivSlow()
189 const T* input2_data, in BroadcastDivSlow()
229 const RuntimeShape& input2_shape, const T* input2_data, in Div()
H A Dmul.h31 const uint8_t* input2_data, uint8_t* output_data) { in MulElementwise()
50 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
69 const std::complex<float>* input2_data, in Mul()
81 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Mul()
95 const uint8_t* input2_data, in BroadcastMul4DSlow()
136 const T* input2_data, in BroadcastMul4DSlow()
184 const std::complex<float>* input2_data, in BroadcastMul4DSlow()
H A Dadd.h31 const RuntimeShape& input2_shape, const T* input2_data, in Add()
52 const T* input1_data, const T* input2_data, in AddElementwise()
86 uint8_t input1_data, const uint8_t* input2_data, in AddScalarBroadcast()
118 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Add()
136 const int16_t* input2_data, in AddGeneralParamScale()
201 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastAdd4DSlow()
246 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastAdd4DSlow()
327 const uint8_t* input2_data = in BroadcastAddFivefold() local
H A 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()
H A Dlegacy_reference_ops.h625 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv()
638 const T* input2_data, const Dims<4>& input2_dims, in Div()
983 const T* input2_data, const Dims<4>& input2_dims, in Comparison()
996 const T* input2_data, const Dims<4>& input2_dims, in Comparison()
1019 const T* input2_data, in BroadcastComparison()
1034 const T* input2_data, in BroadcastComparison()
1262 const uint8* input2_data, const Dims<4>& input2_dims, in Add()
1299 const int32* input2_data, const Dims<4>& input2_dims, in Add()
1316 const uint8* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
1355 const float* input2_data, const Dims<4>& input2_dims, in Add()
[all …]
H A Dreference_ops.h205 const uint8* input2_data = in BroadcastMulFivefold() local
240 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
259 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
288 const RuntimeShape& input2_shape, const int16_t* input2_data, in Sub16()
709 const T* input2_data, const RuntimeShape& output_shape, in Minimum()
723 const RuntimeShape&, const T* input2_data, in Minimum()
731 const T* input2_data, const RuntimeShape& output_shape, in Maximum()
745 const RuntimeShape&, const T* input2_data, in Maximum()
753 const T3* input2_data, const RuntimeShape& output_shape, in ArgMax()
763 const RuntimeShape& input2_shape, const T3* input2_data, in ArgMax()
[all …]
H A Darg_min_max.h37 const T3* input2_data, const RuntimeShape& output_shape, in ArgMinMax()
79 const T3* input2_data, const RuntimeShape& output_shape, in ArgMinMax()
H A Dbinary_function.h32 const RuntimeShape& unextended_input2_shape, const T2* input2_data, in BroadcastBinaryFunction4DSlow()
78 const T2* input2_data, in BinaryFunction()
H A Dbroadcast_args.h26 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastArgs()
H A Dmaximum_minimum.h28 const T* input2_data, in MaximumMinimumBroadcastSlow()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dadd.h40 const int8_t* input2_data, int8_t* output_data, in ElementWise()
52 const int8_t* input2_data, const RuntimeShape& output_shape, in BroadcastBinaryFunction4DSlow()
112 const int8_t* input1_data, const int8_t* input2_data, in AddElementwise()
120 const RuntimeShape& input2_shape, const int8_t* input2_data, in Add()
134 const int8_t* input2_data, in BroadcastAdd4DSlow()
H A Dmul.h29 const T* input1_data, const T* input2_data, in MulElementwise()
49 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
63 const RuntimeShape& input2_shape, const int16_t* input2_data, in Mul()
94 const T* input2_data, const RuntimeShape& output_shape, T* output_data) { in BroadcastMul4DSlow()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Dadd.h37 const int8* input1_data, const int8* input2_data, in AddElementwiseInt8()
170 const int16* input2_data, int16* output_data) { in AddElementwiseInt16()
361 int8 input1_data, const int8* input2_data, in AddScalarBroadcast()
460 const RuntimeShape& input2_shape, const int8* input2_data, in Add()
477 const RuntimeShape& input2_shape, const int16* input2_data, in Add()
496 const int8* input2_data, in BroadcastAddDispatch()
H A Dsub.h32 const int16* input2_data, int16* output_data) { in SubElementwiseInt16()
149 const int16_t* input2_data = in BroadcastSubFiveFold() local
195 const RuntimeShape& input2_shape, const int16* input2_data, in Sub()
214 const int16* input2_data, in BroadcastSubDispatch()
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h165 const T* input2_data = in BinaryBroadcastFiveFold() local
1467 const float* input1_data, const float* input2_data, in AddElementwise()
1519 const RuntimeShape& input2_shape, const float* input2_data, in Add()
1530 const uint8* input1_data, const uint8* input2_data, in AddElementwise()
1622 uint8 input1_data, const uint8* input2_data, in AddScalarBroadcast()
1725 float broadcast_value, const float* input2_data, in AddScalarBroadcast()
1756 const RuntimeShape& input2_shape, const uint8* input2_data, in Add()
1773 const RuntimeShape& input2_shape, const int16* input2_data, in Add()
1812 const T* input2_data, const RuntimeShape& output_shape, T* output_data) { in Add()
1846 const T* input2_data, const RuntimeShape& output_shape, T* output_data) { in BroadcastAddDispatch()
[all …]
H A Dlegacy_optimized_ops.h3438 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv()
3495 const float* input2_data, const Dims<4>& input2_dims, in Add()
3512 const uint8* input2_data, const Dims<4>& input2_dims, in Add()
3549 const int32* input2_data, const Dims<4>& input2_dims, in Add()
3564 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
3579 const uint8* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
3620 int32 input1_multiplier, int input1_shift, const uint8* input2_data, in BroadcastAddFivefold()
3664 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
3676 int input1_shift, const int16* input2_data, in Add()
3703 const float* input2_data, const Dims<4>& input2_dims, in Sub()
[all …]
/aosp_15_r20/external/XNNPACK/src/subgraph/
H A Dconcatenate.c296 const void* input2_data = input2_blob->data; in setup_concatenate2_operator() local
339 const void* input2_data = input2_blob->data; in setup_concatenate3_operator() local
394 const void* input2_data = input2_blob->data; in setup_concatenate4_operator() local
H A Dmaximum2.c112 const void* input2_data = input2_blob->data; in setup_maximum_operator() local
H A Dsquared-difference.c112 const void* input2_data = input2_blob->data; in setup_squared_difference_operator() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Darithmetic_test.cc37 const std::initializer_list<uint8_t>& input2_data) { in ArithmeticOpBaseModel()
86 const std::initializer_list<uint8_t>& input2_data, in AddOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dbinary_elementwise_tester.cc280 std::vector<uint16_t> input2_data(ComputeSize(Input2Shape())); in CreateTfLiteModel() local
289 std::vector<float> input2_data(ComputeSize(Input2Shape())); in CreateTfLiteModel() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dselect_v1_2.mod.py16 def test(name, input0, input1, input2, output0, input0_data, input1_data, input2_data, output_data): argument
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
Dselect_quant8_signed.mod.py16 def test(name, input0, input1, input2, output0, input0_data, input1_data, input2_data, output_data): argument

12