From ad1efeed343a6f82593a15e78ea7e7bd6ceb041c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20K=C3=B6pf?= Date: Fri, 11 Dec 2015 23:47:10 +0100 Subject: Add THNN/ffi conversion of Abs --- generic/Abs.c | 41 ----------------------------------------- 1 file changed, 41 deletions(-) delete mode 100644 generic/Abs.c (limited to 'generic/Abs.c') diff --git a/generic/Abs.c b/generic/Abs.c deleted file mode 100644 index 0d258f8..0000000 --- a/generic/Abs.c +++ /dev/null @@ -1,41 +0,0 @@ -#ifndef TH_GENERIC_FILE -#define TH_GENERIC_FILE "generic/Abs.c" -#else - -static int nn_(Abs_updateOutput)(lua_State *L) -{ - THTensor *input = luaT_checkudata(L, 2, torch_Tensor); - THTensor *output = luaT_getfieldcheckudata(L, 1, "output", torch_Tensor); - - THTensor_(resizeAs)(output, input); - THTensor_(abs)(output, input); - return 1; -} - -static int nn_(Abs_updateGradInput)(lua_State *L) -{ - THTensor *input = luaT_checkudata(L, 2, torch_Tensor); - THTensor *gradOutput = luaT_checkudata(L, 3, torch_Tensor); - THTensor *gradInput = luaT_getfieldcheckudata(L, 1, "gradInput", torch_Tensor); - - THTensor_(resizeAs)(gradInput, input); - TH_TENSOR_APPLY3(real, gradInput, real, gradOutput, real, input, \ - real z = *input_data; \ - *gradInput_data = *gradOutput_data * (z >= 0 ? 1 : -1);) - return 1; -} - -static const struct luaL_Reg nn_(Abs__) [] = { - {"Abs_updateOutput", nn_(Abs_updateOutput)}, - {"Abs_updateGradInput", nn_(Abs_updateGradInput)}, - {NULL, NULL} -}; - -static void nn_(Abs_init)(lua_State *L) -{ - luaT_pushmetatable(L, torch_Tensor); - luaT_registeratname(L, nn_(Abs__), "nn"); - lua_pop(L,1); -} - -#endif -- cgit v1.2.3