cudnn5 branch for tx2

2017-07-14 11:11:51 +02:00
parent 33513468e6
commit e4df86a07c
9 changed files with 49 additions and 276 deletions
@@ -40,7 +40,7 @@ Pooling::Pooling(   Network *net, dataDim_t input_dim,
    }

    checkCUDNN( cudnnSetPooling2dDescriptor(poolingDesc, cudnnPoolingMode_t(pool_mode),
-                winH, winW, 0, 0, strideH, strideW) );
+                CUDNN_NOT_PROPAGATE_NAN, winH, winW, 0, 0, strideH, strideW) );

    checkCUDNN( cudnnSetTensor4dDescriptor(srcTensorDesc, 
                net->tensorFormat, net->dataType, n, c, h, w) );
@@ -108,4 +108,4 @@ value_type* Pooling::infer(dataDim_t &dim, value_type* srcData) {
    return dstData;
 }

-}
+}