diff options
Diffstat (limited to 'generic/SpatialConvolution.c')
-rw-r--r-- | generic/SpatialConvolution.c | 26 |
1 files changed, 13 insertions, 13 deletions
diff --git a/generic/SpatialConvolution.c b/generic/SpatialConvolution.c index 49ccc8d..bfe5698 100644 --- a/generic/SpatialConvolution.c +++ b/generic/SpatialConvolution.c @@ -4,13 +4,13 @@ static int nn_(SpatialConvolution_updateOutput)(lua_State *L) { - THTensor *input = luaT_checkudata(L, 2, torch_(Tensor_id)); + THTensor *input = luaT_checkudata(L, 2, torch_Tensor); int dW = luaT_getfieldcheckint(L, 1, "dW"); int dH = luaT_getfieldcheckint(L, 1, "dH"); - THTensor *weight = luaT_getfieldcheckudata(L, 1, "weight", torch_(Tensor_id)); - THTensor *bias = luaT_getfieldcheckudata(L, 1, "bias", torch_(Tensor_id)); - THTensor *output = luaT_getfieldcheckudata(L, 1, "output", torch_(Tensor_id)); + THTensor *weight = luaT_getfieldcheckudata(L, 1, "weight", torch_Tensor); + THTensor *bias = luaT_getfieldcheckudata(L, 1, "bias", torch_Tensor); + THTensor *output = luaT_getfieldcheckudata(L, 1, "output", torch_Tensor); luaL_argcheck(L, input->nDimension == 3 || input->nDimension == 4, 2, "3D or 4D(batch mode) tensor expected"); @@ -83,14 +83,14 @@ static int nn_(SpatialConvolution_updateOutput)(lua_State *L) static int nn_(SpatialConvolution_updateGradInput)(lua_State *L) { - THTensor *input = luaT_checkudata(L, 2, torch_(Tensor_id)); - THTensor *gradOutput = luaT_checkudata(L, 3, torch_(Tensor_id)); + THTensor *input = luaT_checkudata(L, 2, torch_Tensor); + THTensor *gradOutput = luaT_checkudata(L, 3, torch_Tensor); int dW = luaT_getfieldcheckint(L, 1, "dW"); int dH = luaT_getfieldcheckint(L, 1, "dH"); int nOutputPlane = luaT_getfieldcheckint(L, 1, "nOutputPlane"); - THTensor *weight = luaT_getfieldcheckudata(L, 1, "weight", torch_(Tensor_id)); - THTensor *gradInput = luaT_getfieldcheckudata(L, 1, "gradInput", torch_(Tensor_id)); + THTensor *weight = luaT_getfieldcheckudata(L, 1, "weight", torch_Tensor); + THTensor *gradInput = luaT_getfieldcheckudata(L, 1, "gradInput", torch_Tensor); THArgCheck( nOutputPlane == gradOutput->size[input->nDimension == 4 ? 1 : 0], 1, "Number of output features is not equal to nOutputPlane" ); @@ -112,15 +112,15 @@ static int nn_(SpatialConvolution_updateGradInput)(lua_State *L) static int nn_(SpatialConvolution_accGradParameters)(lua_State *L) { - THTensor *input = luaT_checkudata(L, 2, torch_(Tensor_id)); - THTensor *gradOutput = luaT_checkudata(L, 3, torch_(Tensor_id)); + THTensor *input = luaT_checkudata(L, 2, torch_Tensor); + THTensor *gradOutput = luaT_checkudata(L, 3, torch_Tensor); real scale = luaL_optnumber(L, 4, 1); int dW = luaT_getfieldcheckint(L, 1, "dW"); int dH = luaT_getfieldcheckint(L, 1, "dH"); int nOutputPlane = luaT_getfieldcheckint(L, 1, "nOutputPlane"); - THTensor *gradWeight = luaT_getfieldcheckudata(L, 1, "gradWeight", torch_(Tensor_id)); - THTensor *gradBias = luaT_getfieldcheckudata(L, 1, "gradBias", torch_(Tensor_id)); + THTensor *gradWeight = luaT_getfieldcheckudata(L, 1, "gradWeight", torch_Tensor); + THTensor *gradBias = luaT_getfieldcheckudata(L, 1, "gradBias", torch_Tensor); THArgCheck( nOutputPlane == gradOutput->size[input->nDimension == 4 ? 1 : 0], 1, "Number of output features is not equal to nOutputPlane" ); @@ -186,7 +186,7 @@ static const struct luaL_Reg nn_(SpatialConvolution__) [] = { static void nn_(SpatialConvolution_init)(lua_State *L) { - luaT_pushmetaclass(L, torch_(Tensor_id)); + luaT_pushmetatable(L, torch_Tensor); luaT_registeratname(L, nn_(SpatialConvolution__), "nn"); lua_pop(L,1); } |