| 80 | |
| 81 | volatile bool start_checked = false; |
| 82 | int start_callback(const LiteIO* inputs, const LiteTensor* input_tensors, size_t size) { |
| 83 | start_checked = true; |
| 84 | auto check_func = [&]() { |
| 85 | ASSERT_EQ(size, 1); |
| 86 | ASSERT_EQ(std::string(inputs->name), "data"); |
| 87 | LiteLayout layout; |
| 88 | LITE_get_tensor_layout(*input_tensors, &layout); |
| 89 | ASSERT_EQ(layout.ndim, 4); |
| 90 | ASSERT_EQ(layout.shapes[1], 3); |
| 91 | ASSERT_EQ(layout.shapes[2], 224); |
| 92 | ASSERT_EQ(layout.shapes[3], 224); |
| 93 | }; |
| 94 | check_func(); |
| 95 | return 0; |
| 96 | } |
| 97 | |
| 98 | volatile bool start_checked_with_data = false; |
| 99 | int start_callback_with_data( |
no test coverage detected