Merge with master, works with Jetpack 4.3

Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-01-15 19:24:19 +01:00
parent 23a1365dc4 c32a0be257
commit 0f0c8c28e1
10 changed files with 59 additions and 35 deletions
@@ -124,7 +124,7 @@ dnnType* DeformConv2d::infer(dataDim_t &dim, dnnType* srcData) {
                                                dstTensorDesc, dstData, dstTensorDesc,
                                                dstData, biasTensorDesc, //same tensor descriptor as bias
                                                scales_d, bias_d, mean_d, variance_d,
-                                                CUDNN_BN_MIN_EPSILON) );
+                                                TKDNN_BN_MIN_EPSILON) );
    }

    //update data dimensions