Commit a7b502f0 authored by berak's avatar berak

dnn: preserve name, type strings for ShuffleLayer

parent 0bb2c115
......@@ -14,6 +14,7 @@ public:
ShuffleChannelLayerImpl(const LayerParams& params)
{
group = params.get<int>("group", 1);
setParamsFrom(params);
}
bool getMemoryShapes(const std::vector<MatShape> &inputs,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment