Marat Dukhan | c068bb6 | 2019-10-04 13:24:39 -0700 | [diff] [blame] | 1 | // Copyright 2019 Google LLC |
| 2 | // |
| 3 | // This source code is licensed under the BSD-style license found in the |
| 4 | // LICENSE file in the root directory of this source tree. |
| 5 | |
| 6 | #pragma once |
| 7 | |
| 8 | #include <xnnpack.h> |
| 9 | |
| 10 | #include <memory> |
| 11 | #include <vector> |
| 12 | |
| 13 | namespace models { |
| 14 | |
| 15 | typedef std::vector<std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)>> ExecutionPlan; |
Marat Dukhan | 5f18d26 | 2019-10-31 10:24:14 -0700 | [diff] [blame] | 16 | typedef ExecutionPlan (*ExecutionPlanFactory)(pthreadpool_t threadpool); |
Marat Dukhan | c068bb6 | 2019-10-04 13:24:39 -0700 | [diff] [blame] | 17 | |
| 18 | ExecutionPlan MobileNetV1(pthreadpool_t threadpool); |
| 19 | ExecutionPlan MobileNetV2(pthreadpool_t threadpool); |
Marat Dukhan | c08cdf5 | 2019-12-09 09:17:51 -0800 | [diff] [blame] | 20 | ExecutionPlan MobileNetV3Large(pthreadpool_t threadpool); |
| 21 | ExecutionPlan MobileNetV3Small(pthreadpool_t threadpool); |
Marat Dukhan | c068bb6 | 2019-10-04 13:24:39 -0700 | [diff] [blame] | 22 | |
| 23 | } // namespace models |