Browse Source

fix pclint clean

tags/v1.6.0
liutongtong 4 years ago
parent
commit
1fc11c9599
4 changed files with 10 additions and 4 deletions
  1. +2
    -1
      mindspore/core/ops/grad/max_pool_grad.cc
  2. +3
    -1
      mindspore/core/ops/grad/max_pool_grad.h
  3. +2
    -1
      mindspore/core/ops/non_max_suppression.cc
  4. +3
    -1
      mindspore/core/ops/non_max_suppression.h

+ 2
- 1
mindspore/core/ops/grad/max_pool_grad.cc View File

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

namespace mindspore {
namespace ops {
AbstractBasePtr MaxPoolGradInfer(const std::vector<AbstractBasePtr> &input_args) {
AbstractBasePtr MaxPoolGradInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<AbstractBasePtr> &input_args) {
MS_EXCEPTION_IF_NULL(input_args[0]->BuildValue());
auto x1_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kShape];
auto tensor_type = input_args[0]->BuildType()->cast<TensorTypePtr>();


+ 3
- 1
mindspore/core/ops/grad/max_pool_grad.h View File

@@ -35,7 +35,9 @@ class MS_CORE_API MaxPoolGrad : public PoolGrad {
MS_DECLARE_PARENT(MaxPoolGrad, PoolGrad);
};

AbstractBasePtr MaxPoolGradInfer(const std::vector<AbstractBasePtr> &input_args);
AbstractBasePtr MaxPoolGradInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<AbstractBasePtr> &input_args);
using PrimMaxPoolGradPtr = std::shared_ptr<MaxPoolGrad>;
} // namespace ops
} // namespace mindspore



+ 2
- 1
mindspore/core/ops/non_max_suppression.cc View File

@@ -29,7 +29,8 @@ int64_t NonMaxSuppression::get_center_point_box() const {
}
void NonMaxSuppression::Init(const int64_t center_point_box) { this->set_center_point_box(center_point_box); }

AbstractBasePtr NonMaxSuppressionInfer(const abstract::AnalysisEnginePtr &) {
AbstractBasePtr NonMaxSuppressionInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<AbstractBasePtr> &input_args) {
MS_LOG(INFO) << "NonMaxSuppression infer shape in runtime.";
return std::make_shared<abstract::AbstractTensor>(kInt32, std::vector<int64_t>{});
}


+ 3
- 1
mindspore/core/ops/non_max_suppression.h View File

@@ -40,7 +40,9 @@ class MS_CORE_API NonMaxSuppression : public PrimitiveC {
void set_center_point_box(const int64_t center_point_box);
int64_t get_center_point_box() const;
};
AbstractBasePtr NonMaxSuppressionInfer(const abstract::AnalysisEnginePtr &);
AbstractBasePtr NonMaxSuppressionInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<AbstractBasePtr> &input_args);
using PrimNonMaxSuppressionPtr = std::shared_ptr<NonMaxSuppression>;
} // namespace ops
} // namespace mindspore



Loading…
Cancel
Save