1 /******************************************************************************/\r
2 /*!\r
3 * \file timlUtilSoftmax.c\r
4 */\r
5 /* Copyright (C) 2015 Texas Instruments Incorporated - http://www.ti.com/\r
6 *\r
7 * Redistribution and use in source and binary forms, with or without\r
8 * modification, are permitted provided that the following conditions\r
9 * are met:\r
10 *\r
11 * Redistributions of source code must retain the above copyright\r
12 * notice, this list of conditions and the following disclaimer.\r
13 *\r
14 * Redistributions in binary form must reproduce the above copyright\r
15 * notice, this list of conditions and the following disclaimer in the\r
16 * documentation and/or other materials provided with the\r
17 * distribution.\r
18 *\r
19 * Neither the name of Texas Instruments Incorporated nor the names of\r
20 * its contributors may be used to endorse or promote products derived\r
21 * from this software without specific prior written permission.\r
22 *\r
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\r
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\r
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\r
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\r
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\r
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\r
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\r
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\r
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\r
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\r
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\r
34 *\r
35 ******************************************************************************/\r
36 \r
37 \r
38 /*******************************************************************************\r
39 *\r
40 * INCLUDES\r
41 *\r
42 ******************************************************************************/\r
43 \r
44 #include "../api/timl.h"\r
45 #ifdef TIML_ALT\r
46 #include "../../alt/timlAlt.h"\r
47 #endif\r
48 \r
49 \r
50 /******************************************************************************/\r
51 /*!\r
52 * \ingroup util\r
53 * \brief Softmax function\r
54 * \param[in] x Input\r
55 * \param[out] y Outupt\r
56 * \param[in] row x row size\r
57 * \param[in] col x col size\r
58 * \param[in] channel x channel size\r
59 * \param[in] deviceId Device id\r
60 * \param[in] threadId Thread id\r
61 * \return Error code\r
62 */\r
63 /******************************************************************************/\r
64 \r
65 int timlUtilSoftmax(float *x, float *y, int row, int col, int channel, int deviceId, int threadId)\r
66 {\r
67 #ifdef TIML_CPU\r
68 int i;\r
69 int r;\r
70 int c;\r
71 float max;\r
72 float sum;\r
73 int increment;\r
74 increment = row*col;\r
75 for (r = 0; r < row; r++) {\r
76 for (c = 0; c < col; c++) {\r
77 max = timlUtilVectorMaxFloat(x + r*col + c, channel, row*col);\r
78 sum = 0.0;\r
79 for (i = 0; i < channel; i++) {\r
80 sum += expf(x[i*increment + r*col + c] - max);\r
81 }\r
82 for (i = 0; i < channel; i++) {\r
83 y[i*increment + r*col + c] = expf(x[i*increment + r*col + c] - max)/sum;\r
84 }\r
85 }\r
86 }\r
87 \r
88 return 0;\r
89 #elif defined TIML_ALT\r
90 return timlUtilSoftmaxAlt(x, y, row, col, channel, deviceId, threadId);\r
91 #endif\r
92 }\r