|
|
@ -2506,7 +2506,9 @@ struct Net::Impl |
|
|
|
{ |
|
|
|
{ |
|
|
|
std::vector<LayerPin>& inputLayerIds = layers[id].inputBlobsId; |
|
|
|
std::vector<LayerPin>& inputLayerIds = layers[id].inputBlobsId; |
|
|
|
|
|
|
|
|
|
|
|
if (inOutShapes[0].in[0].empty() && !layers[0].outputBlobs.empty()) |
|
|
|
if (id == 0 && inOutShapes[id].in[0].empty()) |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
if (!layers[0].outputBlobs.empty()) |
|
|
|
{ |
|
|
|
{ |
|
|
|
ShapesVec shapes; |
|
|
|
ShapesVec shapes; |
|
|
|
for (int i = 0; i < layers[0].outputBlobs.size(); i++) |
|
|
|
for (int i = 0; i < layers[0].outputBlobs.size(); i++) |
|
|
@ -2517,6 +2519,12 @@ struct Net::Impl |
|
|
|
} |
|
|
|
} |
|
|
|
inOutShapes[0].in = shapes; |
|
|
|
inOutShapes[0].in = shapes; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
else |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
inOutShapes[0].out.clear(); |
|
|
|
|
|
|
|
return; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
if (inOutShapes[id].in.empty()) |
|
|
|
if (inOutShapes[id].in.empty()) |
|
|
|
{ |
|
|
|
{ |
|
|
|