Home
last modified time | relevance | path

Searched refs:out_quant_args (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Ddepth_to_space_int8.cc63 auto out_quant_args = out_tensor->quant_params(); in Init() local
64 CHECK_LESS_RETURN(out_quant_args.size(), 1); in Init()
65 out_quant_arg_->scale_ = out_quant_args.front().scale; in Init()
66 out_quant_arg_->zp_ = out_quant_args.front().zeroPoint; in Init()
Dslice_int8.cc44 auto out_quant_args = output->quant_params(); in Init() local
45 CHECK_LESS_RETURN(out_quant_args.size(), 1); in Init()
46 param_->quant_arg_.out_args_.scale_ = out_quant_args.front().scale; in Init()
47 param_->quant_arg_.out_args_.zp_ = out_quant_args.front().zeroPoint; in Init()
Dgather_int8.cc38 auto out_quant_args = out_tensors_.at(0)->quant_params(); in Init() local
39 CHECK_LESS_RETURN(out_quant_args.size(), 1); in Init()
40 param_.alpha_ = in_quant_args.front().scale / out_quant_args.front().scale; in Init()
42 param_.zp_out_ = out_quant_args.front().zeroPoint; in Init()
Dsplit_int8.cc51 auto out_quant_args = out_tensor->quant_params(); in Init() local
52 CHECK_LESS_RETURN(out_quant_args.size(), 1); in Init()
53 param->quant_arg_.out_args_[i].scale_ = out_quant_args.front().scale; in Init()
54 param->quant_arg_.out_args_[i].zp_ = out_quant_args.front().zeroPoint; in Init()
Dpower_int8.cc43 auto out_quant_args = output->quant_params(); in Init() local
44 CHECK_LESS_RETURN(out_quant_args.size(), 1); in Init()
45 param_->quant_arg_.out_args_.scale_ = out_quant_args.front().scale; in Init()
46 param_->quant_arg_.out_args_.zp_ = out_quant_args.front().zeroPoint; in Init()
Dcrop_int8.cc39 auto out_quant_args = out_tensor->quant_params(); in Init() local
40 crop_para_->quant_arg.out_args_.scale_ = out_quant_args.front().scale; in Init()
41 crop_para_->quant_arg.out_args_.zp_ = out_quant_args.front().zeroPoint; in Init()
Darithmetic_self_int8.cc40 auto out_quant_args = out_tensor->quant_params(); in Init() local
41 CHECK_LESS_RETURN(out_quant_args.size(), 1); in Init()
42 para_->quant_arg_.out_args_.scale_ = out_quant_args.front().scale; in Init()
43 para_->quant_arg_.out_args_.zp_ = out_quant_args.front().zeroPoint; in Init()
Dargminmax_int8.cc59 auto out_quant_args = out_tensor->quant_params(); in Init() local
60 CHECK_LESS_RETURN(out_quant_args.size(), 1); in Init()
61 out_quant_arg_->scale_ = out_quant_args.front().scale; in Init()
62 out_quant_arg_->zp_ = out_quant_args.front().zeroPoint; in Init()
Dreshape_int8.cc41 auto out_quant_args = out_tensor->quant_params(); in Init() local
42 reshape_param_->quant_para_.out_args_.scale_ = out_quant_args.front().scale; in Init()
43 reshape_param_->quant_para_.out_args_.zp_ = out_quant_args.front().zeroPoint; in Init()
Dunsqueeze_int8.cc40 auto out_quant_args = input_tensor->quant_params(); in Init() local
41 param_->quant_arg.out_quant_args_.scale_ = out_quant_args.front().scale; in Init()
42 param_->quant_arg.out_quant_args_.zp_ = out_quant_args.front().zeroPoint; in Init()
Dbatch_to_space_int8.cc62 auto out_quant_args = out_tensor->quant_params(); in Init() local
63 out_quant_arg_->scale_ = out_quant_args.front().scale; in Init()
64 out_quant_arg_->zp_ = out_quant_args.front().zeroPoint; in Init()
Dleaky_relu_int8.cc54 auto out_quant_args = out_tensor->quant_params(); in Init() local
55 quant_prelu_parm_.quant_arg.out_args_.scale_ = out_quant_args.front().scale; in Init()
56 quant_prelu_parm_.quant_arg.out_args_.zp_ = out_quant_args.front().zeroPoint; in Init()
Darithmetic_int8.cc95 auto out_quant_args = out_tensor->quant_params(); in Init() local
96 CHECK_LESS_RETURN(out_quant_args.size(), 1); in Init()
97 quant_args_.out_args_.scale_ = out_quant_args.front().scale; in Init()
98 quant_args_.out_args_.zp_ = out_quant_args.front().zeroPoint; in Init()
DgatherNd_int8.cc53 auto out_quant_args = out_tensors_.at(0)->quant_params(); in ReSize() local
54 param_.alpha_ = in_quant_args.front().scale / out_quant_args.front().scale; in ReSize()
56 param_.zp_out_ = out_quant_args.front().zeroPoint; in ReSize()
Dsoftmax_int8.cc60 auto out_quant_args = out_tensor->quant_params(); in Init() local
61 quant_param_->out_quant_arg_.scale_ = out_quant_args.front().scale; in Init()
62 quant_param_->out_quant_arg_.zp_ = -out_quant_args.front().zeroPoint; in Init()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/cmsis-nn/int8/
Dsoftmax_int8_coder.cc31 std::vector<LiteQuantParam> out_quant_args = output_tensor_->quant_params(); in Prepare() local
32 MS_CHECK_TRUE(!out_quant_args.empty(), "output quant_params is empty"); in Prepare()
33 quant_params_.out_quant_arg_.scale_ = static_cast<float>(out_quant_args.at(0).scale); in Prepare()
34 quant_params_.out_quant_arg_.zp_ = out_quant_args.at(0).zeroPoint; in Prepare()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dpooling_int8_coder.cc48 std::vector<LiteQuantParam> out_quant_args = out_tensor->quant_params(); in DoCode() local
61 …::QuantArg quant_arg_out = {static_cast<float>(out_quant_args.at(0).scale), out_quant_args.at(0).z… in DoCode()
Dsoftmax_int8_coder.cc37 std::vector<LiteQuantParam> out_quant_args = output_tensor_->quant_params(); in Prepare() local
38 quant_params_.out_quant_arg_.scale_ = out_quant_args.at(0).scale; in Prepare()
39 quant_params_.out_quant_arg_.zp_ = out_quant_args.at(0).zeroPoint; in Prepare()
Dconcat_int8_coder.cc50 auto out_quant_args = output_tensor_->quant_params(); in Prepare() local
51 concat_param_->quant_arg_.out_args_.scale_ = out_quant_args.at(0).scale; in Prepare()
52 concat_param_->quant_arg_.out_args_.zp_ = out_quant_args.at(0).zeroPoint; in Prepare()
/third_party/mindspore/mindspore/lite/micro/coder/generator/component/
Dcommon_component.cc205 std::vector<LiteQuantParam> out_quant_args = out_tensor->quant_params(); in CodeGraphQuantArgsImplement() local
206 if (in_quant_args.empty() || out_quant_args.empty()) { in CodeGraphQuantArgsImplement()
212 …<< "\t\tGraphQuantArgs quan_args = { " << in_quant_args.at(0).scale << ", " << out_quant_args.at(0… in CodeGraphQuantArgsImplement()
213 << in_quant_args.at(0).zeroPoint << ", " << out_quant_args.at(0).zeroPoint << "};\n" in CodeGraphQuantArgsImplement()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/
Dnnacl_int8_serializer.cc99 ::QuantArg *out_quant_args = pooling_parameter.quant_args_[1]; in CodeStruct()
101 MS_CHECK_PTR_IF_NULL(out_quant_args); in CodeStruct()
104 code << "static QuantArg " << out_quant_name << " = " << *out_quant_args << ";\n"; in CodeStruct()