Skip to content

P
pytensor
提交 c02d556a authored 作者: João Victor Risso's avatar João Victor Risso
浏览文件
操作

Rename dims to out_dims in dnn_sptf_desc

Signed-off-by: 's avatarJoão Victor Risso <joaovictor.risso@gmail.com>
上级 f669bd8e
显示空白字符变更
内嵌 并排
正在显示 包含 5 行增加 和 5 行删除
theano/gpuarray/c_code/dnn_sptf_desc.c
浏览文件 @ c02d556a
#section support_code_apply #section support_code_apply
int APPLY_SPECIFIC(dnn_sptf_desc)(PyArrayObject * dims, int APPLY_SPECIFIC(dnn_sptf_desc)(PyArrayObject * out_dims,
cudnnSpatialTransformerDescriptor_t * desc, cudnnSpatialTransformerDescriptor_t * desc,
PARAMS_TYPE * params) PARAMS_TYPE * params)
{ {
cudnnStatus_t err; cudnnStatus_t err;
const int nimages = (int) *((npy_int64 *) PyArray_GETPTR1(dims, 0)); const int nimages = (int) *((npy_int64 *) PyArray_GETPTR1(out_dims, 0));
const int nchannels = (int) *((npy_int64 *) PyArray_GETPTR1(dims, 1)); const int nchannels = (int) *((npy_int64 *) PyArray_GETPTR1(out_dims, 1));
const int height = (int) *((npy_int64 *) PyArray_GETPTR1(dims, 2)); const int height = (int) *((npy_int64 *) PyArray_GETPTR1(out_dims, 2));
const int width = (int) *((npy_int64 *) PyArray_GETPTR1(dims, 3)); const int width = (int) *((npy_int64 *) PyArray_GETPTR1(out_dims, 3));
if ( nimages == 0 || nchannels == 0 || height == 0 || width == 0 ) if ( nimages == 0 || nchannels == 0 || height == 0 || width == 0 )
{ {
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论