This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
fix caffe reduce parser bug
tags/v1.2.0-rc1
cjh9368
5 years ago
parent
3708624a25
commit
4ebdccad11
1 changed files
with
1 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-0
mindspore/lite/tools/converter/parser/caffe/caffe_reduce_parser.cc
+ 1
- 0
mindspore/lite/tools/converter/parser/caffe/caffe_reduce_parser.cc
View File
@@ -55,6 +55,7 @@ PrimitiveC *CaffeReduceParser::ParseLitePrimitive(const caffe::LayerParameter &p
axes = std::vector<int>(1, 0);
}
attr->axes = axes;
attr->reduceToEnd = true;
auto primitive = std::make_unique<schema::PrimitiveT>();
primitive->value.type = schema::PrimitiveType_Reduce;
Write
Preview
Loading…
Cancel
Save