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
arm_logsumexp_dot_prod_f16(const float16_t * pSrcA,const float16_t * pSrcB,uint32_t blockSize,float16_t * pTmpBuffer)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
85