Home
last modified time | relevance | path

Searched defs:weight_tensor (Results 1 – 25 of 47) sorted by relevance

12

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_depthwise_fp32.cc29 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
101 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
109 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
Ddeconvolution_depthwise_fp32.cc81 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
190 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
212 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
Dconvolution_depthwise_slidewindow_fp32.cc63 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
164 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
172 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
Dconvolution_depthwise_slidewindow_x86_fp32.cc67 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
168 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
178 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
Dconvolution_depthwise_3x3_fp32.cc31 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
116 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
124 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
Dconvolution_depthwise_indirect_fp32.cc40 auto weight_tensor = in_tensors_[kWeightIndex]; in Init() local
177 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
190 auto weight_tensor = in_tensors_[kWeightIndex]; in MallocWeightBiasData() local
Ddeconvolution_fp32.cc57 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
82 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
177 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dconvolution_depthwise_fp16.cc27 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
35 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
63 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
Dconvolution_depthwise_slidewindow_fp16.cc58 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
66 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
99 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
Ddeconvolution_depthwise_fp16.cc72 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
95 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
109 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
Dconvolution_depthwise_3x3_fp16.cc30 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
39 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
69 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
Dconvolution_winograd_fp16.cc36 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
82 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
151 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
Dconvolution_1x1_fp16.cc84 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
115 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
151 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
Ddeconvolution_fp16.cc57 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
69 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
191 auto weight_tensor = in_tensors_.at(kWeightIndex); in Init() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dconvolution_base.cc129 auto weight_tensor = in_tensors_.at(kWeightIndex); in InitConvWeightBias() local
226 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocQuantParam() local
269 auto weight_tensor = in_tensors_.at(kWeightIndex); in SetFilterTensorQuantParam() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_conv_parser.cc60 const auto &weight_tensor = tflite_subgraph->tensors.at(tflite_op->inputs[1]); in Parse() local
117 const auto &weight_tensor = tflite_subgraph->tensors.at(tflite_op->inputs.at(1)); in Parse() local
Dtflite_conv_transpose_parser.cc55 const auto &weight_tensor = tflite_subgraph->tensors.at(tflite_op->inputs.at(1)); in Parse() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dprelu.cc37 auto weight_tensor = in_tensors_.at(1); in InitWeights() local
94 auto weight_tensor = in_tensors_.at(1); in CheckSpecs() local
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/quant_helper/
Dconv_quant_type_determiner.cc28 auto &weight_tensor = graph.allTensors.at(node->inputIndex.at(kWeightIndex)); in DetermineQuantWeight() local
Dconv_quant_param_propogator.cc33 auto &weight_tensor = graph->allTensors.at(node.inputIndex.at(1)); in PropogateQuantParams() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dconv1d_inout_adjust.cc68 auto weight_tensor = opt::GetTensorInfo(weight_node); in ExpandFilterShape() local
143 auto weight_tensor = opt::GetTensorInfo(weight_node); in Run() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_transform_fusion.cc69 … const tensor::TensorPtr &weight_tensor, int64_t group, int kernel_num) { in GenerateNewWeightConv2DTranspose()
202 auto weight_tensor = std::dynamic_pointer_cast<tensor::Tensor>(conv_weight_param); in GenNewConvTensor() local
267 …nsformFusion::CalNewWeightTensor(const CNodePtr &conv_node, const tensor::TensorPtr &weight_tensor, in CalNewWeightTensor()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconvolution_depthwise_slidewindow_int8.cc41 auto weight_tensor = in_tensors_.at(kWeightIndex); in InitWeightBias() local
222 auto weight_tensor = in_tensors_.at(kWeightIndex); in ReinitQuantParam() local
Dmatmul_base_int8.cc97 auto weight_tensor = in_tensors_.at(1); in MallocQuantParam() local
143 auto weight_tensor = in_tensors_.at(1); in InitQuantParam() local
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dconvolution_base_npu.cc90 auto weight_tensor = std::make_shared<ge::Tensor>(); in InitWeightConst() local

12