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