From c6494ddbf88bcace1d26d167c3b54eef9276d56a Mon Sep 17 00:00:00 2001
From: lianliguang <lianliguang@huawei.com>
Date: Sun, 25 Apr 2021 15:45:56 +0800
Subject: [PATCH] clean pclint warning

---
 mindspore/core/ops/shape.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mindspore/core/ops/shape.cc b/mindspore/core/ops/shape.cc
index f1cb67814c..764eda5007 100644
--- a/mindspore/core/ops/shape.cc
+++ b/mindspore/core/ops/shape.cc
@@ -52,6 +52,8 @@ ValuePtr ShapeInferValue(const PrimitivePtr &primitive, const std::vector<Abstra
   auto op_name = primitive->name();
   CheckAndConvertUtils::CheckInteger("shape infer", input_args.size(), kEqual, 1, op_name);
   MS_EXCEPTION_IF_NULL(input_args[0]);
+  std::set<TypePtr> valid_params_types = {kTensorType};
+  CheckAndConvertUtils::CheckSubClass("shape type", input_args[0]->BuildType(), valid_params_types, op_name);
   auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape());
   auto inshape = shape_map[kShape];
   auto value = MakeValue(inshape);