Browse Source

Feature: reset shape of dynamic single op

pull/618/head
l00444296 5 years ago
parent
commit
ff54d36efc
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/dynamic_single_op_reset_shape_pass.h

+ 1
- 1
ge/graph/passes/dynamic_single_op_reset_shape_pass.h View File

@@ -28,7 +28,7 @@ class DynamicSingleOpResetShapePass : public GraphPass {
private:
Status ResetOpShape(OpDescPtr &op_desc);
Status CheckAllAicpuNodes(const ComputeGraphPtr &graph, bool &is_not_aicpu);
bool CheckIfConstInput();
bool CheckIfConstInput(const GeTensorDescPtr &input_tensor_desc);
};
} // namespace ge
#endif // GE_GRAPH_PASSES_DYNAMIC_SINGLE_OP_RESET_SHAPE_PASS_H_

Loading…
Cancel
Save