Browse Source

fix_arm32_comple

tags/v1.0.0
sunsuodong 5 years ago
parent
commit
f50a235cee
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/nnacl/fp32/prelu.c

+ 1
- 1
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);


Loading…
Cancel
Save