提交 efb57395 authored 作者: João Victor Risso's avatar João Victor Risso

Use direct pointer to dtheta_dims in dnn_sptf_gt

上级 c25df18f
......@@ -13,7 +13,7 @@ APPLY_SPECIFIC(dnn_sptf_gt)(PyGpuArrayObject * dgrid,
const size_t dtheta_dims[3] = { num_images, 2, 3 };
if ( theano_prep_output( dtheta, 3, &(dtheta_dims[0]), dgrid->ga.typecode,
if ( theano_prep_output( dtheta, 3, dtheta_dims, dgrid->ga.typecode,
GA_C_ORDER, gpu_ctx ) != 0 )
return 1;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论