Browse Source

fix softmax int8 coder

tags/v1.3.0
yangjie159 4 years ago
parent
commit
d429a73404
2 changed files with 5 additions and 5 deletions
  1. +1
    -1
      mindspore/lite/micro/coder/opcoders/nnacl/int8/softmax_int8_coder.cc
  2. +4
    -4
      mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/nnacl_int8_serializer.cc

+ 1
- 1
mindspore/lite/micro/coder/opcoders/nnacl/int8/softmax_int8_coder.cc View File

@@ -90,7 +90,7 @@ int SoftMaxInt8Coder::DoCode(CoderContext *const context) {
MS_CHECK_TRUE(thread_num_ > 0, "thread_num_ <= 0");
int stride = UP_DIV(outter_size, thread_num_);
int count = MSMIN(stride, outter_size - stride * kDefaultTaskId);
code.CodeFunction("SoftmaxInt8", input_tensor_, output_tensor_, count, exp_data_, sum_data_, "quant_args",
code.CodeFunction("SoftmaxInt8", input_tensor_, output_tensor_, count, exp_data_, sum_data_, "&quant_args",
"(SoftmaxParameter *)&softmax_parameter");
context->AppendCode(code.str());
return RET_OK;


+ 4
- 4
mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/nnacl_int8_serializer.cc View File

@@ -140,10 +140,10 @@ void NNaclInt8Serializer::CodeStruct(const std::string &name, const BatchNormPar
}

void NNaclInt8Serializer::CodeStruct(const std::string &name, const SoftmaxQuantArg &softmax_quant_parameter) {
CodeBaseStruct("SoftmaxQuantArg", name, softmax_quant_parameter.in_quant_args_,
softmax_quant_parameter.out_quant_arg_, softmax_quant_parameter.output_activation_min_,
softmax_quant_parameter.output_activation_max_, softmax_quant_parameter.output_multiplier_,
softmax_quant_parameter.shift_left_, softmax_quant_parameter.shift_right_);
CodeBaseStruct<false>("SoftmaxQuantArg", name, softmax_quant_parameter.in_quant_args_,
softmax_quant_parameter.out_quant_arg_, softmax_quant_parameter.output_activation_min_,
softmax_quant_parameter.output_activation_max_, softmax_quant_parameter.output_multiplier_,
softmax_quant_parameter.shift_left_, softmax_quant_parameter.shift_right_);
}

void NNaclInt8Serializer::CodeStruct(const std::string &name, const ConcatParameter &concat_parameter,


Loading…
Cancel
Save