Home
last modified time | relevance | path

Searched refs:scale_param_ (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dscale_fp32.cc31 if (scale_param_->const_scale_) { in ~ScaleCPUKernel()
37 if (scale_param_->const_offset_) { in ~ScaleCPUKernel()
48 scale_param_->const_scale_ = true; in InitScaleOffset()
56 scale_param_->const_scale_ = false; in InitScaleOffset()
61 scale_param_->const_offset_ = true; in InitScaleOffset()
69 scale_param_->const_offset_ = true; in InitScaleOffset()
79 scale_param_->const_offset_ = false; in InitScaleOffset()
92 if (scale_param_->axis_ < 0) { in CalculateParameter()
93 scale_param_->axis_ = scale_param_->axis_ + in_shape.size(); in CalculateParameter()
95 if (scale_shape.size() + scale_param_->axis_ > in_shape.size()) { in CalculateParameter()
[all …]
Dscale_fp32.h31 scale_param_ = reinterpret_cast<ScaleParameter *>(op_parameter_); in ScaleCPUKernel()
43 ScaleParameter *scale_param_;
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dscale_fp32_coder.cc30 scale_param_->const_scale_ = true; in InitScaleOffset()
38 scale_param_->const_scale_ = false; in InitScaleOffset()
43 scale_param_->const_offset_ = true; in InitScaleOffset()
49 scale_param_->const_offset_ = true; in InitScaleOffset()
59 scale_param_->const_offset_ = false; in InitScaleOffset()
71 if (scale_param_->axis_ < 0) { in CalculateParameter()
72 scale_param_->axis_ = scale_param_->axis_ + in_shape.size(); in CalculateParameter()
74 if (scale_shape.size() + scale_param_->axis_ > in_shape.size()) { in CalculateParameter()
78 scale_param_->outer_size_ = 1; in CalculateParameter()
79 scale_param_->axis_size_ = 1; in CalculateParameter()
[all …]
Dscale_fp32_coder.h42 ScaleParameter *scale_param_{nullptr};
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dscale_int8.cc50 scale_param_->const_scale_ = false; in InitScaleOffset()
54 scale_param_->const_scale_ = true; in InitScaleOffset()
70 scale_param_->const_offset_ = false; in InitScaleOffset()
77 scale_param_->const_offset_ = true; in InitScaleOffset()
107 if (scale_param_->axis_ < 0) { in InitParameter()
108 scale_param_->axis_ += in_shape.size(); in InitParameter()
110 if (scale_shape.size() + scale_param_->axis_ > in_shape.size()) { in InitParameter()
116 if (in_shape[i + scale_param_->axis_] != scale_shape[i]) { in InitParameter()
133 size_t len = input0_size - scale_param_->axis_; in InitParameter()
181 scale_param_->input_zp_ = input->quant_params().front().zeroPoint; in InitQuantArgs()
[all …]
Dscale_int8.h34 scale_param_ = reinterpret_cast<ScaleParameter *>(op_parameter_); in ScaleInt8CPUKernel()
51 ScaleParameter *scale_param_ = nullptr; variable
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dscale_fp16.cc75 switch (scale_param_->activation_type_) { in Scale()
77 DoScaleRelu6Fp16(input_, output_, scale_, offset_, task_id, scale_param_); in Scale()
80 Fp16DoScaleRelu(input_, output_, scale_, offset_, task_id, scale_param_); in Scale()
83 DoScaleFp16(input_, output_, scale_, offset_, task_id, scale_param_); in Scale()
86 … MS_LOG(ERROR) << "ScaleFp16 does not support activation type " << scale_param_->activation_type_; in Scale()