p85971406
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!46197 fix cannot create base operator for Histogram. 

Merge pull request !46197 from yangshuo/fix_histogram
This commit is contained in:
i-robot 2022-11-30 01:41:05 +00:00 committed by Gitee
parent f7efff439c f3254263b3
commit 3281b7390e
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/core/ops/histogram.cc
View File

@ -78,7 +78,7 @@ AbstractBasePtr HistogramInfer(const abstract::AnalysisEnginePtr &, const Primit
return abstract::MakeAbstract(shapes, types);
}
MIND_API_BASE_IMPL(Histogram, PrimitiveC, BaseOperator);
MIND_API_OPERATOR_IMPL(Histogram, BaseOperator);
REGISTER_PRIMITIVE_EVAL_IMPL(Histogram, prim::kPrimHistogram, HistogramInfer, nullptr, true);
} // namespace ops
} // namespace mindspore