55 lines
1.6 KiB
C
55 lines
1.6 KiB
C
#include "ref.h"
|
|
|
|
riscv_status ref_mat_scale_f32(const riscv_matrix_instance_f32 *pSrc,
|
|
float32_t scale, riscv_matrix_instance_f32 *pDst)
|
|
{
|
|
uint32_t i;
|
|
uint32_t numSamples; /* total number of elements in the matrix */
|
|
|
|
/* Total number of samples in the input matrix */
|
|
numSamples = (uint32_t)pSrc->numRows * pSrc->numCols;
|
|
|
|
for (i = 0; i < numSamples; i++) {
|
|
pDst->pData[i] = pSrc->pData[i] * scale;
|
|
}
|
|
|
|
return RISCV_MATH_SUCCESS;
|
|
}
|
|
|
|
riscv_status ref_mat_scale_q31(const riscv_matrix_instance_q31 *pSrc, q31_t scale,
|
|
int32_t shift, riscv_matrix_instance_q31 *pDst)
|
|
{
|
|
uint32_t i;
|
|
uint32_t numSamples; /* total number of elements in the matrix */
|
|
int32_t totShift = shift + 1;
|
|
q31_t tmp;
|
|
|
|
/* Total number of samples in the input matrix */
|
|
numSamples = (uint32_t)pSrc->numRows * pSrc->numCols;
|
|
|
|
for (i = 0; i < numSamples; i++) {
|
|
tmp = ((q63_t)pSrc->pData[i] * scale) >> 32;
|
|
pDst->pData[i] = ref_sat_q31((q63_t)tmp << totShift);
|
|
}
|
|
|
|
return RISCV_MATH_SUCCESS;
|
|
}
|
|
|
|
riscv_status ref_mat_scale_q15(const riscv_matrix_instance_q15 *pSrc, q15_t scale,
|
|
int32_t shift, riscv_matrix_instance_q15 *pDst)
|
|
{
|
|
uint32_t i;
|
|
uint32_t numSamples; /* total number of elements in the matrix */
|
|
int32_t totShift = 15 - shift;
|
|
|
|
/* Total number of samples in the input matrix */
|
|
numSamples = (uint32_t)pSrc->numRows * pSrc->numCols;
|
|
|
|
for (i = 0; i < numSamples; i++) {
|
|
pDst->pData[i] =
|
|
ref_sat_q15(((q31_t)pSrc->pData[i] * scale) >> totShift);
|
|
}
|
|
|
|
return RISCV_MATH_SUCCESS;
|
|
}
|