p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix_arm32_comple

This commit is contained in:
sunsuodong 2020-08-24 16:37:48 +08:00
parent 33fdc43f18
commit f50a235cee
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/nnacl/fp32/prelu.c
View File

@ -25,7 +25,7 @@ void PRelu(float *input, float *output, PReluParameter *prelu_param_, int task_i
for (int j = task_id; j < prelu_param_->tile_block_; j += prelu_param_->op_parameter_.thread_num_) {
float *input_ptr = input + j * TILE_NUM * channel_num;
float *output_ptr = input_ptr;
#ifdef ENABLE_NEON
#ifdef ENABLE_ARM64
for (int i = 0; i < c4; i++) {
int c_offset = i * C4NUM;
float32x4_t slope_value = vld1q_f32(negetive_slope_value + c_offset);