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

!49513 Support cyber model convert 

Merge pull request !49513 from JuiceZ/dev_0228
This commit is contained in:
i-robot 2023-03-02 04:12:30 +00:00 committed by Gitee
parent e1eae67540 cfddf4608f
commit ce84f55dea
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/lite/tools/converter/adapter/acl/mapper/reduce_fusion_mapper.cc
View File

@ -68,7 +68,7 @@ STATUS ReduceFusionMapper::Mapper(const CNodePtr &cnode) {
if (axes_ptr != nullptr) {
auto axes = GetValue<std::vector<int32_t>>(axes_ptr);
std::vector<int64_t> axes_vec;
std::transform(axes.begin(), axes.end(), std::back_inserter(axes),
std::transform(axes.begin(), axes.end(), std::back_inserter(axes_vec),
[](int32_t x) { return static_cast<int64_t>(x); });
lp_norm_op.set_axis(axes_vec);
}