Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/torch/nn.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'generic/SpatialConvolution.c')
-rw-r--r--generic/SpatialConvolution.c26
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);
}