| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | comparisons_test.cc | 54 int* input1_dims_data, float* input1_data, in TestComparisonFloat() 72 int* input1_dims_data, bool* input1_data, in TestComparisonBool() 90 int* input1_dims_data, int32_t* input1_data, in TestComparisonInt() 108 int* input1_dims_data, float* input1_data, in TestComparisonQuantizedUInt8() 131 int* input1_dims_data, float* input1_data, in TestComparisonQuantizedInt8() 163 bool input1_data[] = {true, false, true, false}; in TF_LITE_MICRO_TEST() local 179 float input1_data[] = {0.1, 0.9, 0.7, 0.3}; in TF_LITE_MICRO_TEST() local 195 int32_t input1_data[] = {-1, 9, 7, 3}; in TF_LITE_MICRO_TEST() local 210 int32_t input1_data[] = {-1, 9, 7, 3}; in TF_LITE_MICRO_TEST() local 226 int32_t input1_data[] = {-1, 9, 7, 3, 2, 4, 2, 8}; in TF_LITE_MICRO_TEST() local [all …]
|
| D | maximum_minimum_test.cc | 28 const int* input1_dims_data, const float* input1_data, in TestMaxMinFloat() 65 const uint8_t* input1_data, float const input1_scale, in TestMaxMinQuantized() 109 const int32_t* input1_data, const int* input2_dims_data, in TestMaxMinQuantizedInt32()
|
| D | pack_test.cc | 49 const float* input1_data, in TestPackTwoInputsFloat() 82 const int* input1_dims_data, const float* input1_data, in TestPackThreeInputsFloat() 116 const uint8_t* input1_data, in TestPackTwoInputsQuantized() 153 const int32_t* input1_data, in TestPackTwoInputsQuantized32()
|
| D | concatenation_test.cc | 28 const float* input1_data, in TestConcatenateTwoInputs() 71 const int* input1_dims_data, const uint8_t* input1_data, in TestConcatenateQuantizedTwoInputs()
|
| D | arg_min_max.cc | 36 const T1* input1_data, const T3* input2_data, in ArgMinMaxHelper()
|
| D | mul_test.cc | 70 void TestMulFloat(const int* input1_dims_data, const float* input1_data, in TestMulFloat() 93 void TestMulQuantized(const int* input1_dims_data, const float* input1_data, in TestMulQuantized()
|
| D | prelu.cc | 58 const RuntimeShape& unextended_input1_shape, const float* input1_data, in BroadcastPrelu4DSlowFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | sub.h | 34 const float* input1_data, in SubNonBroadcast() 50 const int32_t* input1_data, in SubNonBroadcast() 71 const float* input1_data, in BroadcastSubSlow() 111 const uint8_t* input1_data, in BroadcastSubSlow() 168 const int32_t* input1_data, in BroadcastSubSlow() 208 const int8_t* input1_data, in BroadcastSubSlow() 262 const int64_t* input1_data, in BroadcastSubSlow() 300 const RuntimeShape& input1_shape, const T* input1_data, in BroadcastSubSlow() 338 const int16_t* input1_data, in BroadcastSub16POTSlow() 382 const uint8_t* input1_data, in SubElementwise() [all …]
|
| D | add.h | 27 const RuntimeShape& input1_shape, const T* input1_data, in Add() 40 const RuntimeShape& input1_shape, const float* input1_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() 125 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Add() 142 const int16_t* input1_data, in AddGeneralParamScale() 207 const float* input1_data, in BroadcastAdd4DSlow() 247 const int32_t* input1_data, in BroadcastAdd4DSlow() 292 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastAdd4DSlow() 371 const uint8_t* input1_data = in BroadcastAddFivefold() local
|
| D | mul.h | 27 const uint8_t* input1_data, in MulElementwise() 46 const RuntimeShape& input1_shape, const T* input1_data, in Mul() 63 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Mul() 76 const uint8_t* input1_data, in BroadcastMul4DSlow() 117 const T* input1_data, in BroadcastMul4DSlow()
|
| D | div.h | 29 const uint8* input1_data, const uint8* input2_data, in DivElementwise() 62 const RuntimeShape& input1_shape, const uint8* input1_data, in Div() 76 const uint8* input1_data, in BroadcastDivSlow() 134 const T* input1_data, in BroadcastDivSlow() 175 const RuntimeShape& input1_shape, const T* input1_data, in Div()
|
| D | comparisons.h | 59 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonImpl() 71 const float* input1_data, in Comparison() 82 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonWithScaling() 133 const RuntimeShape& unextended_input1_shape, const T* input1_data, in BroadcastComparison4DSlowImpl() 157 const float* input1_data, in BroadcastComparison4DSlow() 170 const RuntimeShape& unextended_input1_shape, const T* input1_data, in BroadcastComparison4DSlowWithScaling()
|
| D | binary_function.h | 31 const RuntimeShape& unextended_input1_shape, const T1* input1_data, in BroadcastBinaryFunction4DSlow() 65 const T1* input1_data, in BinaryFunction()
|
| D | maximum_minimum.h | 26 const T* input1_data, in MaximumMinimumBroadcastSlow()
|
| D | legacy_reference_ops.h | 622 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv() 635 inline void Div(const T* input1_data, const Dims<4>& input1_dims, in Div() 979 inline void Comparison(const T* input1_data, const Dims<4>& input1_dims, in Comparison() 990 inline void Comparison(int left_shift, const T* input1_data, in Comparison() 1014 inline void BroadcastComparison(const T* input1_data, in BroadcastComparison() 1028 inline void BroadcastComparison(int left_shift, const T* input1_data, in BroadcastComparison() 1256 inline void Add(int left_shift, const uint8* input1_data, in Add() 1295 void Add(const int32* input1_data, const Dims<4>& input1_dims, in Add() 1310 inline void BroadcastAdd(int left_shift, const uint8* input1_data, in BroadcastAdd() 1351 void Add(const float* input1_data, const Dims<4>& input1_dims, in Add() [all …]
|
| D | arg_min_max.h | 25 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax()
|
| D | batch_to_space_nd.h | 40 const T* input1_data, in BatchToSpaceND()
|
| D | space_to_batch_nd.h | 42 const T* input1_data, in SpaceToBatchND()
|
| D | reference_ops.h | 185 const uint8* input1_data = in BroadcastMulFivefold() local 221 const RuntimeShape& input1_shape, const int16* input1_data, in Mul() 240 const RuntimeShape& input1_shape, const int16* input1_data, in Mul() 269 const RuntimeShape& input1_shape, const int16_t* input1_data, in Sub16() 1422 void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum() 1436 inline void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum() 1444 void Maximum(const RuntimeShape& input1_shape, const T* input1_data, in Maximum() 1458 inline void Maximum(const RuntimeShape& input1_shape, const T* input1_data, in Maximum() 1466 void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() 1476 inline void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() [all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | add.h | 38 int size, const ArithmeticParams& params, const int8_t* input1_data, in ElementWise() 50 const int8_t* input1_data, const RuntimeShape& input2_shape, in BroadcastBinaryFunction4DSlow() 111 const int8_t* input1_data, const int8_t* input2_data, in AddElementwise() 118 const RuntimeShape& input1_shape, const int8_t* input1_data, in Add() 131 const int8_t* input1_data, in BroadcastAdd4DSlow()
|
| D | mul.h | 27 const T* input1_data, const T* input2_data, in MulElementwise() 46 const RuntimeShape& input1_shape, const T* input1_data, in Mul() 60 const RuntimeShape& input1_shape, const int16_t* input1_data, in Mul() 91 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | add.h | 36 const int8* input1_data, const int8* input2_data, in AddElementwise() 170 int8 input1_data, const int8* input2_data, in AddScalarBroadcast() 268 const RuntimeShape& input1_shape, const int8* input1_data, in Add() 286 const int8* input1_data, in BroadcastAddDispatch()
|
| D | mul.h | 36 const int8* input1_data, const int8* input2_data, in MulElementwise() 233 const RuntimeShape& input1_shape, const int8* input1_data, in Mul() 247 const int8* input1_data, in BroadcastMulDispatch()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | binary_elementwise_tester.cc | 209 std::vector<uint16_t> input1_data(ComputeSize(Input1Shape())); in CreateTfLiteModel() local 218 std::vector<float> input1_data(ComputeSize(Input1Shape())); in CreateTfLiteModel() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | arithmetic_test.cc | 34 const std::initializer_list<uint8_t>& input1_data, in ArithmeticOpBaseModel() 83 const std::initializer_list<uint8_t>& input1_data, in AddOpModel()
|