Searched refs:FloatSubOpModel (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | sub_test.cc | 48 class FloatSubOpModel : public BaseSubOpModel { class
89 TEST(FloatSubOpModel, NoActivation) { in TEST() argument
90 FloatSubOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
100 TEST(FloatSubOpModel, ActivationRELU_N1_TO_1) { in TEST() argument
101 FloatSubOpModel m( in TEST()
111 TEST(FloatSubOpModel, VariousInputShapes) { in TEST() argument
115 FloatSubOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
128 TEST(FloatSubOpModel, WithBroadcast) { in TEST() argument
132 FloatSubOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 835 class FloatSubOpModel : public SingleOpModelWithNNAPI { class
837 FloatSubOpModel(const TensorData& input1, const TensorData& input2, in FloatSubOpModel() function in tflite::__anon0c68fcb70111::FloatSubOpModel
860 FloatSubOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
|