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

add int8 operator support activation

This commit is contained in:
zengxianglong 2020-10-28 16:05:45 +08:00
parent 59162dc2b1
commit 54bdd9f827
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
mindspore/lite/src/runtime/kernel/arm/int8/add_int8.cc
View File

@ -54,8 +54,23 @@ int QuantizedAddCPUKernel::Init() {
QuantizeMultiplierSmallerThanOne(real_input1_multiplier, &para_.input1_multiplier_, &para_.input1_shift_);
QuantizeMultiplierSmallerThanOne(real_output_multiplier, &para_.output_multiplier_, &para_.output_shift_);
para_.output_activation_min_ = std::numeric_limits<int8_t>::min();
para_.output_activation_max_ = std::numeric_limits<int8_t>::max();
switch (arith_para_->activation_type_) {
case schema::ActivationType_RELU:
para_.output_activation_min_ = 0;
para_.output_activation_max_ = std::numeric_limits<int8_t>::max();
break;
case schema::ActivationType_RELU6:
para_.output_activation_min_ = 0;
para_.output_activation_max_ = 6;
break;
case schema::ActivationType_NO_ACTIVATION:
para_.output_activation_min_ = std::numeric_limits<int8_t>::min();
para_.output_activation_max_ = std::numeric_limits<int8_t>::max();
break;
default:
MS_LOG(ERROR) << "Add does not support activation type " << arith_para_->activation_type_;
return RET_ERROR;
}
int left_shift0 = -para_.input0_shift_ > 0 ? -para_.input0_shift_ : 0;
para_.right_shift0_ = -para_.input0_shift_ > 0 ? 0 : para_.input0_shift_;