diff options
Diffstat (limited to 'generic/TemporalConvolution.c')
| -rw-r--r-- | generic/TemporalConvolution.c | 26 |
1 files changed, 13 insertions, 13 deletions
diff --git a/generic/TemporalConvolution.c b/generic/TemporalConvolution.c index fa14a22..091a0cd 100644 --- a/generic/TemporalConvolution.c +++ b/generic/TemporalConvolution.c @@ -4,15 +4,15 @@ static int nn_(TemporalConvolution_updateOutput)(lua_State *L) { - THTensor *input = luaT_checkudata(L, 2, torch_(Tensor_id)); + THTensor *input = luaT_checkudata(L, 2, torch_Tensor); int kW = luaT_getfieldcheckint(L, 1, "kW"); int dW = luaT_getfieldcheckint(L, 1, "dW"); int inputFrameSize = luaT_getfieldcheckint(L, 1, "inputFrameSize"); int outputFrameSize = luaT_getfieldcheckint(L, 1, "outputFrameSize"); - 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); THTensor *outputWindow, *inputWindow; int nInputFrame, nOutputFrame; @@ -72,15 +72,15 @@ static int nn_(TemporalConvolution_updateOutput)(lua_State *L) static int nn_(TemporalConvolution_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 kW = luaT_getfieldcheckint(L, 1, "kW"); int dW = luaT_getfieldcheckint(L, 1, "dW"); long nInputFrame = input->size[0]; long nOutputFrame = gradOutput->size[0]; - 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); THTensor *gradOutputWindow; THTensor *gradInputWindow; @@ -121,16 +121,16 @@ static int nn_(TemporalConvolution_updateGradInput)(lua_State *L) static int nn_(TemporalConvolution_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 kW = luaT_getfieldcheckint(L, 1, "kW"); int dW = luaT_getfieldcheckint(L, 1, "dW"); long nInputFrame = input->size[0]; long nOutputFrame = gradOutput->size[0]; - 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); THTensor *gradOutputWindow; THTensor *inputWindow; @@ -186,7 +186,7 @@ static const struct luaL_Reg nn_(TemporalConvolution__) [] = { static void nn_(TemporalConvolution_init)(lua_State *L) { - luaT_pushmetaclass(L, torch_(Tensor_id)); + luaT_pushmetatable(L, torch_Tensor); luaT_registeratname(L, nn_(TemporalConvolution__), "nn"); lua_pop(L,1); } |