Browse Source

!646 parser 仓对外头文件回退

Merge pull request !646 from xujiuxu/master
pull/644/MERGE
xujiuxu i-robot 3 years ago
parent
commit
fa668f39ab
5 changed files with 42 additions and 39 deletions
  1. +0
    -1
      CMakeLists.txt
  2. +14
    -1
      inc/external/parser/caffe_parser.h
  3. +14
    -1
      inc/external/parser/onnx_parser.h
  4. +0
    -34
      inc/external/parser/parser_common.h
  5. +14
    -2
      inc/external/parser/tensorflow_parser.h

+ 0
- 1
CMakeLists.txt View File

@@ -253,7 +253,6 @@ install(TARGETS _caffe_parser parser_common fmk_onnx_parser fmk_parser parser_he
install(FILES ${PARSER_DIR}/inc/external/parser/onnx_parser.h install(FILES ${PARSER_DIR}/inc/external/parser/onnx_parser.h
${PARSER_DIR}/inc/external/parser/caffe_parser.h ${PARSER_DIR}/inc/external/parser/caffe_parser.h
${PARSER_DIR}/inc/external/parser/tensorflow_parser.h ${PARSER_DIR}/inc/external/parser/tensorflow_parser.h
${PARSER_DIR}/inc/external/parser/parser_common.h
DESTINATION ${INSTALL_INCLUDE_DIR}/parser/external/parser COMPONENT opensdk EXCLUDE_FROM_ALL DESTINATION ${INSTALL_INCLUDE_DIR}/parser/external/parser COMPONENT opensdk EXCLUDE_FROM_ALL
) )




+ 14
- 1
inc/external/parser/caffe_parser.h View File

@@ -17,6 +17,20 @@
#ifndef INC_EXTERNAL_ACL_GRAPH_CAFFE_H_ #ifndef INC_EXTERNAL_ACL_GRAPH_CAFFE_H_
#define INC_EXTERNAL_ACL_GRAPH_CAFFE_H_ #define INC_EXTERNAL_ACL_GRAPH_CAFFE_H_


#if defined(_MSC_VER)
#ifdef FUNC_VISIBILITY
#define PARSER_FUNC_VISIBILITY _declspec(dllexport)
#else
#define PARSER_FUNC_VISIBILITY
#endif
#else
#ifdef FUNC_VISIBILITY
#define PARSER_FUNC_VISIBILITY __attribute__((visibility("default")))
#else
#define PARSER_FUNC_VISIBILITY
#endif
#endif

#include <memory> #include <memory>
#include <string> #include <string>
#include <vector> #include <vector>
@@ -24,7 +38,6 @@
#include "graph/ascend_string.h" #include "graph/ascend_string.h"
#include "graph/ge_error_codes.h" #include "graph/ge_error_codes.h"
#include "graph/graph.h" #include "graph/graph.h"
#include "parser_common.h"


namespace ge { namespace ge {
PARSER_FUNC_VISIBILITY graphStatus aclgrphParseCaffe(const char *model_file, const char *weights_file, PARSER_FUNC_VISIBILITY graphStatus aclgrphParseCaffe(const char *model_file, const char *weights_file,


+ 14
- 1
inc/external/parser/onnx_parser.h View File

@@ -17,11 +17,24 @@
#ifndef INC_EXTERNAL_PARSER_ONNX_PARSER_H_ #ifndef INC_EXTERNAL_PARSER_ONNX_PARSER_H_
#define INC_EXTERNAL_PARSER_ONNX_PARSER_H_ #define INC_EXTERNAL_PARSER_ONNX_PARSER_H_


#if defined(_MSC_VER)
#ifdef FUNC_VISIBILITY
#define PARSER_FUNC_VISIBILITY _declspec(dllexport)
#else
#define PARSER_FUNC_VISIBILITY
#endif
#else
#ifdef FUNC_VISIBILITY
#define PARSER_FUNC_VISIBILITY __attribute__((visibility("default")))
#else
#define PARSER_FUNC_VISIBILITY
#endif
#endif

#include <map> #include <map>
#include "graph/ascend_string.h" #include "graph/ascend_string.h"
#include "graph/ge_error_codes.h" #include "graph/ge_error_codes.h"
#include "graph/graph.h" #include "graph/graph.h"
#include "parser_common.h"


namespace ge { namespace ge {
PARSER_FUNC_VISIBILITY graphStatus aclgrphParseONNX(const char *model_file, PARSER_FUNC_VISIBILITY graphStatus aclgrphParseONNX(const char *model_file,


+ 0
- 34
inc/external/parser/parser_common.h View File

@@ -1,34 +0,0 @@
/**
* Copyright (c) Huawei Technologies Co., Ltd. 2020~2022. All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

#ifndef INC_EXTERNAL_ACL_PARSER_COMMON_H_
#define INC_EXTERNAL_ACL_PARSER_COMMON_H_

#if defined(_MSC_VER)
#ifdef FUNC_VISIBILITY
#define PARSER_FUNC_VISIBILITY _declspec(dllexport)
#else
#define PARSER_FUNC_VISIBILITY
#endif
#else
#ifdef FUNC_VISIBILITY
#define PARSER_FUNC_VISIBILITY __attribute__((visibility("default")))
#else
#define PARSER_FUNC_VISIBILITY
#endif
#endif

#endif // INC_EXTERNAL_ACL_PARSER_COMMON_H_

+ 14
- 2
inc/external/parser/tensorflow_parser.h View File

@@ -17,15 +17,27 @@
#ifndef INC_EXTERNAL_ACL_PARSER_TENSORFLOW_H_ #ifndef INC_EXTERNAL_ACL_PARSER_TENSORFLOW_H_
#define INC_EXTERNAL_ACL_PARSER_TENSORFLOW_H_ #define INC_EXTERNAL_ACL_PARSER_TENSORFLOW_H_


#if defined(_MSC_VER)
#ifdef FUNC_VISIBILITY
#define PARSER_FUNC_VISIBILITY _declspec(dllexport)
#else
#define PARSER_FUNC_VISIBILITY
#endif
#else
#ifdef FUNC_VISIBILITY
#define PARSER_FUNC_VISIBILITY __attribute__((visibility("default")))
#else
#define PARSER_FUNC_VISIBILITY
#endif
#endif

#include <atomic> #include <atomic>
#include <memory> #include <memory>
#include <string> #include <string>
#include <vector> #include <vector>

#include "graph/ascend_string.h" #include "graph/ascend_string.h"
#include "graph/ge_error_codes.h" #include "graph/ge_error_codes.h"
#include "graph/graph.h" #include "graph/graph.h"
#include "parser_common.h"


namespace ge { namespace ge {
PARSER_FUNC_VISIBILITY graphStatus aclgrphParseTensorFlow(const char *model_file, ge::Graph &graph); PARSER_FUNC_VISIBILITY graphStatus aclgrphParseTensorFlow(const char *model_file, ge::Graph &graph);


Loading…
Cancel
Save