Remove dependency on nb_dims parameter in spatialtf descriptor

上级 fda82536
...@@ -27,7 +27,7 @@ int APPLY_SPECIFIC(spatialtf_desc)(cudnnSpatialTransformerDescriptor_t * desc, ...@@ -27,7 +27,7 @@ int APPLY_SPECIFIC(spatialtf_desc)(cudnnSpatialTransformerDescriptor_t * desc,
// Currently, only the bilinear sampler is supported by cuDNN, // Currently, only the bilinear sampler is supported by cuDNN,
// so it is not available as a parameter // so it is not available as a parameter
err = cudnnSetSpatialTransformerNdDescriptor( *desc, CUDNN_SAMPLER_BILINEAR, err = cudnnSetSpatialTransformerNdDescriptor( *desc, CUDNN_SAMPLER_BILINEAR,
params->dtype, params->nb_dims, out_tensor_dims ); params->dtype, 4, out_tensor_dims );
if ( CUDNN_STATUS_SUCCESS != err ) if ( CUDNN_STATUS_SUCCESS != err )
{ {
PyErr_Format( PyExc_MemoryError, PyErr_Format( PyExc_MemoryError,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论