1  /* ----------------------------------------------------------------------
 2   * Project:      CMSIS DSP Library
 3   * Title:        arm_logsumexp_f16.c
 4   * Description:  LogSumExp
 5   *
 6   * $Date:        23 April 2021
 7   * $Revision:    V1.9.0
 8   *
 9   * Target Processor: Cortex-M and Cortex-A cores
10   * -------------------------------------------------------------------- */
11  /*
12   * Copyright (C) 2010-2021 ARM Limited or its affiliates. All rights reserved.
13   *
14   * SPDX-License-Identifier: Apache-2.0
15   *
16   * Licensed under the Apache License, Version 2.0 (the License); you may
17   * not use this file except in compliance with the License.
18   * You may obtain a copy of the License at
19   *
20   * www.apache.org/licenses/LICENSE-2.0
21   *
22   * Unless required by applicable law or agreed to in writing, software
23   * distributed under the License is distributed on an AS IS BASIS, WITHOUT
24   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
25   * See the License for the specific language governing permissions and
26   * limitations under the License.
27   */
28  
29  #include "dsp/statistics_functions_f16.h"
30  
31  #if defined(ARM_FLOAT16_SUPPORTED)
32  
33  #include <limits.h>
34  #include <math.h>
35  
36  /**
37    @ingroup groupStats
38   */
39  
40  /**
41    @defgroup LogSumExp LogSumExp
42  
43    LogSumExp optimizations to compute sum of probabilities with Gaussian distributions
44  
45   */
46  
47  /**
48   * @addtogroup LogSumExp
49   * @{
50   */
51  
52  
53  /**
54   * @brief Dot product with log arithmetic
55   *
56   * Vectors are containing the log of the samples
57   *
58   * @param[in]       *pSrcA points to the first input vector
59   * @param[in]       *pSrcB points to the second input vector
60   * @param[in]       blockSize number of samples in each vector
61   * @param[in]       *pTmpBuffer temporary buffer of length blockSize
62   * @return The log of the dot product.
63   *
64   */
65  
66  
67  float16_t arm_logsumexp_dot_prod_f16(const float16_t * pSrcA,
68    const float16_t * pSrcB,
69    uint32_t blockSize,
70    float16_t *pTmpBuffer)
71  {
72      float16_t result;
73      arm_add_f16((float16_t*)pSrcA, (float16_t*)pSrcB, pTmpBuffer, blockSize);
74  
75      result = arm_logsumexp_f16(pTmpBuffer, blockSize);
76      return(result);
77  }
78  
79  /**
80   * @} end of LogSumExp group
81   */
82  
83  #endif /* #if defined(ARM_FLOAT16_SUPPORTED) */ 
84