Home
last modified time | relevance | path

Searched defs:filter_tensor (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/bolt/
Dconvolution_bolt.cc60 std::vector<Tensor> filter_tensor(filter_desc.size()); in InitWeightBiasDesc() local
158 auto filter_tensor = weight_tensors_[0]; in InferForwardAlgorithm() local
210 auto filter_tensor = weight_tensors_[0]; in InferFilterTransformBytes() local
247 auto filter_tensor = weight_tensors_[0]; in TransformFilter() local
324 Tensor filter_tensor = this->weight_tensors_[0]; in MallocTmpTensor() local
416 auto filter_tensor = weight_tensors_[0]; in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Dconvolution_fp16.cc31 auto filter_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
42 auto filter_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
105 auto filter_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dconvolution_im2col_base_fp32.cc82 auto filter_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
189 auto filter_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
214 auto filter_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
Dconvolution_slidewindow_fp32.cc43 auto filter_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
195 auto filter_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
208 auto filter_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
Dconvolution_fp32.cc96 auto filter_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
237 auto filter_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
268 auto filter_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
Dconvolution_1x1_fp32.cc131 auto filter_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
314 auto filter_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
341 auto filter_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
Dconvolution_winograd_base_fp32.cc128 auto filter_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
245 auto filter_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
Dadder_fp32.cc69 auto filter_tensor = in_tensors_.at(kWeightIndex); in InitWeightBias() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dconvolution_slidewindow.c83 TensorC *filter_tensor = conv->base_.in_[SECOND_INPUT]; in ConvSWPackWeight() local
103 TensorC *filter_tensor = conv->base_.in_[SECOND_INPUT]; in ConvSWMallocWeightBiasData() local
205 TensorC *filter_tensor = self->in_[SECOND_INPUT]; in ConvolutionSWPrepare() local
Dconvolution_base.c157 TensorC *filter_tensor = conv->base_.in_[SECOND_INPUT]; in ConvBaseCheckResizeValid() local
Ddeconvolution.c162 TensorC *filter_tensor = conv->base_.in_[SECOND_INPUT]; in DeConvCheckvResizeValid() local
Dconvolution_winograd_base.c126 TensorC *filter_tensor = self->in_[SECOND_INPUT]; in ConvolutionWinogradBasePrepare() local
Dconvolution_1x1.c114 TensorC *filter_tensor = conv->base_.in_[SECOND_INPUT]; in Conv1x1PackWeight() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/
Dconvolution_base.cc222 auto filter_tensor = in_tensors_.at(kWeightIndex); in CheckResizeValid() local
235 auto filter_tensor = in_tensors_.at(kWeightIndex); in CheckDeconvResizeValid() local
251 auto filter_tensor = in_tensors_.at(kWeightIndex); in SetIfPerChannel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Dpower_fp32_coder.cc40 Tensor *filter_tensor = input_tensors_.at(kWeightIndex); in DoCode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Dconvolution_int8.cc54 auto filter_tensor = in_tensors_.at(kWeightIndex); in InitWeightBias() local
Dconvolution_1x1_int8.cc220 auto filter_tensor = in_tensors_.at(kWeightIndex); in InitWeightBias() local
270 auto filter_tensor = in_tensors_.at(kWeightIndex); in InitWeightBiasArm32() local
Dconvolution_3x3_int8.cc83 auto filter_tensor = in_tensors_.at(kWeightIndex); in InitWeightBias() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dconv2d.cc108 auto *filter_tensor = in_tensors_.at(kWeightIndex); in IsFilterConst() local
125 auto *filter_tensor = in_tensors_.at(kWeightIndex); in FilterBiasCheckSpecs() local
442 auto filter_tensor = in_tensors_.at(1); in InitFilter() local
Dwinograd.cc155 auto filter_tensor = in_tensors_.at(1); in InitFilter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dspace_to_depth_split.cc78 tensor::TensorPtr filter_tensor = CreateTensor(node); in CreateValueNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32-sparsity/
Dmatmul_fp32_tests.cc51 auto *filter_tensor = in GetInputs() local