| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | add_test.cc | 64 class FloatAddOpModel : public BaseAddOpModel { class
129 TEST(FloatAddOpModel, NoActivation) { in TEST() argument
130 FloatAddOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
139 TEST(FloatAddOpModel, ActivationRELU_N1_TO_1) { in TEST() argument
140 FloatAddOpModel m( in TEST()
149 TEST(FloatAddOpModel, VariousInputShapes) { in TEST() argument
153 FloatAddOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
165 TEST(FloatAddOpModel, WithBroadcast) { in TEST() argument
169 FloatAddOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
182 TEST(FloatAddOpModel, WithBroadcastGeneric) { in TEST() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_c_api_test.cc | 67 class FloatAddOpModel : public SingleOpModelWithNnapiDelegateCApi { class
69 FloatAddOpModel(const TensorData& input1, const TensorData& input2, in FloatAddOpModel() function in tflite::__anon51a303080111::FloatAddOpModel
75 FloatAddOpModel(const TfLiteNnapiDelegateOptions& options, in FloatAddOpModel() function in tflite::__anon51a303080111::FloatAddOpModel
112 FloatAddOpModel m(options, {TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
128 FloatAddOpModel m(options, {TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
147 FloatAddOpModel m(options, {TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
157 FloatAddOpModel m(options, {TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
|
| H A D | nnapi_delegate_errno_test.cc | 53 class FloatAddOpModel : public SingleOpModelWithNNAPI { class
55 FloatAddOpModel(const NnApi* nnapi, const TensorData& input1, in FloatAddOpModel() function in tflite::__anon206e91520111::FloatAddOpModel
91 FloatAddOpModel m(nnapi_mock_->GetNnApi(), {TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST_F()
104 FloatAddOpModel m(nnapi_mock_->GetNnApi(), {TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST_F()
118 FloatAddOpModel m(nnapi_mock_->GetNnApi(), {TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST_F()
135 FloatAddOpModel m(nnapi_mock_->GetNnApi(), {TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST_F()
162 FloatAddOpModel m(nnapi_mock_->GetNnApi(), {TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST_F()
|
| H A D | nnapi_delegate_test.cc | 165 class FloatAddOpModel : public SingleOpModelWithNNAPI { class
167 FloatAddOpModel(const TensorData& input1, const TensorData& input2, in FloatAddOpModel() function in tflite::__anon1a74836d0111::FloatAddOpModel
174 FloatAddOpModel(const StatefulNnApiDelegate::Options& options, in FloatAddOpModel() function in tflite::__anon1a74836d0111::FloatAddOpModel
210 FloatAddOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
221 FloatAddOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
234 FloatAddOpModel m( in TEST()
246 FloatAddOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
257 FloatAddOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
284 FloatAddOpModel m(options, in TEST()
353 FloatAddOpModel m(options, {TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
[all …]
|
| H A D | nnapi_delegate_device_selection_test.cc | 42 class FloatAddOpModel : public SingleOpModel { class
44 FloatAddOpModel() = default;
116 FloatAddOpModel m;
|