Merge with master, works with Jetpack 4.3

Signed-off-by: xavier <micaelaverucchi@gmail.com>
This commit is contained in:
xavier
2020-01-15 19:24:19 +01:00
10 changed files with 59 additions and 35 deletions
+1 -1
View File
@@ -113,7 +113,7 @@ void Conv2d::inferCUDNN(dnnType* srcData, bool back) {
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) );
}
}