lib/THNN/generic/THNN.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

#ifndef TH_GENERIC_FILE
#define TH_GENERIC_FILE "generic/THNN.h"
#else

#ifndef THIndexTensor
#define THIndexTensor THLongTensor
#endif

TH_API void THNN_(Abs_updateOutput)(
          THNNState *state,
          THTensor *input,
          THTensor *output);
TH_API void THNN_(Abs_updateGradInput)(
          THNNState *state,
          THTensor *input,
          THTensor *gradOutput,
          THTensor *gradInput);

#endif