|
|
|
@ -24,13 +24,7 @@ STATUS CaffeFlattenParser::Parse(const caffe::LayerParameter &proto, const caffe
|
|
|
|
|
// MS_LOG(ERROR) << "null pointer dereferencing.";
|
|
|
|
|
return RET_NULL_PTR;
|
|
|
|
|
}
|
|
|
|
|
std::unique_ptr<schema::FullConnectionT> attr(new schema::FullConnectionT());
|
|
|
|
|
const caffe::FlattenParameter flattenParam = proto.flatten_param();
|
|
|
|
|
|
|
|
|
|
attr->axis = (int32_t)flattenParam.axis();
|
|
|
|
|
attr->useAxis = true;
|
|
|
|
|
attr->hasBias = false;
|
|
|
|
|
attr->activationType = schema::ActivationType_NO_ACTIVATION;
|
|
|
|
|
std::unique_ptr<schema::FlattenT> attr = std::make_unique<schema::FlattenT>();
|
|
|
|
|
|
|
|
|
|
op->primitive = std::make_unique<schema::PrimitiveT>();
|
|
|
|
|
op->primitive->value.type = schema::PrimitiveType_Flatten;
|
|
|
|
|