diff --git a/src/common/util/timlUtilSigmoidDerivative.c b/src/common/util/timlUtilSigmoidDerivative.c
index 3de086f9c39ec621ccfaf4df78364f9c89460df6..6ed90cff132f4cbd027277d7f43e032467017e17 100644 (file)
******************************************************************************/\r
\r
#include "../api/timl.h"\r
-#ifdef TIML_ALT\r
+#ifdef TIML_CPU_ALT\r
#include "../../alt/timlAlt.h"\r
+#elif defined TIML_ARM_DSP\r
+#include "../../dsp/timlDSP.h"\r
#endif\r
\r
\r
\r
int timlUtilSigmoidDerivative(float *x, float *y, int n, int deviceId, int threadId)\r
{\r
-#ifdef TIML_CPU\r
+#ifdef TIML_CPU_ALT\r
+ return timlUtilSigmoidDerivativeAlt(x, y, n, deviceId, threadId);\r
+#else\r
int i;\r
float tmp;\r
for (i = 0; i < n; i++) {\r
y[i] = (1.0 - tmp)*tmp;\r
}\r
return 0;\r
-#elif defined TIML_ALT\r
- return timlUtilSigmoidDerivativeAlt(x, y, n, deviceId, threadId);\r
+\r
#endif\r
}\r