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

Fix comment on grid width and height in dnn_sptf_sampler

上级 d9d2adce
...@@ -76,8 +76,8 @@ APPLY_SPECIFIC(dnn_sptf_sampler)(PyGpuArrayObject * input, ...@@ -76,8 +76,8 @@ APPLY_SPECIFIC(dnn_sptf_sampler)(PyGpuArrayObject * input,
out_dims[0] = (size_t) PyGpuArray_DIM(input, 0); // num_images out_dims[0] = (size_t) PyGpuArray_DIM(input, 0); // num_images
out_dims[1] = (size_t) PyGpuArray_DIM(input, 1); // num_channels out_dims[1] = (size_t) PyGpuArray_DIM(input, 1); // num_channels
out_dims[2] = (size_t) PyGpuArray_DIM(grid, 1); // grid width out_dims[2] = (size_t) PyGpuArray_DIM(grid, 1); // grid height
out_dims[3] = (size_t) PyGpuArray_DIM(grid, 2); // grid height out_dims[3] = (size_t) PyGpuArray_DIM(grid, 2); // grid width
if ( out_dims[0] == 0 || out_dims[1] == 0 || out_dims[2] == 0 || out_dims[3] == 0 ) if ( out_dims[0] == 0 || out_dims[1] == 0 || out_dims[2] == 0 || out_dims[3] == 0 )
{ {
...@@ -126,3 +126,4 @@ APPLY_SPECIFIC(dnn_sptf_sampler)(PyGpuArrayObject * input, ...@@ -126,3 +126,4 @@ APPLY_SPECIFIC(dnn_sptf_sampler)(PyGpuArrayObject * input,
return 0; return 0;
} }
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论