Browse Source

fix caffe reduce parser bug

tags/v1.2.0-rc1
cjh9368 5 years ago
parent
commit
4ebdccad11
1 changed files with 1 additions and 0 deletions
  1. +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;


Loading…
Cancel
Save