diff options
Diffstat (limited to 'generic/HardShrink.c')
-rw-r--r-- | generic/HardShrink.c | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/generic/HardShrink.c b/generic/HardShrink.c index d3bcfcd..d75e071 100644 --- a/generic/HardShrink.c +++ b/generic/HardShrink.c @@ -2,7 +2,7 @@ #define TH_GENERIC_FILE "generic/HardShrink.c" #else -static int nn_(HardShrink_forward)(lua_State *L) +static int nn_(HardShrink_updateOutput)(lua_State *L) { THTensor *input = luaT_checkudata(L, 2, torch_(Tensor_id)); THTensor *output = luaT_getfieldcheckudata(L, 1, "output", torch_(Tensor_id)); @@ -16,7 +16,7 @@ static int nn_(HardShrink_forward)(lua_State *L) return 1; } -static int nn_(HardShrink_backward)(lua_State *L) +static int nn_(HardShrink_updateGradInput)(lua_State *L) { THTensor *input = luaT_checkudata(L, 2, torch_(Tensor_id)); THTensor *gradOutput = luaT_checkudata(L, 3, torch_(Tensor_id)); @@ -32,8 +32,8 @@ static int nn_(HardShrink_backward)(lua_State *L) } static const struct luaL_Reg nn_(HardShrink__) [] = { - {"HardShrink_forward", nn_(HardShrink_forward)}, - {"HardShrink_backward", nn_(HardShrink_backward)}, + {"HardShrink_updateOutput", nn_(HardShrink_updateOutput)}, + {"HardShrink_updateGradInput", nn_(HardShrink_updateGradInput)}, {NULL, NULL} }; |