Searched refs:cur_exp (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | power_fp32.cc | 75 float *cur_exp = nullptr; in RunImpl() local
77 cur_exp = exp_addr; in RunImpl()
79 cur_exp = exp_addr + stride * task_id; in RunImpl()
81 CHECK_NULL_RETURN(cur_exp); in RunImpl()
83 …Power(x_addr + stride * task_id, cur_exp, output_addr + stride * task_id, len, scale_, shift_, bro… in RunImpl()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | power_fp16.cc | 115 float16_t *cur_exp = nullptr; in RunImpl() local
117 cur_exp = exp_data_; in RunImpl()
119 cur_exp = exp_data_ + stride * task_id; in RunImpl()
121 CHECK_NULL_RETURN(cur_exp); in RunImpl()
123 …PowerFp16(x_addr + stride * task_id, cur_exp, output_addr + stride * task_id, len, scale_, shift_,… in RunImpl()
|