| 25 | namespace { |
| 26 | |
| 27 | void TestRound(std::initializer_list<int> input_dims_data, |
| 28 | std::initializer_list<float> input_data, |
| 29 | std::initializer_list<float> expected_output_data, |
| 30 | float* output_data) { |
| 31 | TfLiteIntArray* input_dims = IntArrayFromInitializer(input_dims_data); |
| 32 | TfLiteIntArray* output_dims = IntArrayFromInitializer(input_dims_data); |
| 33 | const int output_dims_count = ElementCount(*output_dims); |
| 34 | constexpr int inputs_size = 1; |
| 35 | constexpr int outputs_size = 1; |
| 36 | constexpr int tensors_size = inputs_size + outputs_size; |
| 37 | TfLiteTensor tensors[tensors_size] = { |
| 38 | CreateFloatTensor(input_data, input_dims, "input_tensor"), |
| 39 | CreateFloatTensor(output_data, output_dims, "output_tensor"), |
| 40 | }; |
| 41 | TfLiteContext context; |
| 42 | PopulateContext(tensors, tensors_size, &context); |
| 43 | ::tflite::ops::micro::AllOpsResolver resolver; |
| 44 | const TfLiteRegistration* registration = |
| 45 | resolver.FindOp(tflite::BuiltinOperator_ROUND, 1); |
| 46 | TF_LITE_MICRO_EXPECT_NE(nullptr, registration); |
| 47 | |
| 48 | int inputs_array_data[] = {1, 0}; |
| 49 | TfLiteIntArray* inputs_array = IntArrayFromInts(inputs_array_data); |
| 50 | int outputs_array_data[] = {1, 1}; |
| 51 | TfLiteIntArray* outputs_array = IntArrayFromInts(outputs_array_data); |
| 52 | TfLiteIntArray* temporaries_array = IntArrayFromInitializer({0}); |
| 53 | TfLiteNode node; |
| 54 | node.inputs = inputs_array; |
| 55 | node.outputs = outputs_array; |
| 56 | node.temporaries = temporaries_array; |
| 57 | node.user_data = nullptr; |
| 58 | node.builtin_data = nullptr; |
| 59 | node.custom_initial_data = nullptr; |
| 60 | node.custom_initial_data_size = 0; |
| 61 | node.delegate = nullptr; |
| 62 | TF_LITE_MICRO_EXPECT_NE(nullptr, registration->invoke); |
| 63 | TF_LITE_MICRO_EXPECT_EQ(kTfLiteOk, registration->invoke(&context, &node)); |
| 64 | for (int i = 0; i < output_dims_count; ++i) { |
| 65 | TF_LITE_MICRO_EXPECT_NEAR(expected_output_data.begin()[i], output_data[i], |
| 66 | 1e-5f); |
| 67 | } |
| 68 | } |
| 69 | |
| 70 | } // namespace |
| 71 | } // namespace testing |
no test coverage detected