[NeoML] BaseLayer use BlobDesc.MemorySize

Signed-off-by: Kirill Golikov <[email protected]>
neoml-lib · Apr 25, 2024 · 6bd5ea7 · 6bd5ea7
1 parent d2a783b
commit 6bd5ea7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/NeoML/src/Dnn/BaseLayer.cpp b/NeoML/src/Dnn/BaseLayer.cpp
@@ -346,15 +346,15 @@ void CBaseLayer::reshape()
 {
 	NeoAssert( dnn != 0 ); // possible only in a network
 
-	if( !isReshapeNeeded && !forcedReshape) {
+	if( !isReshapeNeeded && !forcedReshape ) {
 		return;
 	}
 	isReshapeNeeded = false;
 
 	CArray<CBlobDesc> prevInputDescs;
-	inputDescs.MoveTo( prevInputDescs );
-	inputDescs.SetSize(inputs.Size());
-	
+	inputDescs.CopyTo( prevInputDescs ); // do not delete, do not loose the memory_size
+	inputDescs.SetSize( inputs.Size() );
+
 	// Call the input layers reshape recursively, reset the input blobs
 	for( int i = 0; i < GetInputCount(); ++i ) {
 		GetInputLayer(i)->reshape();