Browse Source

remove dependence of parser

pull/236/head
wjm 5 years ago
parent
commit
4565825ecf
45 changed files with 282 additions and 65 deletions
  1. +0
    -1
      parser/caffe/caffe_custom_parser_adapter.cc
  2. +0
    -1
      parser/caffe/caffe_data_parser.cc
  3. +3
    -5
      parser/caffe/caffe_parser.cc
  4. +2
    -2
      parser/caffe/caffe_reshape_parser.cc
  5. +0
    -1
      parser/common/acl_graph_parser_util.cc
  6. +1
    -2
      parser/common/data_op_parser.cc
  7. +1
    -1
      parser/common/data_op_parser.h
  8. +1
    -1
      parser/common/model_saver.cc
  9. +1
    -1
      parser/common/op_def/ir_pb_converter.cc
  10. +0
    -1
      parser/common/op_def/op_schema.cc
  11. +1
    -1
      parser/common/op_def/operator.cc
  12. +0
    -1
      parser/common/op_parser_factory.cc
  13. +1
    -1
      parser/common/parser_api.cc
  14. +0
    -1
      parser/common/parser_factory.cc
  15. +1
    -1
      parser/common/parser_utils.cc
  16. +0
    -1
      parser/common/pass_manager.cc
  17. +1
    -1
      parser/common/pre_checker.cc
  18. +0
    -1
      parser/common/proto_file_parser.cc
  19. +0
    -1
      parser/common/register_tbe.cc
  20. +1
    -1
      parser/common/tuple.h
  21. +248
    -0
      parser/common/util.h
  22. +1
    -1
      parser/onnx/onnx_constant_parser.cc
  23. +1
    -0
      parser/onnx/onnx_constant_parser.h
  24. +2
    -2
      parser/onnx/onnx_parser.cc
  25. +1
    -2
      parser/tensorflow/graph_optimizer.cc
  26. +0
    -1
      parser/tensorflow/iterator_fusion_pass.cc
  27. +3
    -3
      parser/tensorflow/iterator_fusion_pass.h
  28. +0
    -1
      parser/tensorflow/tensorflow_arg_parser.cc
  29. +0
    -1
      parser/tensorflow/tensorflow_constant_parser.cc
  30. +0
    -1
      parser/tensorflow/tensorflow_data_parser.cc
  31. +1
    -1
      parser/tensorflow/tensorflow_enter_parser.cc
  32. +0
    -2
      parser/tensorflow/tensorflow_fill_parser.cc
  33. +0
    -1
      parser/tensorflow/tensorflow_frameworkop_parser.cc
  34. +0
    -2
      parser/tensorflow/tensorflow_fusion_op_parser.cc
  35. +0
    -1
      parser/tensorflow/tensorflow_fusionop_util.cc
  36. +0
    -2
      parser/tensorflow/tensorflow_fusionop_util.h
  37. +1
    -1
      parser/tensorflow/tensorflow_merge_parser.cc
  38. +1
    -1
      parser/tensorflow/tensorflow_no_op_parser.cc
  39. +4
    -7
      parser/tensorflow/tensorflow_parser.cc
  40. +1
    -1
      parser/tensorflow/tensorflow_parser_register.h
  41. +1
    -2
      parser/tensorflow/tensorflow_reshape_parser.cc
  42. +1
    -3
      parser/tensorflow/tensorflow_squeeze_parser.cc
  43. +1
    -1
      parser/tensorflow/tensorflow_util.cc
  44. +1
    -1
      parser/tensorflow/tensorflow_var_is_initialized_op_parser.cc
  45. +0
    -1
      parser/tensorflow/tensorflow_variable_v2_parser.cc

+ 0
- 1
parser/caffe/caffe_custom_parser_adapter.cc View File

@@ -17,7 +17,6 @@
#include "parser/caffe/caffe_custom_parser_adapter.h"
#include <memory>
#include <vector>
#include "common/debug/log.h"
#include "parser/common/acl_graph_parser_util.h"
#include "common/util.h"
#include "framework/common/debug/ge_log.h"


+ 0
- 1
parser/caffe/caffe_data_parser.cc View File

@@ -17,7 +17,6 @@
#include "parser/caffe/caffe_data_parser.h"
#include <unordered_map>
#include <utility>
#include "common/debug/log.h"
#include "framework/omg/parser/parser_types.h"
#include "common/util.h"
#include "common/util/error_manager/error_manager.h"


