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
@@ -114,6 +114,7 @@ public:
    //fp16
    __half *data16_h, *bias16_h;
    __half *data16_d, *bias16_d;
+    __half *bias216_h, *bias216_d;

    __half *power16_h,    *power16_d;
    __half *scales16_h,   *scales16_d;