|
|
@ -59,9 +59,6 @@ int ElementSqrtFp16(float16_t *input, float16_t *output, int element_size) {
|
|
|
|
|
|
|
|
|
|
|
|
int ElementRsqrtFp16(float16_t *input, float16_t *output, int element_size) {
|
|
|
|
int ElementRsqrtFp16(float16_t *input, float16_t *output, int element_size) {
|
|
|
|
for (int i = 0; i < element_size; i++) {
|
|
|
|
for (int i = 0; i < element_size; i++) {
|
|
|
|
if (input[i] <= 0) {
|
|
|
|
|
|
|
|
return NNACL_ERRCODE_RSQRT_NEGATIVE_OR_ZERO;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
output[i] = 1.f / sqrtf(input[i]);
|
|
|
|
output[i] = 1.f / sqrtf(input[i]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return NNACL_OK;
|
|
|
|
return NNACL_OK;
|
|
|
|