+ 3
- 5
parser/caffe/caffe_parser.cc View File

@@ -22,17 +22,14 @@
#include <memory>
#include <algorithm>
#include "parser/common/convert/pb2json.h"
#include "common/debug/log.h"
#include "parser/common/acl_graph_parser_util.h"
#include "common/op_map.h"
#include "common/util/error_manager/error_manager.h"
#include "common/ge_types.h"
#include "common/string_util.h"
#include "external/graph/operator_factory.h"
#include "external/parser/caffe_parser.h"
#include "external/ge/ge_api_types.h"
#include "framework/common/debug/ge_log.h"
#include "graph/optimize/common/params.h"
#include "graph/utils/graph_utils.h"
#include <google/protobuf/compiler/importer.h>
#include <google/protobuf/descriptor.h>
@@ -52,6 +49,7 @@
#include "parser/common/acl_graph_parser_util.h"
#include "parser/common/proto_file_parser.h"
#include "register/op_registry.h"
#include "register/register_fmk_types.h"

using domi::caffe::LayerParameter;
using domi::caffe::NetParameter;
@@ -79,7 +77,7 @@ graphStatus aclgrphParseCaffe(const char *model_file, const char *weights_file,
GE_CHECK_NOTNULL(model_file);
GetParserContext().type = domi::CAFFE;
std::map<string, string> options;
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(ge::CAFFE)));
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(domi::CAFFE)));

// load custom plugin so and proto
AclGrphParseUtil acl_graph_parse_util;
@@ -126,7 +124,7 @@ graphStatus aclgrphParseCaffe(const char *model_file, const char *weights_file,
GE_CHECK_NOTNULL(model_file);
GetParserContext().type = domi::CAFFE;
std::map<string, string> options;
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(ge::CAFFE)));
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(domi::CAFFE)));

// load custom plugin so and proto
AclGrphParseUtil acl_graph_parse_util;


+ 2
- 2
parser/caffe/caffe_reshape_parser.cc View File

@@ -16,9 +16,7 @@

