diff --git a/ge/graph/passes/dynamic_single_op_reset_shape_pass.cc b/ge/graph/passes/dynamic_single_op_reset_shape_pass.cc index 1a59f348..43fbe86b 100644 --- a/ge/graph/passes/dynamic_single_op_reset_shape_pass.cc +++ b/ge/graph/passes/dynamic_single_op_reset_shape_pass.cc @@ -23,6 +23,9 @@ #include "graph/debug/ge_attr_define.h" namespace ge { +namespace { +const int64_t kDynamicShapeDim = -2; +} Status DynamicSingleOpResetShapePass::Run(ComputeGraphPtr graph) { GE_CHECK_NOTNULL(graph); for (const auto &node : graph->GetDirectNode()) { @@ -38,7 +41,7 @@ Status DynamicSingleOpResetShapePass::Run(ComputeGraphPtr graph) { } auto op_desc = node->GetOpDesc(); - std::vector dynamic_shape_dims = {-2}; + std::vector dynamic_shape_dims = {kDynamicShapeDim}; GeShape dynamic_shape(dynamic_shape_dims); for (size_t i = 0; i < op_desc->GetAllInputsDesc().size(); i++) { auto input_desc = op_desc->MutableInputDesc(static_cast(i)); diff --git a/ge/graph/passes/dynamic_single_op_reset_shape_pass.h b/ge/graph/passes/dynamic_single_op_reset_shape_pass.h index e82d4e16..a8a1a724 100644 --- a/ge/graph/passes/dynamic_single_op_reset_shape_pass.h +++ b/ge/graph/passes/dynamic_single_op_reset_shape_pass.h @@ -22,7 +22,7 @@ namespace ge { class DynamicSingleOpResetShapePass : public GraphPass { public: - Status Run(ComputeGraphPtr graph); + Status Run(ComputeGraphPtr graph) override; }; } // namespace ge #endif // GE_GRAPH_PASSES_DYNAMIC_SINGLE_OP_RESET_SHAPE_PASS_H_