From 6e87bbdf70b26234c615ec95f55efd171ba42376 Mon Sep 17 00:00:00 2001 From: zhao-lupeng Date: Fri, 8 Apr 2022 14:33:31 +0800 Subject: [PATCH] fix com --- parser/tensorflow/tensorflow_parser.cc | 7 ++++--- parser/tensorflow/tensorflow_parser.h | 6 +++--- parser/tensorflow/tensorflow_shape_n_parser.cc | 4 ++-- parser/tensorflow/tensorflow_shape_n_parser.h | 4 ++-- 4 files changed, 11 insertions(+), 10 deletions(-) diff --git a/parser/tensorflow/tensorflow_parser.cc b/parser/tensorflow/tensorflow_parser.cc index f4100ac..033cc2f 100644 --- a/parser/tensorflow/tensorflow_parser.cc +++ b/parser/tensorflow/tensorflow_parser.cc @@ -3291,7 +3291,7 @@ Status TensorFlowModelParser::FusionNodeParseParams(shared_ptr &op_par * @return false optimize failed * */ -Status TensorFlowModelParser::OptimizeConstNodes4CustomOp(domi::tensorflow::GraphDef *graph_def) { +Status TensorFlowModelParser::OptimizeConstNodes4CustomOp(domi::tensorflow::GraphDef *graph_def) const { GE_CHECK_NOTNULL(graph_def); // 1. find all the nodes in the graph and save them to all_nodedef_map map all_nodedef_map; @@ -3628,7 +3628,7 @@ Status TensorFlowModelParser::RecordFusionResult(const std::shared_ptr> &inputs, const std::vector> &outputs) { int32_t in_index = 0; @@ -3909,7 +3909,8 @@ Status TensorFlowModelParser::AddFusionNodeDef(shared_ptr &scope } Status TensorFlowModelParser::AddScopeInnerNode(TensorFlowModelParser *parser, ge::ComputeGraphPtr &graph, - std::mutex *graph_mutex, const domi::tensorflow::NodeDef *node_def) { + std::mutex *const graph_mutex, + const domi::tensorflow::NodeDef *node_def) { // This is an internal function. The pointer input parameter is not empty when this function is invoked. string node_name = node_def->name(); string node_op = node_def->op(); diff --git a/parser/tensorflow/tensorflow_parser.h b/parser/tensorflow/tensorflow_parser.h index c7206c2..1de44ab 100644 --- a/parser/tensorflow/tensorflow_parser.h +++ b/parser/tensorflow/tensorflow_parser.h @@ -533,7 +533,7 @@ class PARSER_FUNC_VISIBILITY TensorFlowModelParser : public domi::ModelParser { * @return false optimize failed * */ - Status OptimizeConstNodes4CustomOp(domi::tensorflow::GraphDef *graph_def); + Status OptimizeConstNodes4CustomOp(domi::tensorflow::GraphDef *graph_def) const; /** * @ingroup domi_omg @@ -593,7 +593,7 @@ class PARSER_FUNC_VISIBILITY TensorFlowModelParser : public domi::ModelParser { static Status GetFunctionProto(const string &file, domi::tensorflow::GraphDefLibrary &graph_def_library); - Status SetOriginNodeContext(NodeDef *node_def, OpNodeContext &op_node_context, + Status SetOriginNodeContext(const NodeDef *node_def, OpNodeContext &op_node_context, const std::vector> &inputs, const std::vector> &outputs); @@ -624,7 +624,7 @@ class PARSER_FUNC_VISIBILITY TensorFlowModelParser : public domi::ModelParser { Status AddFusionNodeDef(shared_ptr &scope_graph, vector &node_name_list); static Status AddScopeInnerNode(TensorFlowModelParser *parser, ge::ComputeGraphPtr &graph, - std::mutex *graph_mutex, const domi::tensorflow::NodeDef *node_def); + std::mutex *const graph_mutex, const domi::tensorflow::NodeDef *node_def); static void DumpNodeContext(const string &node_name, const OpNodeContext &ctx, const string &phase); void DumpAllNodeContext(const string &phase) const; diff --git a/parser/tensorflow/tensorflow_shape_n_parser.cc b/parser/tensorflow/tensorflow_shape_n_parser.cc index 3d3d058..a9a422a 100644 --- a/parser/tensorflow/tensorflow_shape_n_parser.cc +++ b/parser/tensorflow/tensorflow_shape_n_parser.cc @@ -154,13 +154,13 @@ Status TensorFlowShapeNParser::ParseParams(const Message *op_src, ge::OpDescPtr } // AUTO GEN PLEASE DO NOT MODIFY IT -Status TensorFlowShapeNParser::PreParseParams(const domi::tensorflow::NodeDef *node, ShapeNOperator *op) { +Status TensorFlowShapeNParser::PreParseParams(const domi::tensorflow::NodeDef *node, const ShapeNOperator *op) { (void)node; (void)op; return SUCCESS; } -Status TensorFlowShapeNParser::PostParseParams(const domi::tensorflow::NodeDef *node, ShapeNOperator *op) { +Status TensorFlowShapeNParser::PostParseParams(const domi::tensorflow::NodeDef *node, const ShapeNOperator *op) { (void)node; (void)op; return SUCCESS; diff --git a/parser/tensorflow/tensorflow_shape_n_parser.h b/parser/tensorflow/tensorflow_shape_n_parser.h index 3c1b776..244d6da 100644 --- a/parser/tensorflow/tensorflow_shape_n_parser.h +++ b/parser/tensorflow/tensorflow_shape_n_parser.h @@ -27,8 +27,8 @@ class PARSER_FUNC_VISIBILITY TensorFlowShapeNParser : public TensorFlowOpParser Status ParseParams(const Message *op_src, ge::OpDescPtr &op_dest) override; protected: - Status PreParseParams(const domi::tensorflow::NodeDef *node, ShapeNOperator *op); - Status PostParseParams(const domi::tensorflow::NodeDef *node, ShapeNOperator *op); + Status PreParseParams(const domi::tensorflow::NodeDef *node, const ShapeNOperator *op); + Status PostParseParams(const domi::tensorflow::NodeDef *node, const ShapeNOperator *op); static Status ParseN(const domi::tensorflow::NodeDef *node, ShapeNOperator *op); static Status ParseInType(const domi::tensorflow::NodeDef *node, ShapeNOperator *op);