Home
last modified time | relevance | path

Searched refs:kWeightIndex (Results 1 – 25 of 78) sorted by relevance

1234

/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dmatmul_int8_coder.cc23 filter_tensor_ = input_tensors_.at(kWeightIndex); in Prepare()
47 param_->col_ = o_shape[o_shape.size() - kWeightIndex]; in ReSize()
48 …ram_->a_transpose_ ? x_shape[x_shape.size() - kBiasIndex] : x_shape[x_shape.size() - kWeightIndex]; in ReSize()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dconvolution_base.cc129 auto weight_tensor = in_tensors_.at(kWeightIndex); in InitConvWeightBias()
158 …origin_weight_ = origin_weight_ != nullptr ? origin_weight_ : in_tensors_.at(kWeightIndex)->Mutabl… in RepackWeight()
177 auto filter_tensor = in_tensors_.at(kWeightIndex); in CheckResizeValid()
189 auto filter_tensor = in_tensors_.at(kWeightIndex); in SetIfPerChannel()
226 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocQuantParam()
269 auto weight_tensor = in_tensors_.at(kWeightIndex); in SetFilterTensorQuantParam()
420 if (in_tensors_.at(kWeightIndex)->data() != nullptr) { in UpdateOriginWeightAndBias()
421 origin_weight_ = in_tensors_.at(kWeightIndex)->data(); in UpdateOriginWeightAndBias()
Dgroup_convolution_creator.cc150 …std::vector<std::pair<int, std::vector<int>>> const_tensor_list{std::make_pair(kWeightIndex, filte… in NewConstTensor()
165 int new_in_channel = origin_inputs_.at(kWeightIndex)->Channel(); in SetShapeOfTensors()
171 new_out_channel = origin_inputs_.at(kWeightIndex)->Batch() / conv_param_->group_; in SetShapeOfTensors()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dconvolution_depthwise_fp16.cc27 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
35 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
63 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
Dconvolution_depthwise_3x3_fp16.cc30 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
39 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
69 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
Dconvolution_fp16.cc31 auto filter_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
42 auto filter_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
100 auto filter_tensor = in_tensors_.at(kWeightIndex); in Init()
Dconvolution_depthwise_slidewindow_fp16.cc58 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
66 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
99 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
Ddeconvolution_fp16.cc57 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
69 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
187 CHECK_NULL_RETURN(in_tensors_.at(kWeightIndex)); in Init()
191 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
Ddeconvolution_depthwise_fp16.cc72 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
95 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
109 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dfullconnection.cc66 if (!in_tensors_.at(kWeightIndex)->IsConst()) { in CheckSpecs()
72 if (in_tensors_.at(kWeightIndex)->shape().size() != DIMENSION_2D) { in CheckSpecs()
76 if (intensor_shape.C != in_tensors_.at(kWeightIndex)->shape()[1]) { in CheckSpecs()
155 …void *src_data = stored_weight_ == nullptr ? in_tensors_.at(kWeightIndex)->data() : stored_weight_; in InitFilter()
159 bool isModelFp16 = in_tensors_.at(kWeightIndex)->data_type() == kNumberTypeFloat16; in InitFilter()
303 stored_weight_ = StoreTensorData(in_tensors_.at(kWeightIndex)); in StoreConstData()
Ddepthwise_conv2d.cc48 if (!in_tensors_.at(kWeightIndex)->IsConst()) { in CheckSpecs()
113 …void *origin_weight = stored_weight_ == nullptr ? in_tensors_.at(kWeightIndex)->data() : stored_we… in InitWeights()
153 auto src_type = in_tensors_.at(kWeightIndex)->data_type(); in InitWeights()
179 auto src_type = in_tensors_.at(kWeightIndex)->data_type(); in InitBias()
312 stored_weight_ = StoreTensorData(in_tensors_.at(kWeightIndex)); in StoreConstData()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_depthwise_fp32.cc29 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
101 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
109 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
Dconvolution_depthwise_3x3_fp32.cc31 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
116 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
124 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
Ddeconvolution_fp32.cc57 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
82 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
169 CHECK_NULL_RETURN(in_tensors_.at(kWeightIndex)); in Init()
177 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
191 if (in_tensors_.at(kWeightIndex)->data() != nullptr) { in Init()
Dconvolution_depthwise_slidewindow_x86_fp32.cc67 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
168 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
178 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
Dconvolution_depthwise_slidewindow_fp32.cc63 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
164 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
172 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
Dconvolution_depthwise_indirect_fp32.cc40 auto weight_tensor = in_tensors_[kWeightIndex]; in Init()
177 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
190 auto weight_tensor = in_tensors_[kWeightIndex]; in MallocWeightBiasData()
Dconvolution_fp32.cc81 auto filter_tensor = in_tensors_.at(kWeightIndex); in Init()
171 auto filter_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
202 auto filter_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
Dconvolution_winograd_fp32.cc107 auto filter_tensor = in_tensors_.at(kWeightIndex); in Init()
189 auto filter_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
246 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
Dconvolution_slidewindow_fp32.cc40 auto filter_tensor = in_tensors_.at(kWeightIndex); in Init()
189 auto filter_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
202 auto filter_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
Ddeconvolution_depthwise_fp32.cc81 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init()
190 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData()
212 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight()
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/quant_helper/
Dconv_quant_type_determiner.cc27 …MS_CHECK_TRUE_MSG(graph.allTensors.size() > node->inputIndex.at(kWeightIndex), false, "Out of vect… in DetermineQuantWeight()
28 auto &weight_tensor = graph.allTensors.at(node->inputIndex.at(kWeightIndex)); in DetermineQuantWeight()
Dmatmul_quant_type_determiner.cc27 …MS_CHECK_TRUE_MSG(graph.allTensors.size() > node->inputIndex.at(kWeightIndex), false, "Out of vect… in DetermineQuantWeight()
28 auto &input_tensor2 = graph.allTensors.at(node->inputIndex.at(kWeightIndex)); in DetermineQuantWeight()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dsplice_fp32_coder.cc35 int src_row = src_shape.at(kWeightIndex); in DoCode()
36 int dst_row = dst_shape.at(kWeightIndex); in DoCode()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Ddepthwise_conv2d_infer.c46 param->kernel_h_ = param->kernel_h_ != -1 ? param->kernel_h_ : GetHeight(inputs[kWeightIndex]); in DepthwiseConv2dInferShape()
47 param->kernel_w_ = param->kernel_w_ != -1 ? param->kernel_w_ : GetWidth(inputs[kWeightIndex]); in DepthwiseConv2dInferShape()

1234