Searched refs:dst_weight (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_transform_fusion.cc | 54 void GenerateNewWeightConv2D(float *dst_weight, const float *conv_weight, const float *scale_weight, in GenerateNewWeightConv2D() argument
56 MS_ASSERT(dst_weight != nullptr && conv_weight != nullptr && scale_weight != nullptr); in GenerateNewWeightConv2D()
63 dst_weight[i * kernel_size + j] = conv_weight[i * kernel_size + j] * scale_weight[i]; in GenerateNewWeightConv2D()
68 void GenerateNewWeightConv2DTranspose(float *dst_weight, const float *scale_weight, in GenerateNewWeightConv2DTranspose() argument
70 MS_ASSERT(dst_weight != nullptr && scale_weight != nullptr && weight_tensor != nullptr); in GenerateNewWeightConv2DTranspose()
81 dst_weight[i + j * kernel_num + k * area_size * kernel_num] = in GenerateNewWeightConv2DTranspose()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | conv2d_3x3_int8_coder.cc | 27 void ProcessFilterUint8(int8_t *origin_weight, int16_t *dst_weight, ConvParameter *conv_param) { in ProcessFilterUint8() argument
43 Conv3x3Int8FilterTransform(tmp_addr, dst_weight, iC8, output_channel, kernel_plane); in ProcessFilterUint8()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | convolution_3x3_int8.h | 49 int ProcessFilterUint8(const int8_t *origin_weight, int16_t *dst_weight, const ConvParameter *conv_…
|
| D | convolution_3x3_int8.cc | 28 int ProcessFilterUint8(const int8_t *origin_weight, int16_t *dst_weight, const ConvParameter *conv_… in ProcessFilterUint8() argument
43 Conv3x3Int8FilterTransform(tmp_addr, dst_weight, iC8, output_channel, kernel_plane); in ProcessFilterUint8()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | deconv_winograd_fp32.c | 143 float *dst_weight = (float *)unit->weight_; in PackDeConvWgDataFp32() local
145 memset(dst_weight, 0, size * sizeof(float)); in PackDeConvWgDataFp32()
153 dst_weight[dst_index] = current_unit_weight[src_index]; in PackDeConvWgDataFp32()
157 TransposeWeight(dst_weight, size / 16); in PackDeConvWgDataFp32()
|
| /third_party/ffmpeg/libavcodec/mips/ |
| D | h264dsp_msa.c | 232 int32_t dst_weight, int32_t offset_in) in avc_biwgt_4x2_msa() argument
240 offset_in += (128 * (src_weight + dst_weight)); in avc_biwgt_4x2_msa()
243 dst_wgt = __msa_fill_b(dst_weight); in avc_biwgt_4x2_msa()
265 int32_t dst_weight, int32_t offset_in) in avc_biwgt_4x4_msa() argument
273 offset_in += (128 * (src_weight + dst_weight)); in avc_biwgt_4x4_msa()
276 dst_wgt = __msa_fill_b(dst_weight); in avc_biwgt_4x4_msa()
299 int32_t dst_weight, int32_t offset_in) in avc_biwgt_4x8_msa() argument
307 offset_in += (128 * (src_weight + dst_weight)); in avc_biwgt_4x8_msa()
310 dst_wgt = __msa_fill_b(dst_weight); in avc_biwgt_4x8_msa()
339 int32_t dst_weight, int32_t offset_in) in avc_biwgt_8x4_msa() argument
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | deconv_winograd_fp16.c | 324 float16_t *dst_weight = (float16_t *)unit->weight_; in PackDeConvWgDataFp16() local
326 memset(dst_weight, 0, size * sizeof(float16_t)); in PackDeConvWgDataFp16()
334 dst_weight[dst_index] = current_unit_weight[src_index]; in PackDeConvWgDataFp16()
|