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

fix warning

This commit is contained in:
duzhixing 2023-02-06 14:19:47 +08:00
parent 232bc67d26
commit 809ad272bc
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/lite/src/litert/kernel/cpu/int8/relux_int8.h
View File

@ -23,6 +23,9 @@
#include "nnacl/int8/relux_int8.h"
namespace mindspore::kernel {
constexpr size_t kRelu6Min = 0;
constexpr size_t kRelu6Max = 6;
class ReluXInt8CPUKernel : public LiteKernel {
public:
ReluXInt8CPUKernel(OpParameter *parameter, const std::vector<lite::Tensor *> &inputs,
@ -69,8 +72,10 @@ class Relu6Int8CPUKernel : public ReluXInt8CPUKernel {
int Prepare() override {
auto ret = ReluXInt8CPUKernel::Prepare();
quant_arg_.quantized_output_min = QuantizeToInt8(0, quant_arg_.output_arg.scale_, quant_arg_.output_arg.zp_);
quant_arg_.quantized_output_max = QuantizeToInt8(6, quant_arg_.output_arg.scale_, quant_arg_.output_arg.zp_);
quant_arg_.quantized_output_min =
QuantizeToInt8(kRelu6Min, quant_arg_.output_arg.scale_, quant_arg_.output_arg.zp_);
quant_arg_.quantized_output_max =
QuantizeToInt8(kRelu6Max, quant_arg_.output_arg.scale_, quant_arg_.output_arg.zp_);
return ret;
};
};

3
mindspore/lite/src/litert/kernel/cpu/int8/softmax_int8.cc
View File

@ -96,7 +96,8 @@ int SoftmaxInt8CPUKernel::DoSoftmax(int task_id) {
auto output_ptr = reinterpret_cast<int8_t *>(out_tensors_.at(0)->MutableData());
MS_ASSERT(output_ptr);
int outter_size = 1, inner_size = 1;
int outter_size = 1;
int inner_size = 1;
for (int i = 0; i < softmax_param_->axis_; i++) {
outter_size *= softmax_param_->input_shape_[i];
}