Searched refs:FloatAddOpModel (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_test.cc | 48 class FloatAddOpModel : public BaseAddOpModel { class
89 TEST(FloatAddOpModel, NoActivation) { in TEST() argument
90 FloatAddOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
99 TEST(FloatAddOpModel, ActivationRELU_N1_TO_1) { in TEST() argument
100 FloatAddOpModel m( in TEST()
109 TEST(FloatAddOpModel, VariousInputShapes) { in TEST() argument
113 FloatAddOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
125 TEST(FloatAddOpModel, WithBroadcast) { in TEST() argument
129 FloatAddOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 89 class FloatAddOpModel : public SingleOpModelWithNNAPI { class
91 FloatAddOpModel(const TensorData& input1, const TensorData& input2, in FloatAddOpModel() function in tflite::__anon0c68fcb70111::FloatAddOpModel
117 FloatAddOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
130 FloatAddOpModel m( in TEST()
141 FloatAddOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
153 FloatAddOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
|