Home
last modified time | relevance | path

Searched refs:in_quant_args (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Ddepth_to_space_int8.cc52 auto in_quant_args = input_tensor->quant_params(); in Init() local
53 CHECK_LESS_RETURN(in_quant_args.size(), 1); in Init()
54 in_quant_arg_->scale_ = in_quant_args.front().scale; in Init()
55 in_quant_arg_->zp_ = in_quant_args.front().zeroPoint; in Init()
Dslice_int8.cc39 auto in_quant_args = input->quant_params(); in Init() local
40 CHECK_LESS_RETURN(in_quant_args.size(), 1); in Init()
41 param_->quant_arg_.in_args_.scale_ = in_quant_args.front().scale; in Init()
42 param_->quant_arg_.in_args_.zp_ = in_quant_args.front().zeroPoint; in Init()
Dsqueeze_int8.cc56 auto in_quant_args = in_tensors_.front()->quant_params(); in Init() local
57 MS_ASSERT(in_quant_args.size() > 0); in Init()
58 quant_squeeze_param_->in_quant_args_->scale_ = in_quant_args.front().scale; in Init()
59 quant_squeeze_param_->in_quant_args_->zp_ = in_quant_args.front().zeroPoint; in Init()
Dgather_int8.cc36 auto in_quant_args = in_tensors_.at(0)->quant_params(); in Init() local
37 CHECK_LESS_RETURN(in_quant_args.size(), 1); in Init()
40 param_.alpha_ = in_quant_args.front().scale / out_quant_args.front().scale; in Init()
41 param_.zp_in_ = in_quant_args.front().zeroPoint; in Init()
Dsplit_int8.cc44 auto in_quant_args = in_tensor->quant_params(); in Init() local
45 CHECK_LESS_RETURN(in_quant_args.size(), 1); in Init()
46 param->quant_arg_.in_args_.scale_ = in_quant_args.front().scale; in Init()
47 param->quant_arg_.in_args_.zp_ = in_quant_args.front().zeroPoint; in Init()
Dpower_int8.cc38 auto in_quant_args = input->quant_params(); in Init() local
39 CHECK_LESS_RETURN(in_quant_args.size(), 1); in Init()
40 param_->quant_arg_.in_args_.scale_ = in_quant_args.front().scale; in Init()
41 param_->quant_arg_.in_args_.zp_ = in_quant_args.front().zeroPoint; in Init()
Dcrop_int8.cc34 auto in_quant_args = input_tensor->quant_params(); in Init() local
35 crop_para_->quant_arg.in_args_.scale_ = in_quant_args.front().scale; in Init()
36 crop_para_->quant_arg.in_args_.zp_ = in_quant_args.front().zeroPoint; in Init()
Darithmetic_self_int8.cc34 auto in_quant_args = input_tensor->quant_params(); in Init() local
35 CHECK_LESS_RETURN(in_quant_args.size(), 1); in Init()
36 para_->quant_arg_.in_args_.scale_ = in_quant_args.front().scale; in Init()
37 para_->quant_arg_.in_args_.zp_ = in_quant_args.front().zeroPoint * (-1); in Init()
Dargminmax_int8.cc53 auto in_quant_args = input_tensor->quant_params(); in Init() local
54 CHECK_LESS_RETURN(in_quant_args.size(), 1); in Init()
55 in_quant_arg_->scale_ = in_quant_args.front().scale; in Init()
56 in_quant_arg_->zp_ = in_quant_args.front().zeroPoint; in Init()
Dreshape_int8.cc36 auto in_quant_args = input_tensor->quant_params(); in Init() local
37 reshape_param_->quant_para_.in_args_.scale_ = in_quant_args.front().scale; in Init()
38 reshape_param_->quant_para_.in_args_.zp_ = in_quant_args.front().zeroPoint; in Init()
Dconcat_int8.cc50 auto in_quant_args = input_tensor->quant_params(); in Init() local
51 MS_CHECK_TRUE_RET(!in_quant_args.empty(), RET_ERROR); in Init()
52 concat_param_->quant_arg_.in_args_[i].scale_ = in_quant_args.front().scale; in Init()
53 concat_param_->quant_arg_.in_args_[i].zp_ = in_quant_args.front().zeroPoint; in Init()
Dbatch_to_space_int8.cc51 auto in_quant_args = input_tensor->quant_params(); in Init() local
52 in_quant_arg_->scale_ = in_quant_args.front().scale; in Init()
53 in_quant_arg_->zp_ = in_quant_args.front().zeroPoint; in Init()
Dleaky_relu_int8.cc49 auto in_quant_args = input_tensor->quant_params(); in Init() local
50 quant_prelu_parm_.quant_arg.in_args_.scale_ = in_quant_args.front().scale; in Init()
51 quant_prelu_parm_.quant_arg.in_args_.zp_ = in_quant_args.front().zeroPoint; in Init()
DgatherNd_int8.cc52 auto in_quant_args = in_tensors_.at(0)->quant_params(); in ReSize() local
54 param_.alpha_ = in_quant_args.front().scale / out_quant_args.front().scale; in ReSize()
55 param_.zp_in_ = in_quant_args.front().zeroPoint; in ReSize()
Dsoftmax_int8.cc53 auto in_quant_args = input_tensor->quant_params(); in Init() local
54 quant_param_->in_quant_args_.scale_ = in_quant_args.front().scale; in Init()
55 quant_param_->in_quant_args_.zp_ = -in_quant_args.front().zeroPoint; in Init()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dpooling_int8_coder.cc47 std::vector<LiteQuantParam> in_quant_args = in_tensor->quant_params(); in DoCode() local
60 …::QuantArg quant_arg_in = {static_cast<float>(in_quant_args.at(0).scale), in_quant_args.at(0).zero… in DoCode()
Dsoftmax_int8_coder.cc33 std::vector<LiteQuantParam> in_quant_args = input_tensor_->quant_params(); in Prepare() local
34 quant_params_.in_quant_args_.scale_ = in_quant_args.at(0).scale; in Prepare()
35 quant_params_.in_quant_args_.zp_ = -in_quant_args.at(0).zeroPoint; in Prepare()
Dconcat_int8_coder.cc45 auto in_quant_args = input_tensor->quant_params(); in Prepare() local
46 concat_param_->quant_arg_.in_args_[i].scale_ = in_quant_args.at(0).scale; in Prepare()
47 concat_param_->quant_arg_.in_args_[i].zp_ = in_quant_args.at(0).zeroPoint; in Prepare()
/third_party/mindspore/mindspore/lite/micro/coder/generator/component/
Dcommon_component.cc204 std::vector<LiteQuantParam> in_quant_args = in_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.cc98 ::QuantArg *in_quant_args = pooling_parameter.quant_args_[0]; in CodeStruct()
100 MS_CHECK_PTR_IF_NULL(in_quant_args); in CodeStruct()
103 code << "static QuantArg " << in_quant_name << " = " << *in_quant_args << ";\n"; in CodeStruct()