| |
| /* ---------------------------------------------------------------------- |
| * Project: CMSIS DSP Library |
| * Title: arm_cosine_distance_f16.c |
| * Description: Cosine distance between two vectors |
| * |
| * $Date: 23 April 2021 |
| * $Revision: V1.9.0 |
| * |
| * Target Processor: Cortex-M and Cortex-A cores |
| * -------------------------------------------------------------------- */ |
| /* |
| * Copyright (C) 2010-2021 ARM Limited or its affiliates. All rights reserved. |
| * |
| * SPDX-License-Identifier: Apache-2.0 |
| * |
| * Licensed under the Apache License, Version 2.0 (the License); you may |
| * not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an AS IS BASIS, WITHOUT |
| * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "dsp/distance_functions_f16.h" |
| |
| #if defined(ARM_FLOAT16_SUPPORTED) |
| |
| #include <limits.h> |
| #include <math.h> |
| |
| /** |
| @ingroup FloatDist |
| */ |
| |
| /** |
| @defgroup CosineDist Cosine distance |
| |
| Cosine distance |
| */ |
| |
| |
| /** |
| @addtogroup CosineDist |
| @{ |
| */ |
| |
| |
| |
| /** |
| * @brief Cosine distance between two vectors |
| * |
| * @param[in] pA First vector |
| * @param[in] pB Second vector |
| * @param[in] blockSize vector length |
| * @return distance |
| * |
| * @par Description |
| * cosine_distance(u,v) is 1 - u . v / (Norm(u) Norm(v)) |
| */ |
| |
| float16_t arm_cosine_distance_f16(const float16_t *pA,const float16_t *pB, uint32_t blockSize) |
| { |
| float16_t pwra,pwrb,dot,tmp; |
| |
| arm_power_f16(pA, blockSize, &pwra); |
| arm_power_f16(pB, blockSize, &pwrb); |
| |
| arm_dot_prod_f16(pA,pB,blockSize,&dot); |
| |
| arm_sqrt_f16((_Float16)pwra * (_Float16)pwrb, &tmp); |
| return(1.0f16 - (_Float16)dot / (_Float16)tmp); |
| |
| } |
| |
| |
| |
| /** |
| * @} end of CosineDist group |
| */ |
| |
| #endif /* #if defined(ARM_FLOAT16_SUPPORTED) */ |
| |