| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | resize_bilinear_gpu_kernel.h | 45 float w_scale = Scaling(input_w_, output_w_, align_corners_); in Launch()
46 …CalResizeBilinear(input, n_, c_, input_h_, input_w_, output_h_, output_w_, h_scale, w_scale, outpu… in Launch()
80 output_w_ = SizeToInt(output_shape[3]); in Init()
102 output_w_ = 0; in ResetResource()
130 int output_w_; variable
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | conv1x1_fp32_tests.cc | 51 conv_param->output_w_ = 5; in TEST_F()
92 conv_param->output_w_ = 4; in TEST_F()
106 conv_param->output_h_ = conv_param->output_w_ = 3; in TEST_F()
125 conv_param->output_h_ = conv_param->output_w_ = 3; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | resize_bilinear_impl.cuh | 22 const int output_h_, const int output_w_, const float h_scale, const float w_scale, T *output,
25 …const int output_h_, const int output_w_, const float h_scale, const float w_scale, half *output, …
28 …const int output_h_, const int output_w_, const float h_scale, const float w_scale, float *output,…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | deconv_fp16.c | 24 size_t output_plane = conv_param->output_w_ * conv_param->output_h_; in DeConvPostFp16()
31 int dst_oh_stride = conv_param->output_w_ * C8NUM; in DeConvPostFp16()
33 int dst_kh_stride = conv_param->dilation_h_ * conv_param->output_w_ * C8NUM; in DeConvPostFp16()
49 …int kw_end = MSMIN(conv_param->kernel_w_, UP_DIV(conv_param->output_w_ - ow, conv_param->dilation_… in DeConvPostFp16()
|
| D | conv_depthwise_fp16.c | 351 int units = UP_DIV(conv_param->output_w_, C2NUM); in ConvDw3x3Fp16()
360 …float16_t *dst = output_data + b * conv_param->output_h_ * conv_param->output_w_ * conv_param->out… in ConvDw3x3Fp16()
368 ConvDw3x3InitTopFp16(src, lines, conv_param->output_w_, conv_param->input_channel_); in ConvDw3x3Fp16()
371 … ConvDw3x3InitRowFp16(src + oh * line, lines, conv_param->output_w_, conv_param->input_channel_); in ConvDw3x3Fp16()
374 …ConvDw3x3LineFp16(dst + oh * line, lines, weight_data, bias_data, conv_param->output_w_, conv_para… in ConvDw3x3Fp16()
378 …ConvDw3x3RowFp16(src + oh * line + line, lines, conv_param->output_w_, conv_param->input_channel_); in ConvDw3x3Fp16()
380 ConvDw3x3LineFp16(dst + oh * line, lines, weight_data, bias_data, conv_param->output_w_, in ConvDw3x3Fp16()
385 ConvDw3x3BottomFp16(lines, conv_param->output_w_, conv_param->input_channel_); in ConvDw3x3Fp16()
388 …ConvDw3x3RowFp16(src + oh * line + line, lines, conv_param->output_w_, conv_param->input_channel_); in ConvDw3x3Fp16()
391 …ConvDw3x3LineFp16(dst + oh * line, lines, weight_data, bias_data, conv_param->output_w_, conv_para… in ConvDw3x3Fp16()
[all …]
|
| D | conv_fp16.c | 33 int output_hw = conv_param->output_h_ * conv_param->output_w_; in ConvFp16()
75 int output_hw = conv_param->output_h_ * conv_param->output_w_; in ConvOutNc8hw8Fp16()
199 int out_w_block = UP_DIV(conv_param->output_w_, conv_param->output_unit_); in ConvWinogardFp16()
221 …out_batch_offset = b * conv_param->output_channel_ * conv_param->output_h_ * conv_param->output_w_; in ConvWinogardFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | deconv_fp32.c | 42 size_t output_plane = conv_param->output_w_ * conv_param->output_h_; in DeConvPostFp32C8()
54 int dst_oh_stride = conv_param->output_w_ * C8NUM; in DeConvPostFp32C8()
56 int dst_kh_stride = conv_param->dilation_h_ * conv_param->output_w_ * C8NUM; in DeConvPostFp32C8()
74 …int kw_end = MSMIN(conv_param->kernel_w_, UP_DIV(conv_param->output_w_ - ow, conv_param->dilation_… in DeConvPostFp32C8()
|
| D | conv_depthwise_fp32.c | 47 …float *dst = output_data + b * conv_param->output_h_ * conv_param->output_w_ * conv_param->output_… in ConvDw()
49 float *dst_data = dst + oh * conv_param->output_w_ * conv_param->output_channel_; in ConvDw()
55 for (int ow = 0; ow < conv_param->output_w_; ow++) { in ConvDw()
69 int out_w_end = MSMIN(conv_param->output_w_, (conv_param->input_w_ + conv_param->pad_l_ - in ConvDw()
84 Fp32Relu(dst_data, conv_param->output_w_ * conv_param->output_channel_, dst_data); in ConvDw()
87 Fp32Relu6(dst_data, conv_param->output_w_ * conv_param->output_channel_, dst_data); in ConvDw()
99 int right = conv_param->output_w_; in InitSlidingParam()
125 sliding->out_step_ = conv_param->output_h_ * conv_param->output_w_ * sliding->block_channel_; in InitSlidingParam()
128 sliding->out_h_step_ = conv_param->output_w_ * block; in InitSlidingParam()
129 sliding->out_c_step_ = block * conv_param->output_h_ * conv_param->output_w_; in InitSlidingParam()
[all …]
|
| D | conv_winograd_fp32.c | 31 int out_w_block = UP_DIV(conv_param->output_w_, conv_param->output_unit_); in ConvWinogardFp32()
60 …int out_batch_offset = b * conv_param->output_channel_ * conv_param->output_w_ * conv_param->outpu… in ConvWinogardFp32()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | convolution_grad_input.c | 27 int out_w = conv_param->output_w_; in ConvDwInputGrad16()
31 int out_spatial = conv_param->output_h_ * conv_param->output_w_; in ConvDwInputGrad16()
132 int out_w = conv_param->output_w_; in ConvDwInputGrad8()
136 int out_spatial = conv_param->output_h_ * conv_param->output_w_; in ConvDwInputGrad8()
215 int out_w = conv_param->output_w_; in ConvDwInputGrad4()
219 int out_spatial = conv_param->output_h_ * conv_param->output_w_; in ConvDwInputGrad4()
291 int out_w = conv_param->output_w_; in ConvDwInputGradFp16()
295 int out_spatial = conv_param->output_h_ * conv_param->output_w_; in ConvDwInputGradFp16()
|
| D | convolution_grad_filter.c | 36 int out_w = conv_param->output_w_; in FilterGrad32Arm()
114 int out_w = conv_param->output_w_; in FilterGrad16Arm()
172 int out_w = conv_param->output_w_; in FilterGrad8Arm()
221 int out_w = conv_param->output_w_; in FilterGrad4Arm()
269 int out_w = conv_param->output_w_; in FilterGradLeftoverArm()
319 int out_w = conv_param->output_w_; in ConvDwFilterFp16Grad()
|
| D | pack_fp16_ext.c | 37 const int output_w = conv_param->output_w_; in RollingIm2ColPackDwUnitFp16()
87 const int output_w = conv_param->output_w_; in RollingIm2ColPackUnitFp16()
154 const int output_w = conv_param->output_w_; in RollingCol2ImPackUnitFp16()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | deconvolution_depthwise_int8.cc | 85 conv_param_->output_w_ = in_tensors_.front()->shape().at(kNHWC_W); in InitSlideParam()
109 …utput_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w_ * C4NUM * in InitBuffer()
120 …conv_param_->output_h_ * conv_param_->output_w_ * C4NUM * conv_param_->thread_num_ * sizeof(int32_… in InitBuffer()
169 auto buffer = output_buffer_ + conv_param_->output_h_ * conv_param_->output_w_ * C4NUM * task_id; in DoExecute()
228 … conv_param_->output_h_ * conv_param_->output_w_, conv_param_->output_channel_); in Run()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | pooling_grad_fp32_tests.cc | 42 pooling_param->output_w_ = 28; in InitPoolingParamFP32()
72 …tput_batch_ * pooling_param->output_channel_ * pooling_param->output_h_ * pooling_param->output_w_; in TEST_F()
127 …tput_batch_ * pooling_param->output_channel_ * pooling_param->output_h_ * pooling_param->output_w_; in TEST_F()
271 pool->output_w_ = 14; in TEST_F()
287 …std::vector<int> dim_y({pool->output_batch_, pool->output_h_, pool->output_w_, pool->output_channe… in TEST_F()
335 pool->output_w_ = 10; in TEST_F()
351 …std::vector<int> dim_y({pool->output_batch_, pool->output_h_, pool->output_w_, pool->output_channe… in TEST_F()
402 …tput_batch_ * pooling_param->output_channel_ * pooling_param->output_h_ * pooling_param->output_w_; in TEST_F()
538 maxpool->output_w_ = 14; in TEST_F()
554 …std::vector<int> dim_y({maxpool->output_batch_, maxpool->output_h_, maxpool->output_w_, maxpool->o… in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | conv1x1_base.c | 29 char *dst_h_ptr = dst + dst_h * conv_param->output_w_ * conv_param->input_channel_ * data_size; in Conv1x1InputPack()
30 for (int dst_w = 0; dst_w < conv_param->output_w_; dst_w++) { in Conv1x1InputPack()
|
| D | conv_common_base.c | 27 conv_param->input_channel_ == conv_param->output_channel_ && conv_param->output_w_ >= 4 && in CheckConvDw1DWinograd()
49 int out_w = conv_param->output_w_; in SelectOutputUnit()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | deconv_int8.c | 25 int output_plane = conv_param->output_w_ * conv_param->output_h_; in DeConvPostInt8C4()
33 int dst_oh_stride = conv_param->output_w_ * C4NUM; in DeConvPostInt8C4()
35 int dst_kh_stride = conv_param->dilation_h_ * conv_param->output_w_ * C4NUM; in DeConvPostInt8C4()
51 …int kw_end = MSMIN(conv_param->kernel_w_, UP_DIV(conv_param->output_w_ - ow, conv_param->dilation_… in DeConvPostInt8C4()
|
| D | conv_depthwise_int8.c | 98 …int8_t *dst = output_data + b * conv_param->output_h_ * conv_param->output_w_ * conv_param->output… in ConvDwInt8()
100 int8_t *dst_data = dst + oh * conv_param->output_w_ * conv_param->output_channel_; in ConvDwInt8()
107 for (int ow = 0; ow < conv_param->output_w_; ow++) { in ConvDwInt8()
120 int out_w_end = MSMIN(conv_param->output_w_, (conv_param->input_w_ + conv_param->pad_l_ - in ConvDwInt8()
135 …ConvDwInt8Post(dst_data, row_buffer, conv_param->output_w_, conv_param->output_channel_, output_zp, in ConvDwInt8()
318 …int8_t *dst = output_data + b * conv_param->output_h_ * conv_param->output_w_ * conv_param->output… in ConvDw3x3Int8()
319 start_oh * conv_param->output_w_ * conv_param->output_channel_ + in ConvDw3x3Int8()
326 dst += conv_param->output_w_ * conv_param->output_channel_; in ConvDw3x3Int8()
425 int output_row_size = conv_param->output_w_ * conv_param->output_channel_; in ConvDw3x3Int8Pad()
434 output_data + b * conv_param->output_h_ * conv_param->output_w_ * conv_param->output_channel_; in ConvDw3x3Int8Pad()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | pack_ext.c | 38 const int output_w = conv_param->output_w_; in RollingIm2ColPackDwUnitFp32()
88 const int output_w = conv_param->output_w_; in rolling_im2col_hwc()
157 const int in_width = conv_param->output_w_; in rolling_im2row_hwc()
208 const int output_w = conv_param->output_w_; in col2im_hwc()
255 const int output_w = conv_param->output_w_; in rolling_col2im_hwc()
|
| D | convolution_grad_filter.c | 34 int out_w = conv_param->output_w_; in FilterGrad16Arm()
111 int out_w = conv_param->output_w_; in FilterGrad12Arm()
181 int out_w = conv_param->output_w_; in FilterGrad8Arm()
240 int out_w = conv_param->output_w_; in FilterGrad4Arm()
290 int out_w = conv_param->output_w_; in Filtergrad2Arm()
338 int out_w = conv_param->output_w_; in ConvDwFilterGrad()
|
| D | convolution_grad_input.c | 26 int out_w = conv_param->output_w_; in ConvDwInputGrad()
29 int out_spatial = conv_param->output_h_ * conv_param->output_w_; in ConvDwInputGrad()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/ |
| D | pack_tests.cc | 41 conv_param->output_w_ = 28; in InitConvParamPack()
70 int out_w = conv_param->output_w_; in TEST_F()
133 int out_w = conv_param->output_w_; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_depthwise_fp16.cc | 42 conv_param_->output_w_ = in_tensors_.front()->shape().at(kNHWC_W); in InitSlideParam()
59 …t_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w_ * C8NUM * C8; in InitPackedInputOutput()
209 … conv_param_->output_h_ * conv_param_->output_w_, conv_param_->output_channel_); in Run()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | deconvolution_depthwise_fp32.cc | 41 conv_param_->output_w_ = in_tensors_.front()->shape().at(kNHWC_W); in InitSlideParam()
59 …_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w_ * C4NUM * OC4; in InitPackedInputOutput()
183 … conv_param_->output_h_ * conv_param_->output_w_, conv_param_->output_channel_, C4NUM); in Run()
|
| D | convolution_depthwise_3x3_fp32.cc | 60 int units = UP_DIV(conv_param_->output_w_, C2NUM); // F(2, 3) contains 2 conv units in DoExecute()
86 int units = UP_DIV(conv_param_->output_w_, C2NUM); // F(2, 3) contains 2 conv units in Run()
|