1/*
2 * Copyright (c) 2023 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_V2_0_IPREPAREDMODEL_H
17#define OHOS_HDI_NNRT_V2_0_IPREPAREDMODEL_H
18
19#include <stdint.h>
20#include <vector>
21#include <hdf_base.h>
22#include <hdi_base.h>
23#include "nnrt/v2_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 V2_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_GET_INPUT_DIM_RANGES = 2,
57    CMD_PREPARED_MODEL_RUN = 3,
58};
59
60class IPreparedModel : public HdiBase {
61public:
62    DECLARE_HDI_DESCRIPTOR(u"ohos.hdi.nnrt.v2_0.IPreparedModel");
63
64    virtual ~IPreparedModel() = default;
65
66    virtual int32_t ExportModelCache(std::vector<OHOS::HDI::Nnrt::V2_0::SharedBuffer>& modelCache) = 0;
67
68    virtual int32_t GetInputDimRanges(std::vector<std::vector<uint32_t>>& minInputDims,
69         std::vector<std::vector<uint32_t>>& maxInputDims) = 0;
70
71    virtual int32_t Run(const std::vector<OHOS::HDI::Nnrt::V2_0::IOTensor>& inputs,
72         const std::vector<OHOS::HDI::Nnrt::V2_0::IOTensor>& outputs, std::vector<std::vector<int32_t>>& outputDims) = 0;
73
74    virtual int32_t GetVersion(uint32_t& majorVer, uint32_t& minorVer)
75    {
76        majorVer = 2;
77        minorVer = 0;
78        return HDF_SUCCESS;
79    }
80
81    virtual bool IsProxy()
82    {
83        return false;
84    }
85
86    virtual const std::u16string GetDesc()
87    {
88        return metaDescriptor_;
89    }
90};
91} // V2_0
92} // Nnrt
93} // HDI
94} // OHOS
95
96#endif // OHOS_HDI_NNRT_V2_0_IPREPAREDMODEL_H
97
98