• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /**
2  * Copyright 2021 Huawei Technologies Co., Ltd
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef MINDSPORE_LITE_TOOLS_CONVERTER_CONFIG_PARSER_ACL_OPTION_PARAM_PARSER_H_
18 #define MINDSPORE_LITE_TOOLS_CONVERTER_CONFIG_PARSER_ACL_OPTION_PARAM_PARSER_H_
19 #include <string>
20 #include "tools/converter/config_parser/config_file_parser.h"
21 #include "tools/converter/adapter/acl/common/acl_types.h"
22 #include "include/errorcode.h"
23 
24 namespace mindspore {
25 namespace lite {
26 class AclOptionParamParser {
27  public:
28   STATUS ParseAclOptionCfg(const AclOptionCfgString &acl_option_string, acl::AclModelOptionCfg *acl_option_cfg);
29 
30  private:
31   STATUS ParseCommon(const AclOptionCfgString &acl_option_string, acl::AclModelOptionCfg *acl_option_cfg);
32   STATUS ParseDeviceId(const std::string &device_id, acl::AclModelOptionCfg *acl_option_cfg);
33   STATUS ParseOutputType(const std::string &output_type, acl::AclModelOptionCfg *acl_option_cfg);
34   STATUS ParseDynamicBatchSize(const std::string &dynamic_size, acl::AclModelOptionCfg *acl_option_cfg);
35   STATUS ParseInputShapeVector(const std::string &input_shape_vector, acl::AclModelOptionCfg *acl_option_cfg);
36 };
37 }  // namespace lite
38 }  // namespace mindspore
39 
40 #endif  // MINDSPORE_LITE_TOOLS_CONVERTER_CONFIG_PARSER_ACL_OPTION_PARAM_PARSER_H_
41