1
2 /* ----------------------------------------------------------------------
3 * Project: CMSIS DSP Library
4 * Title: arm_cosine_distance_f16.c
5 * Description: Cosine distance between two vectors
6 *
7 * $Date: 23 April 2021
8 * $Revision: V1.9.0
9 *
10 * Target Processor: Cortex-M and Cortex-A cores
11 * -------------------------------------------------------------------- */
12 /*
13 * Copyright (C) 2010-2021 ARM Limited or its affiliates. All rights reserved.
14 *
15 * SPDX-License-Identifier: Apache-2.0
16 *
17 * Licensed under the Apache License, Version 2.0 (the License); you may
18 * not use this file except in compliance with the License.
19 * You may obtain a copy of the License at
20 *
21 * www.apache.org/licenses/LICENSE-2.0
22 *
23 * Unless required by applicable law or agreed to in writing, software
24 * distributed under the License is distributed on an AS IS BASIS, WITHOUT
25 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
26 * See the License for the specific language governing permissions and
27 * limitations under the License.
28 */
29
30 #include "dsp/distance_functions_f16.h"
31
32 #if defined(ARM_FLOAT16_SUPPORTED)
33
34 #include <limits.h>
35 #include <math.h>
36
37 /**
38 @ingroup FloatDist
39 */
40
41 /**
42 @defgroup CosineDist Cosine distance
43
44 Cosine distance
45 */
46
47
48 /**
49 @addtogroup CosineDist
50 @{
51 */
52
53
54
55 /**
56 * @brief Cosine distance between two vectors
57 *
58 * @param[in] pA First vector
59 * @param[in] pB Second vector
60 * @param[in] blockSize vector length
61 * @return distance
62 *
63 */
64
arm_cosine_distance_f16(const float16_t * pA,const float16_t * pB,uint32_t blockSize)65 float16_t arm_cosine_distance_f16(const float16_t *pA,const float16_t *pB, uint32_t blockSize)
66 {
67 float16_t pwra,pwrb,dot,tmp;
68
69 arm_power_f16(pA, blockSize, &pwra);
70 arm_power_f16(pB, blockSize, &pwrb);
71
72 arm_dot_prod_f16(pA,pB,blockSize,&dot);
73
74 arm_sqrt_f16(pwra * pwrb, &tmp);
75 return(1.0f - dot / tmp);
76
77 }
78
79
80
81 /**
82 * @} end of CosineDist group
83 */
84
85 #endif /* #if defined(ARM_FLOAT16_SUPPORTED) */
86
87
