m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!5049 [MS][LITE][Develop]fix_arm32_comple

Browse Source 
Merge pull request !5049 from sunsuodong/fix_arm32_comple
pull/5049/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
parent 33a562de3d f50a235cee
commit 6c309d8f62
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
mindspore/lite/nnacl/fp32/prelu.c
Unescape 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_) { 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 *input_ptr = input + j * TILE_NUM * channel_num;
float *output_ptr = input_ptr; float *output_ptr = input_ptr;
#ifdef ENABLE_NEON #ifdef ENABLE_ARM64
for (int i = 0; i < c4; i++) { for (int i = 0; i < c4; i++) {
int c_offset = i * C4NUM; int c_offset = i * C4NUM;
float32x4_t slope_value = vld1q_f32(negetive_slope_value + c_offset); float32x4_t slope_value = vld1q_f32(negetive_slope_value + c_offset);

Write Preview
Loading…
Cancel
Save