Rev 2 | Details | Compare with Previous | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 2 | mjames | 1 | /* |
| 2 | * Copyright (C) 2010-2018 Arm Limited or its affiliates. All rights reserved. |
||
| 3 | * |
||
| 4 | * SPDX-License-Identifier: Apache-2.0 |
||
| 5 | * |
||
| 6 | * Licensed under the Apache License, Version 2.0 (the License); you may |
||
| 7 | * not use this file except in compliance with the License. |
||
| 8 | * You may obtain a copy of the License at |
||
| 9 | * |
||
| 10 | * www.apache.org/licenses/LICENSE-2.0 |
||
| 11 | * |
||
| 12 | * Unless required by applicable law or agreed to in writing, software |
||
| 13 | * distributed under the License is distributed on an AS IS BASIS, WITHOUT |
||
| 14 | * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||
| 15 | * See the License for the specific language governing permissions and |
||
| 16 | * limitations under the License. |
||
| 17 | */ |
||
| 18 | |||
| 19 | /* ---------------------------------------------------------------------- |
||
| 20 | * Project: CMSIS NN Library |
||
| 21 | * Title: arm_nn_activations_q7.c |
||
| 22 | * Description: Q7 neural network activation function using direct table look-up |
||
| 23 | * |
||
| 24 | * $Date: 17. January 2018 |
||
| 25 | * $Revision: V.1.0.0 |
||
| 26 | * |
||
| 27 | * Target Processor: Cortex-M cores |
||
| 28 | * |
||
| 29 | * -------------------------------------------------------------------- */ |
||
| 30 | |||
| 31 | #include "arm_math.h" |
||
| 32 | #include "arm_common_tables.h" |
||
| 33 | #include "arm_nnfunctions.h" |
||
| 34 | |||
| 35 | /** |
||
| 36 | * @ingroup groupNN |
||
| 37 | */ |
||
| 38 | |||
| 39 | /** |
||
| 40 | * @addtogroup Acti |
||
| 41 | * @{ |
||
| 42 | */ |
||
| 43 | |||
| 44 | /** |
||
| 45 | * @brief Q7 neural network activation function using direct table look-up |
||
| 46 | * @param[in,out] data pointer to input |
||
| 47 | * @param[in] size number of elements |
||
| 48 | * @param[in] int_width bit-width of the integer part, assume to be smaller than 3 |
||
| 49 | * @param[in] type type of activation functions |
||
| 50 | * @return none. |
||
| 51 | * |
||
| 52 | * @details |
||
| 53 | * |
||
| 54 | * This is the direct table look-up approach. |
||
| 55 | * |
||
| 56 | * Assume here the integer part of the fixed-point is <= 3. |
||
| 57 | * More than 3 just not making much sense, makes no difference with |
||
| 58 | * saturation followed by any of these activation functions. |
||
| 59 | */ |
||
| 60 | |||
| 61 | void arm_nn_activations_direct_q7(q7_t * data, uint16_t size, uint16_t int_width, arm_nn_activation_type type) |
||
| 62 | { |
||
| 63 | uint16_t i = size; |
||
| 64 | q7_t *pIn = data; |
||
| 65 | q7_t *pOut = data; |
||
| 66 | q7_t in; |
||
| 67 | q7_t out; |
||
| 68 | uint16_t shift_size = 3 - int_width; |
||
| 69 | const q7_t *lookup_table; |
||
| 70 | switch (type) |
||
| 71 | { |
||
| 72 | case ARM_SIGMOID: |
||
| 73 | lookup_table = sigmoidTable_q7; |
||
| 74 | break; |
||
| 75 | case ARM_TANH: |
||
| 76 | default: |
||
| 77 | lookup_table = tanhTable_q7; |
||
| 78 | break; |
||
| 79 | } |
||
| 80 | while (i) |
||
| 81 | { |
||
| 82 | in = *pIn++; |
||
| 83 | out = lookup_table[(uint8_t) (in >> shift_size)]; |
||
| 84 | *pOut++ = out; |
||
| 85 | i--; |
||
| 86 | } |
||
| 87 | } |
||
| 88 | |||
| 89 | /** |
||
| 90 | * @} end of Acti group |
||
| 91 | */ |