Home
last modified time | relevance | path

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

12

/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 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 …]
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()
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()
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
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()
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()
Dbinary_function.h32 const RuntimeShape& unextended_input2_shape, const T2* input2_data, in BroadcastBinaryFunction4DSlow()
78 const T2* input2_data, in BinaryFunction()
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 …]
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 …]
Dmaximum_minimum.h28 const T* input2_data, in MaximumMinimumBroadcastSlow()
Dbroadcast_args.h26 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastArgs()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
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()
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()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
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()
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.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()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Darithmetic_test.cc37 const std::initializer_list<uint8_t>& input2_data) { in ArithmeticOpBaseModel()
86 const std::initializer_list<uint8_t>& input2_data, in AddOpModel()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
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
/external/XNNPACK/src/subgraph/
Dmaximum2.c112 const void* input2_data = input2_blob->data; in setup_maximum_operator() local
Dsquared-difference.c112 const void* input2_data = input2_blob->data; in setup_squared_difference_operator() local
Dminimum2.c112 const void* input2_data = input2_blob->data; in setup_minimum_operator() local
Ddivide.c115 const void* input2_data = input2_blob->data; in setup_divide_operator() local
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
Dsubtract.c151 const void* input2_data = input2_blob->data; in setup_subtract_operator() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
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 …]

12