Fix conv2d with additional bias for tensorRT. Fix reshape deserialize. Mb2512 works with tensorRT

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-03-17 13:18:05 +01:00
parent 14d850751d
commit 1a50065bde
4 changed files with 23 additions and 6 deletions
@@ -445,8 +445,6 @@ int main()
    tk::dnn::Reshape reshape_conf2(&net, newdim_c);

    tk::dnn::Softmax sm_1(&net, &newdim_c, true);
-    // tk::dnn::Flatten fl_l_7(&net);
-    // tk::dnn::Reshape reshape_conf3(&net,dim_resh, true);
    tk::dnn::Layer *conf = &sm_1;

    //concat locations