mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Merge with master works

Browse Source 
Signed-off-by:  Micaela Verucchi <micaelaverucchi@gmail.com>
		Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-04-08 14:59:42 +02:00
parent 6ec417d89f 62c8f528a4
commit 0951f4644f
16 changed files with 948 additions and 190 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Route.cpp
+7 -4
View File
@@ -7,11 +7,14 @@ namespace tk { namespace dnn {
Route::Route(Network *net, Layer **layers, int layers_n, bool final) : Layer(net, final) {
this->layers_n = layers_n;
if(layers_n > MAX_INPUT_LAYERS)
FatalError("Route: MAX INPUT LAYERS overload");
for(int i=0; i<layers_n; i++)
// copy input layers
if(layers_n > MAX_LAYERS) {
FatalError("ROUTE: reached max number of input layers");
}
for(int i=0; i<layers_n; i++) {
this->layers[i] = layers[i];
}
this->layers_n = layers_n;
//get dims
output_dim.l = 1;