| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | deconvolution_depthwise_fp32.cc | 96 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
129 auto weight_tensor = in_tensors_.at(kWeightIndex); in ReSize() local
219 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
244 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
|
| D | convolution_depthwise_fp32.cc | 32 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
183 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
191 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
|
| D | convolution_depthwise_slidewindow_x86_fp32.cc | 76 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
182 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
192 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
|
| D | convolution_depthwise_3x3_fp32.cc | 33 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
128 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
136 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
|
| D | convolution_depthwise_slidewindow_fp32.cc | 75 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
181 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
189 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
|
| D | convolution_depthwise_indirect_fp32.cc | 42 auto weight_tensor = in_tensors_[kWeightIndex]; in Prepare() local
195 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
208 auto weight_tensor = in_tensors_[kWeightIndex]; in MallocWeightBiasData() local
|
| D | deconvolution_fp32.cc | 49 auto weight_tensor = in_tensors_.at(kWeightIndex); in ReSize() local
69 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
95 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
203 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
|
| D | deconvolution_winograd_fp32.cc | 222 auto weight_tensor = in_tensors_[1]; in InitComputeParam() local
316 auto weight_tensor = in_tensors_.at(kWeightIndex); in InitDataParam() local
370 auto weight_tensor = in_tensors_.at(kWeightIndex); in ReSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/ |
| D | convolution_depthwise_fp16.cc | 27 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
35 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
66 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
|
| D | convolution_depthwise_slidewindow_fp16.cc | 58 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
66 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
98 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
|
| D | deconvolution_depthwise_fp16.cc | 72 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
98 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
112 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
|
| D | convolution_depthwise_3x3_fp16.cc | 30 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 Prepare() local
|
| D | convolution_winograd_fp16.cc | 36 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
85 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
176 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
|
| D | convolution_1x1_fp16.cc | 84 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
117 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
152 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
|
| D | deconvolution_fp16.cc | 57 auto weight_tensor = in_tensors_.at(kWeightIndex); in PackWeight() local
69 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocWeightBiasData() local
237 auto weight_tensor = in_tensors_.at(kWeightIndex); in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | convolution_base.cc | 159 auto weight_tensor = in_tensors_.at(kWeightIndex); in InitConvWeightBias() local
298 auto weight_tensor = in_tensors_.at(kWeightIndex); in MallocQuantParam() local
347 auto weight_tensor = in_tensors_.at(kWeightIndex); in SetFilterTensorQuantParam() local
515 auto weight_tensor = in_tensors_.at(kWeightIndex); in CheckInputsValid() local
529 auto weight_tensor = in_tensors_.at(kWeightIndex); in CheckAndGetWeightParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | weight_quant_batch_matmul.cc | 31 … const BaseTensorPtr &weight_tensor, const BaseTensorPtr &antiquant_scale_tensor, in WeightQuantBatchMatmulV2AscendCall()
43 ValueTuplePtr GetTransposePerm(const BaseTensorPtr &weight_tensor) { in GetTransposePerm()
61 …t std::shared_ptr<OpRunner> &op, const BaseTensorPtr &x_tensor, const BaseTensorPtr &weight_tensor, in WeightQuantBatchMatmulV2AscendCustomize()
|
| D | dense.cc | 30 ValueTuplePtr GetTransposePerm(const BaseTensorPtr &weight_tensor) { in GetTransposePerm()
49 … const BaseTensorPtr &weight_tensor, const std::optional<BaseTensorPtr> &bias_tensor) { in DenseAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/ |
| D | dense.cc | 29 ValueTuplePtr GetTransposePerm(const BaseTensorPtr &weight_tensor) { in GetTransposePerm()
48 … const BaseTensorPtr &weight_tensor, const std::optional<BaseTensorPtr> &bias_tensor) { in DenseCPUCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/pyboost/customize/ |
| D | dense.cc | 28 ValueTuplePtr GetTransposePerm(const BaseTensorPtr &weight_tensor) { in GetTransposePerm()
47 … const BaseTensorPtr &weight_tensor, const std::optional<BaseTensorPtr> &bias_tensor) { in DenseGPUCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | conv2d_infer.c | 75 int CheckConvAttr(const int input_c, const TensorC *weight_tensor, const ConvParameter *param) { in CheckConvAttr()
122 const TensorC *weight_tensor = inputs[1]; in Conv2dInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/ |
| D | prelu.cc | 38 auto weight_tensor = in_tensors_.at(1); in InitWeights() local
125 auto weight_tensor = in_tensors_.at(1); in CheckSpecs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | binary_cross_entropy_with_logits.cc | 105 const auto &weight_tensor = input_values[kIndex2]->cast<tensor::BaseTensorPtr>(); in InferShape() local
145 const auto &weight_tensor = input_values[kInputIndex2]->cast<tensor::BaseTensorPtr>(); in InferType() local
|
| D | batch_norm_ext.cc | 122 const auto &weight_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferShape() local
130 const auto &weight_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | convolution_depthwise.c | 78 TensorC *weight_tensor = conv->base_.in_[SECOND_INPUT]; in ConvDwMallocWeightBiasData() local
147 TensorC *weight_tensor = self->in_[SECOND_INPUT]; in ConvolutionDepthwisePrepare() local
|