1/*
2 * Copyright (c) 2022 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#ifndef OHOS_HDI_NNRT_V1_0_IPREPAREDMODEL_H
17#define OHOS_HDI_NNRT_V1_0_IPREPAREDMODEL_H
18
19#include <stdint.h>
20#include <vector>
21#include <hdf_base.h>
22#include <hdi_base.h>
23#include "nnrt/v1_0/nnrt_types.h"
24
25#ifndef HDI_BUFF_MAX_SIZE
26#define HDI_BUFF_MAX_SIZE (1024 * 200)
27#endif
28
29#ifndef HDI_CHECK_VALUE_RETURN
30#define HDI_CHECK_VALUE_RETURN(lv, compare, rv, ret) do { \
31    if ((lv) compare (rv)) { \
32        return ret; \
33    } \
34} while (false)
35#endif
36
37#ifndef HDI_CHECK_VALUE_RET_GOTO
38#define HDI_CHECK_VALUE_RET_GOTO(lv, compare, rv, ret, value, table) do { \
39    if ((lv) compare (rv)) { \
40        ret = value; \
41        goto table; \
42    } \
43} while (false)
44#endif
45
46namespace OHOS {
47namespace HDI {
48namespace Nnrt {
49namespace V1_0 {
50using namespace OHOS;
51using namespace OHOS::HDI;
52
53enum {
54    CMD_PREPARED_MODEL_GET_VERSION = 0,
55    CMD_PREPARED_MODEL_EXPORT_MODEL_CACHE = 1,
56    CMD_PREPARED_MODEL_RUN = 2,
57};
58
59class IPreparedModel : public HdiBase {
60public:
61    DECLARE_HDI_DESCRIPTOR(u"ohos.hdi.nnrt.v1_0.IPreparedModel");
62
63    virtual ~IPreparedModel() = default;
64
65    virtual int32_t ExportModelCache(std::vector<OHOS::HDI::Nnrt::V1_0::SharedBuffer>& modelCache) = 0;
66
67    virtual int32_t Run(const std::vector<OHOS::HDI::Nnrt::V1_0::IOTensor>& inputs,
68         const std::vector<OHOS::HDI::Nnrt::V1_0::IOTensor>& outputs, std::vector<std::vector<int32_t>>& outputsDims, std::vector<bool>& isOutputBufferEnough) = 0;
69
70    virtual int32_t GetVersion(uint32_t& majorVer, uint32_t& minorVer)
71    {
72        majorVer = 1;
73        minorVer = 0;
74        return HDF_SUCCESS;
75    }
76
77    virtual bool IsProxy()
78    {
79        return false;
80    }
81
82    virtual const std::u16string GetDesc()
83    {
84        return metaDescriptor_;
85    }
86};
87} // V1_0
88} // Nnrt
89} // HDI
90} // OHOS
91
92#endif // OHOS_HDI_NNRT_V1_0_IPREPAREDMODEL_H
93
94