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 * @par Description
64 * cosine_distance(u,v) is 1 - u . v / (Norm(u) Norm(v))
65 */
66
arm_cosine_distance_f16(const float16_t * pA,const float16_t * pB,uint32_t blockSize)67 float16_t arm_cosine_distance_f16(const float16_t *pA,const float16_t *pB, uint32_t blockSize)
68 {
69 float16_t pwra,pwrb,dot,tmp;
70
71 arm_power_f16(pA, blockSize, &pwra);
72 arm_power_f16(pB, blockSize, &pwrb);
73
74 arm_dot_prod_f16(pA,pB,blockSize,&dot);
75
76 arm_sqrt_f16((_Float16)pwra * (_Float16)pwrb, &tmp);
77 return(1.0f16 - (_Float16)dot / (_Float16)tmp);
78
79 }
80
81
82
83 /**
84 * @} end of CosineDist group
85 */
86
87 #endif /* #if defined(ARM_FLOAT16_SUPPORTED) */
88
89