#include "parser/caffe/caffe_reshape_parser.h"
#include <vector>
#include "common/debug/log.h"
#include "parser/common/acl_graph_parser_util.h"
#include "common/op/op_parser_util.h"
#include "common/util.h"
#include "framework/common/debug/ge_log.h"
#include "graph/utils/graph_utils.h"
@@ -33,6 +31,8 @@ namespace ge {
namespace {
const int kAnchorIndexZero = 0;
const int kAnchorIndexOne = 1;
const int32_t RESHAPE_AXIS_DEFAULT_VALUE = 0;
const int32_t RESHAPE_NUM_AXES_DEFAULT_VALUE = -1;
} // namespace

Status CaffeReshapeParser::ParseParams(const Message *op_src, ge::OpDescPtr &op) {


+ 0
- 1
parser/common/acl_graph_parser_util.cc View File

@@ -23,7 +23,6 @@
#include <ctime>
#include <fstream>

#include "common/debug/log.h"
#include "common/op/ge_op_utils.h"
#include "common/string_util.h"
#include "common/types.h"


+ 1
- 2
parser/common/data_op_parser.cc View File

@@ -16,12 +16,11 @@

#include "parser/common/data_op_parser.h"
#include <cstdlib>
#include "common/debug/log.h"
#include "common/op/ge_op_utils.h"
#include "parser/common/acl_graph_parser_util.h"
#include "omg/parser/parser_inner_ctx.h"
#include "common/util.h"
#include "graph/utils/type_utils.h"
#include "omg/omg.h"

namespace {
const int kDataMemAlignSize = 32;


+ 1
- 1
parser/common/data_op_parser.h View File

@@ -19,11 +19,11 @@

#include <google/protobuf/text_format.h>
#include <vector>
#include "common/debug/log.h"
#include "common/op/attr_value_util.h"
#include "framework/omg/parser/parser_types.h"
#include "omg/omg_inner_types.h"
#include "proto/om.pb.h"
#include "ge/ge_api_error_codes.h"

#include "graph/attr_value.h"
#include "graph/compute_graph.h"


+ 1
- 1
parser/common/model_saver.cc View File

@@ -19,7 +19,7 @@

#include "parser/common/model_saver.h"
#include "framework/common/debug/ge_log.h"
#include "framework/common/debug/log.h"
#include "common/util.h"
#include "common/util/error_manager/error_manager.h"
#include "mmpa/mmpa_api.h"



+ 1
- 1
parser/common/op_def/ir_pb_converter.cc View File

@@ -24,7 +24,7 @@
#include "graph/buffer.h"
#include "framework/common/debug/ge_log.h"
#include "framework/omg/parser/parser_types.h"
#include "framework/common/util.h"
#include "common/util.h"

namespace ge {
static void ConvertList(const std::pair<std::string, OpAttribute> &op_attr_pair, ge::OpDescPtr op_def) {


+ 0
- 1
parser/common/op_def/op_schema.cc View File

@@ -18,7 +18,6 @@
#include <iostream>
#include <utility>
#include "framework/common/debug/ge_log.h"
#include "framework/common/debug/log.h"

namespace ge {
OpSchema::FormalParameter::FormalParameter(const std::string &name, FormalParameterOption param_option)


+ 1
- 1
parser/common/op_def/operator.cc View File

@@ -17,7 +17,7 @@
#include "operator.h"
#include <utility>
#include "framework/common/fmk_types.h"
#include "framework/common/util.h"
#include "common/util.h"
#include "framework/common/debug/ge_log.h"

using ge::BoolTuple;


+ 0
- 1
parser/common/op_parser_factory.cc View File

@@ -15,7 +15,6 @@
*/

#include "parser/common/op_parser_factory.h"
#include "common/debug/log.h"
#include "framework/common/debug/ge_log.h"
#include "graph/utils/type_utils.h"



+ 1
- 1
parser/common/parser_api.cc View File

@@ -15,8 +15,8 @@
*/

#include "framework/omg/parser/parser_api.h"
#include "common/debug/log.h"

#include "common/util.h"
#include "tbe_plugin_loader.h"
#include "framework/common/debug/ge_log.h"
#include "parser/common/register_tbe.h"


+ 0
- 1
parser/common/parser_factory.cc View File

@@ -15,7 +15,6 @@
*/

#include "omg/parser/parser_factory.h"
#include "common/debug/log.h"
#include "framework/common/debug/ge_log.h"

namespace domi {


+ 1
- 1
parser/common/parser_utils.cc View File

@@ -17,7 +17,7 @@
#include "parser_utils.h"
#include "external/ge/ge_api_types.h"
#include "framework/common/debug/ge_log.h"
#include "framework/common/util.h"
#include "common/util.h"
#include "framework/omg/parser/parser_types.h"
#include "graph/anchor.h"
#include "graph/compute_graph.h"


+ 0
- 1
parser/common/pass_manager.cc View File

@@ -17,7 +17,6 @@
#include "parser/common/pass_manager.h"
#include "framework/omg/parser/parser_types.h"
#include "parser/common/acl_graph_parser_util.h"
#include "common/debug/log.h"
#include "graph/utils/node_utils.h"
#include "omg/omg_inner_types.h"



+ 1
- 1
parser/common/pre_checker.cc View File

@@ -21,9 +21,9 @@
#include "common/util.h"
#include "common/util/error_manager/error_manager.h"
#include "framework/common/debug/ge_log.h"
#include "omg/omg.h"
#include "parser/common/op_parser_factory.h"
#include "parser/common/model_saver.h"
#include "omg/parser/parser_inner_ctx.h"
#include "register/op_registry.h"

namespace ge {


+ 0
- 1
parser/common/proto_file_parser.cc View File

@@ -26,7 +26,6 @@
#include "common/string_util.h"
#include "common/types.h"
#include "common/util.h"
#include "common/debug/log.h"
#include "parser/common/acl_graph_parser_util.h"
#include "ge/ge_api_types.h"
#include "framework/common/debug/ge_log.h"


+ 0
- 1
parser/common/register_tbe.cc View File

@@ -18,7 +18,6 @@
#include <map>
#include <memory>
#include <string>
#include "common/debug/log.h"
#include "parser/common/acl_graph_parser_util.h"
#include "common/op/ge_op_utils.h"
#include "common/op_map.h"


+ 1
- 1
parser/common/tuple.h View File

@@ -23,7 +23,7 @@
#include <type_traits>
#include <utility>
#include <vector>
#include "framework/common/debug/log.h"
#include "framework/common/debug/ge_log.h"

namespace ge {
template <typename ValueType>


+ 248
- 0
parser/common/util.h View File

@@ -0,0 +1,248 @@
/**
* Copyright 2019-2020 Huawei Technologies Co., Ltd
*
* 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 PARSER_COMMON_UTIL_H_
#define PARSER_COMMON_UTIL_H_

#include "framework/common/debug/ge_log.h"
#include "mmpa/mmpa_api.h"

#define CHECK_FALSE_EXEC(expr, exec_expr, ...) \
{ \
bool b = (expr); \
if (!b) { \
exec_expr; \
} \
}

// For propagating errors when calling a function.
#define GE_RETURN_IF_ERROR(expr) \
do { \
const ::ge::Status _status = (expr); \
if (_status) return _status; \
} while (0)

#define GE_RETURN_WITH_LOG_IF_ERROR(expr, ...) \
do { \
const ::ge::Status _status = (expr); \
if (_status) { \
GELOGE(ge::FAILED, __VA_ARGS__); \
return _status; \
} \
} while (0)

// check whether the parameter is true. If it is, return FAILED and record the error log
#define GE_RETURN_WITH_LOG_IF_TRUE(condition, ...) \
do { \
if (condition) { \
GELOGE(ge::FAILED, __VA_ARGS__); \
return ge::FAILED; \
} \
} while (0)

// Check if the parameter is false. If yes, return FAILED and record the error log
#define GE_RETURN_WITH_LOG_IF_FALSE(condition, ...) \
do { \
bool _condition = (condition); \
if (!_condition) { \
GELOGE(ge::FAILED, __VA_ARGS__); \
return ge::FAILED; \
} \
} while (0)

// Check if the parameter is false. If yes, return PARAM_INVALID and record the error log
#define GE_RT_PARAM_INVALID_WITH_LOG_IF_FALSE(condition, ...) \
do { \
bool _condition = (condition); \
if (!_condition) { \
GELOGE(ge::FAILED, __VA_ARGS__); \
return ge::PARAM_INVALID; \
} \
} while (0)

// Check if the parameter is null. If yes, return PARAM_INVALID and record the error
#define GE_CHECK_NOTNULL(val) \
do { \
if (val == nullptr) { \
GELOGE(ge::FAILED, "param[%s] must not be null.", #val); \
return ge::PARAM_INVALID; \
} \
} while (0)

// Check whether the parameter is null. If so, execute the exec_expr expression and record the error log
#define GE_CHECK_NOTNULL_EXEC(val, exec_expr) \
do { \
if (val == nullptr) { \
GELOGE(ge::FAILED, "param[%s] must not be null.", #val); \
exec_expr; \
} \
} while (0)

// Check if the value on the left is greater than or equal to the value on the right
#define GE_CHECK_GE(lhs, rhs) \
do { \
if (lhs < rhs) { \
GELOGE(ge::FAILED, "param[%s] is less than[%s]", #lhs, #rhs); \
return ge::PARAM_INVALID; \
} \
} while (0)

#define GE_DELETE_NEW_SINGLE(var) \
do { \
if (var != nullptr) { \
delete var; \
var = nullptr; \
} \
} while (0)

#define GE_DELETE_NEW_ARRAY(var) \
do { \
if (var != nullptr) { \
delete[] var; \
var = nullptr; \
} \
} while (0)

// If expr is true, execute exec_expr without printing logs
#define GE_IF_BOOL_EXEC(expr, exec_expr) \
{ \
if (expr) { \
exec_expr; \
} \
}

// If expr is not true, print the log and execute a custom statement
#define GE_CHK_BOOL_TRUE_EXEC_INFO(expr, exec_expr, ...) \
{ \
bool b = (expr); \
if (b) { \
GELOGI(__VA_ARGS__); \
exec_expr; \
} \
}

// If expr is not true, print the log and return the specified status
#define GE_CHK_BOOL_RET_STATUS(expr, _status, ...) \
do { \
bool b = (expr); \
if (!b) { \
GELOGE(_status, __VA_ARGS__); \
return _status; \
} \
} while (0);

// If expr is not SUCCESS, print the log and execute the expression + return _status
#define GE_CHK_BOOL_TRUE_EXEC_RET_STATUS(expr, _status, exec_expr, ...) \
{ \
bool b = (expr); \
if (b) { \
GELOGE(ge::FAILED, __VA_ARGS__); \
exec_expr; \
return _status; \
} \
}

// If expr is not SUCCESS, print the log and return the same value
#define GE_CHK_STATUS_RET(expr, ...) \
do { \
const ge::Status _status = (expr); \
if (_status != ge::SUCCESS) { \
GELOGE(ge::FAILED, __VA_ARGS__); \
return _status; \
} \
} while (0);

// If expr is true, print logs and execute custom statements
#define GE_CHK_BOOL_TRUE_EXEC_WITH_LOG(expr, exec_expr, ...) \
{ \
bool b = (expr); \
if (b) { \
GELOGE(ge::FAILED, __VA_ARGS__); \
exec_expr; \
} \
}

// If expr is not SUCCESS, print the log and do not execute return
#define GE_CHK_STATUS(expr, ...) \
do { \
const ge::Status _status = (expr); \
if (_status != ge::SUCCESS) { \
GELOGE(ge::FAILED, __VA_ARGS__); \
} \
} while (0);

#define GE_LOGE_IF(condition, ...) \
if ((condition)) { \
GELOGE(ge::FAILED, __VA_ARGS__); \
}

// If expr is not true, print the log and execute a custom statement
#define GE_CHK_BOOL_EXEC(expr, exec_expr, ...) \
{ \
bool b = (expr); \
if (!b) { \
GELOGE(ge::FAILED, __VA_ARGS__); \
exec_expr; \
} \
}

// ge marco
#define GE_LOGI_IF(condition, ...) \
if ((condition)) { \
GELOGI(__VA_ARGS__); \
}

#define GE_LOGW_IF(condition, ...) \
if ((condition)) { \
GELOGW(__VA_ARGS__); \
}

// If expr is not true, execute a custom statement
#define GE_CHK_BOOL_EXEC_NOLOG(expr, exec_expr) \
{ \
bool b = (expr); \
if (!b) { \
exec_expr; \
} \
}

// If expr is not SUCCESS, print the log and execute a custom statement
#define GE_CHK_STATUS_EXEC(expr, exec_expr, ...) \
do { \
const ge::Status _status = (expr); \
GE_CHK_BOOL_EXEC(_status == SUCCESS, exec_expr, __VA_ARGS__); \
} while (0);

// If expr is not true, print the log and execute a custom statement
#define GE_CHK_BOOL_EXEC_INFO(expr, exec_expr, ...) \
{ \
bool b = (expr); \
if (!b) { \
GELOGI(__VA_ARGS__); \
exec_expr; \
} \
}

// If make_shared is abnormal, print the log and execute the statement
#define GE_MAKE_SHARED(exec_expr0, exec_expr1) \
try { \
exec_expr0; \
} catch (const std::bad_alloc &) { \
GELOGE(ge::FAILED, "Make shared failed"); \
exec_expr1; \
}

#endif // PARSER_COMMON_UTIL_H_

+ 1
- 1
parser/onnx/onnx_constant_parser.cc View File

@@ -18,7 +18,7 @@
#include <map>
#include <vector>
#include "parser/common/acl_graph_parser_util.h"
#include "common/util.h"
#include "framework/omg/parser/parser_inner_ctx.h"
#include "graph/ge_tensor.h"
#include "graph/utils/tensor_adapter.h"


+ 1
- 0
parser/onnx/onnx_constant_parser.h View File

@@ -18,6 +18,7 @@
#define GE_PARSER_ONNX_ONNX_CONSTANT_PARSER_H_

#include <string>
#include "common/util.h"
#include "parser/common/data_op_parser.h"
#include "parser/onnx/onnx_op_parser.h"



+ 2
- 2
parser/onnx/onnx_parser.cc View File

@@ -19,7 +19,6 @@
#include <iostream>
#include "common/convert/pb2json.h"
#include "common/util.h"
#include "common/ge_types.h"
#include "common/util/error_manager/error_manager.h"
#include "external/graph/operator_factory.h"
#include "external/register/register_error_codes.h"
@@ -37,6 +36,7 @@
#include "parser/common/parser_utils.h"
#include "parser/onnx/onnx_util.h"
#include "register/op_registry.h"
#include "register/register_fmk_types.h"

namespace ge {
graphStatus PrepareBeforeParse(AclGrphParseUtil &acl_graph_parse_util,
@@ -44,7 +44,7 @@ graphStatus PrepareBeforeParse(AclGrphParseUtil &acl_graph_parse_util,
ge::Graph &graph, std::shared_ptr<domi::ModelParser> &model_parser) {
GetParserContext().type = domi::ONNX;
std::map<string, string> options;
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(ge::ONNX)));
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(domi::ONNX)));

if (acl_graph_parse_util.AclParserInitialize(options) != ge::SUCCESS) {
GELOGE(ge::FAILED, "Acl parser initialize failed.");


+ 1
- 2
parser/tensorflow/graph_optimizer.cc View File

@@ -22,15 +22,14 @@
#include "./graph_insert_trans_op.h"
#include "cce/cce.h"
#include "cce/dnn.h"
#include "common/debug/log.h"
#include "parser/common/acl_graph_parser_util.h"
#include "common/op/ge_op_utils.h"
#include "common/op_map.h"
#include "common/op_types.h"
#include "common/types_map.h"
#include "framework/common/debug/ge_log.h"
#include "framework/omg/parser/parser_inner_ctx.h"
#include "framework/omg/parser/parser_types.h"
#include "graph/common/omg_util.h"
#include "graph/debug/ge_attr_define.h"
#include "graph/ge_tensor.h"
#include "graph/types.h"


+ 0
- 1
parser/tensorflow/iterator_fusion_pass.cc View File

@@ -18,7 +18,6 @@

#include <memory>

#include "common/debug/log.h"
#include "framework/omg/parser/parser_types.h"
#include "common/util.h"
#include "graph_optimizer.h"


+ 3
- 3
parser/tensorflow/iterator_fusion_pass.h View File

@@ -17,13 +17,13 @@
#ifndef GE_GRAPH_PASSES_ITERATOR_FUSION_PASS_H_
#define GE_GRAPH_PASSES_ITERATOR_FUSION_PASS_H_

#include "framework/common/ge_types.h"
#include "inc/graph_pass.h"
#include "register/register_fmk_types.h"

namespace ge {
class IteratorFusionPass : public GraphPass {
public:
IteratorFusionPass(ge::FrameworkType type, bool local_fmk_op_flag)
IteratorFusionPass(domi::FrameworkType type, bool local_fmk_op_flag)
: fmk_type_(type), local_fmk_op_flag_(local_fmk_op_flag) {}

virtual ~IteratorFusionPass() {}
@@ -31,7 +31,7 @@ class IteratorFusionPass : public GraphPass {
Status Run(ge::ComputeGraphPtr graph) final;

private:
ge::FrameworkType fmk_type_;
domi::FrameworkType fmk_type_;
bool local_fmk_op_flag_;
};
} // namespace ge


+ 0
- 1
parser/tensorflow/tensorflow_arg_parser.cc View File

@@ -14,7 +14,6 @@
* limitations under the License.
*/

#include "common/debug/log.h"
#include "parser/common/op_def/arg_op.h"
#include "framework/common/debug/ge_log.h"
#include "framework/omg/parser/parser_inner_ctx.h"


+ 0
- 1
parser/tensorflow/tensorflow_constant_parser.cc View File

@@ -18,7 +18,6 @@
#include <map>
#include <memory>
#include <vector>
#include "common/debug/log.h"
#include "parser/common/acl_graph_parser_util.h"
#include "common/op/ge_op_utils.h"
#include "parser/common/op_def/constant_op.h"


+ 0
- 1
parser/tensorflow/tensorflow_data_parser.cc View File

@@ -16,7 +16,6 @@

#include "parser/tensorflow/tensorflow_data_parser.h"
#include <unordered_map>
#include "common/debug/log.h"
#include "common/util.h"
#include "framework/common/debug/ge_log.h"
#include "framework/omg/parser/parser_inner_ctx.h"


+ 1
- 1
parser/tensorflow/tensorflow_enter_parser.cc View File

@@ -16,7 +16,7 @@

#include "parser/tensorflow/tensorflow_enter_parser.h"
#include "framework/common/debug/ge_log.h"
#include "framework/common/debug/log.h"
#include "common/util.h"
#include "graph/debug/ge_attr_define.h"
#include "parser/common/op_parser_factory.h"
#include "framework/omg/parser/parser_types.h"


+ 0
- 2
parser/tensorflow/tensorflow_fill_parser.cc View File

@@ -15,10 +15,8 @@
*/

// Copyright (c) <2018>, <Huawei Technologies Co., Ltd>
#include "common/debug/log.h"
#include "common/op/attr_value_util.h"
#include "parser/common/op_def/fill_op.h"
#include "common/util.h"
#include "parser/tensorflow/tensorflow_parser_register.h"
#include "framework/omg/parser/parser_types.h"



+ 0
- 1
parser/tensorflow/tensorflow_frameworkop_parser.cc View File

@@ -14,7 +14,6 @@
* limitations under the License.
*/

#include "common/debug/log.h"
#include "parser/common/op_def/frameworkop_op.h"
#include "framework/common/debug/ge_log.h"
#include "parser/common/op_parser_factory.h"


+ 0
- 2
parser/tensorflow/tensorflow_fusion_op_parser.cc View File

@@ -16,11 +16,9 @@

#include "parser/tensorflow/tensorflow_fusion_op_parser.h"
#include <memory>
#include "common/debug/log.h"
#include "parser/common/acl_graph_parser_util.h"
#include "common/util.h"
#include "framework/common/debug/ge_log.h"
#include "omg/omg.h"
#include "parser/common/parser_fp16_t.h"
#include "parser/tensorflow/tensorflow_op_parser.h"
#include "register/tensor_assign.h"


+ 0
- 1
parser/tensorflow/tensorflow_fusionop_util.cc View File

@@ -16,7 +16,6 @@

#include "parser/tensorflow/tensorflow_fusionop_util.h"
#include "common/util/error_manager/error_manager.h"
#include "common/debug/log.h"
#include "common/op/ge_op_utils.h"
#include "framework/common/debug/ge_log.h"
#include "parser/tensorflow/tensorflow_parser.h"


+ 0
- 2
parser/tensorflow/tensorflow_fusionop_util.h View File

@@ -20,10 +20,8 @@
#include <string>
#include <utility>
#include <vector>
#include "common/debug/log.h"
#include "common/string_util.h"
#include "framework/omg/parser/parser_types.h"
#include "common/util.h"
#include "omg/omg_inner_types.h"
#include "proto/tensorflow/graph.pb.h"
#include "external/register/scope/scope_fusion_pass_register.h"


+ 1
- 1
parser/tensorflow/tensorflow_merge_parser.cc View File

@@ -17,7 +17,7 @@
#include "parser/tensorflow/tensorflow_merge_parser.h"

#include "framework/common/debug/ge_log.h"
#include "framework/common/util.h"
#include "common/util.h"
#include "graph/debug/ge_attr_define.h"
#include "parser/common/op_parser_factory.h"
#include "framework/omg/parser/parser_types.h"


+ 1
- 1
parser/tensorflow/tensorflow_no_op_parser.cc View File

@@ -15,7 +15,7 @@
*/

#include "parser/tensorflow/tensorflow_no_op_parser.h"
#include "framework/common/util.h"
#include "common/util.h"
#include "framework/common/debug/ge_log.h"
#include "parser/common/op_def/ir_pb_converter.h"
#include "parser/common/op_def/no_op_op.h"


+ 4
- 7
parser/tensorflow/tensorflow_parser.cc View File

@@ -17,8 +17,8 @@
#include "parser/tensorflow/tensorflow_parser.h"
#include <algorithm>
#include <iostream>
#include "ge/ge_api_types.h"
#include "parser/common/convert/pb2json.h"
#include "common/debug/log.h"
#include "parser/common/acl_graph_parser_util.h"
#include "common/util/error_manager/error_manager.h"
#include "external/graph/operator_factory.h"
@@ -28,13 +28,10 @@
#include "framework/omg/parser/parser_api.h"
#include "framework/omg/parser/parser_inner_ctx.h"
#include "graph/debug/ge_attr_define.h"
#include "graph/optimize/common/params.h"
#include "graph/passes/variable_format_pass.h"
#include "graph/utils/graph_utils.h"
#include "graph/utils/node_utils.h"
#include "graph/utils/type_utils.h"
#include "iterator_fusion_pass.h"
#include "omg/omg.h"
#include "omg/parser/op_parser.h"
#include "omg/parser/parser_factory.h"
#include "parser/common/acl_graph_parser_util.h"
@@ -93,7 +90,7 @@ graphStatus aclgrphParseTensorFlow(const char *model_file, ge::Graph &graph) {
GE_CHECK_NOTNULL(model_file);
GetParserContext().type = domi::TENSORFLOW;
std::map<string, string> options;
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(ge::TENSORFLOW)));
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(domi::TENSORFLOW)));

// load custom plugin so and proto
AclGrphParseUtil acl_graph_parse_util;
@@ -132,7 +129,7 @@ graphStatus aclgrphParseTensorFlow(const char *model_file, const std::map<Ascend
GE_CHECK_NOTNULL(model_file);
GetParserContext().type = domi::TENSORFLOW;
std::map<string, string> options;
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(ge::TENSORFLOW)));
options.insert(std::pair<string, string>(string(ge::FRAMEWORK_TYPE), to_string(domi::TENSORFLOW)));

// load custom plugin so and proto
AclGrphParseUtil acl_graph_parse_util;
@@ -2244,7 +2241,7 @@ Status TensorFlowModelParser::ParseProto(const google::protobuf::Message *proto,
ge::parser::PassManager iterator_fusion_pass;
try {
(void)iterator_fusion_pass.AddPass("ParseProto::IteratorFusionPass",
new ge::IteratorFusionPass(ge::TENSORFLOW, false));
new ge::IteratorFusionPass(domi::TENSORFLOW, false));
} catch (std::bad_alloc &e) {
GELOGE(INTERNAL_ERROR, "Add pass failed, bad memory allocation occurs.");
return INTERNAL_ERROR;


+ 1
- 1
parser/tensorflow/tensorflow_parser_register.h View File

@@ -21,7 +21,7 @@
#include <functional>
#include <memory>
#include <string>
#include "framework/common/util.h"
#include "common/util.h"
#include "framework/omg/parser/op_parser.h"
#include "parser/common/op_def/ir_pb_converter.h"
#include "parser/common/op_def/operator.h"


+ 1
- 2
parser/tensorflow/tensorflow_reshape_parser.cc View File

@@ -16,13 +16,12 @@

#include "parser/tensorflow/tensorflow_reshape_parser.h"
#include "framework/common/debug/ge_log.h"
#include "framework/common/debug/log.h"
#include "framework/common/op/ge_op_utils.h"
#include "framework/omg/omg.h"
#include "graph/utils/type_utils.h"
#include "parser/common/op_parser_factory.h"
#include "parser/tensorflow/tensorflow_util.h"
#include "parser/common/acl_graph_parser_util.h"
#include "omg/parser/parser_inner_ctx.h"

using domi::TENSORFLOW;
using namespace ge::parser;


+ 1
- 3
parser/tensorflow/tensorflow_squeeze_parser.cc View File

@@ -18,10 +18,8 @@
#include <memory>
#include <vector>
#include "framework/common/debug/ge_log.h"
#include "framework/common/debug/log.h"
#include "framework/common/op/attr_value_util.h"
#include "framework/common/op/op_parser_util.h"
#include "framework/common/util.h"
#include "common/util.h"
#include "framework/omg/parser/parser_inner_ctx.h"
#include "graph/utils/type_utils.h"
#include "parser/common/op_parser_factory.h"


+ 1
- 1
parser/tensorflow/tensorflow_util.cc View File

@@ -19,8 +19,8 @@
#include <cstdlib>
#include <iostream>
#include <memory>
#include "common/string_util.h"
#include "framework/common/debug/ge_log.h"
#include "framework/common/debug/log.h"
#include "framework/common/op/ge_op_utils.h"
#include "framework/omg/parser/parser_types.h"
#include "graph/debug/ge_attr_define.h"


+ 1
- 1
parser/tensorflow/tensorflow_var_is_initialized_op_parser.cc View File

@@ -15,7 +15,7 @@
*/

#include "framework/common/debug/ge_log.h"
#include "framework/common/util.h"
//#include "common/util.h"
#include "framework/common/op/ge_op_utils.h"
#include "parser/common/op_def/var_is_initialized_op_op.h"
#include "parser/common/op_parser_factory.h"


+ 0
- 1
parser/tensorflow/tensorflow_variable_v2_parser.cc View File

@@ -15,7 +15,6 @@
*/

#include "framework/common/debug/ge_log.h"
#include "framework/common/debug/log.h"
#include "framework/common/op/ge_op_utils.h"
#include "graph/compute_graph.h"
#include "graph/ge_attr_value.h"


Loading…
Cancel
Save