| /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()
78 input_w_ = SizeToInt(input_shape[3]); in Init()
100 input_w_ = 0; in ResetResource()
128 int input_w_; variable
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | conv1x1_fp32_tests.cc | 48 conv_param->input_w_ = 13; in TEST_F()
89 conv_param->input_w_ = 10; in TEST_F()
105 conv_param->input_h_ = conv_param->input_w_ = 3; in TEST_F()
124 conv_param->input_h_ = conv_param->input_w_ = 3; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | resize_bilinear_impl.cuh | 21 …lResizeBilinear(const T *input, const int n_, const int c_, const int input_h_, const int input_w_,
24 …BilinearGrad(const half *input, const int n_, const int c_, const int input_h_, const int input_w_,
27 …ilinearGrad(const float *input, const int n_, const int c_, const int input_h_, const int input_w_,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | deconv_fp16.c | 22 size_t input_plane = conv_param->input_w_ * conv_param->input_h_; in DeConvPostFp16()
28 int src_ih_stride = conv_param->input_w_ * C8NUM; in DeConvPostFp16()
42 for (int iw = 0; iw < conv_param->input_w_; iw++) { in DeConvPostFp16()
|
| D | pooling_fp16.c | 26 int in_w = pooling_param->input_w_; in AvgPoolingFp16()
142 int in_w = pooling_param->input_w_; in MaxPoolingC8Fp16()
186 int in_w = pooling_param->input_w_; in MaxPoolingC4Fp16()
232 int in_w = pooling_param->input_w_; in MaxPoolingC1Fp16()
262 int in_w = pooling_param->input_w_; in MaxPoolingFp16()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | deconvolution_depthwise_int8.cc | 81 conv_param_->input_w_ = out_tensors_.front()->shape().at(kNHWC_W); in InitSlideParam()
90 sliding_->in_h_step_ = conv_param_->input_w_ * C4NUM; in InitSlideParam()
91 sliding_->in_sh_step_ = conv_param_->input_w_ * C4NUM * conv_param_->stride_h_; // stride H in InitSlideParam()
93 sliding_->in_kh_step_ = conv_param_->input_w_ * C4NUM * conv_param_->dilation_h_; // kernel H in InitSlideParam()
99 …int pack_input_size = conv_param_->input_batch_ * conv_param_->input_h_ * conv_param_->input_w_ * … in InitBuffer()
|
| D | deconvolution_int8.cc | 132 matmul_param_->row_ = conv_param_->input_h_ * conv_param_->input_w_; in InitParam()
188 …UP_ROUND(conv_param_->input_h_ * conv_param_->input_w_, C4NUM) * UP_ROUND(conv_param_->input_chann… in InitData()
198 int size = UP_ROUND(conv_param_->input_h_ * conv_param_->input_w_, C4NUM) * in InitRunBuf()
258 int input_plane = conv_param_->input_h_ * conv_param_->input_w_; in DoDeconv()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | deconvolution_grad_filter.cc | 51 conv_param->input_w_ = x_tensor->shape().at(kNHWC_W); in Init()
59 …ws_size = chunk * conv_param->input_w_ * conv_param->kernel_h_ * conv_param->kernel_w_ * conv_para… in Init()
64 size_t mat_alloc = MatSizeTotal(n, m, chunk * conv_param->input_w_, conv_param->input_channel_); in Init()
89 int in_w = conv_param->input_w_; in Execute()
|
| D | convolution_grad_input.cc | 45 conv_param->input_w_ = dx_tensor->shape()[(kNHWC_W)]; in ReSize()
93 int in_w = conv_param->input_w_; in Execute()
175 int in_w = conv_param->input_w_; in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | conv1x1_base.c | 28 …const char *src_h_ptr = src + src_h * conv_param->input_w_ * conv_param->input_channel_ * data_siz… in Conv1x1InputPack()
32 if (src_w < 0 || src_w >= conv_param->input_w_) { in Conv1x1InputPack()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | deconv_fp32.c | 40 size_t input_plane = conv_param->input_w_ * conv_param->input_h_; in DeConvPostFp32C8()
51 int src_ih_stride = conv_param->input_w_ * C8NUM; in DeConvPostFp32C8()
67 for (int iw = 0; iw < conv_param->input_w_; iw++) { in DeConvPostFp32C8()
|
| D | conv_depthwise_fp32.c | 46 …const float *src = input_data + b * conv_param->input_h_ * conv_param->input_w_ * conv_param->inpu… in ConvDw()
62 const float *src_kh = src + ih * conv_param->input_w_ * conv_param->input_channel_; in ConvDw()
69 int out_w_end = MSMIN(conv_param->output_w_, (conv_param->input_w_ + conv_param->pad_l_ - in ConvDw()
107 conv_param->input_w_ && in InitSlidingParam()
154 …sliding->in_step_ = conv_param->input_h_ * conv_param->input_w_ * sliding->ic_align_; // for batc… in AppendSlidingParamConv()
155 sliding->in_h_step_ = conv_param->input_w_ * sliding->ic_align_; in AppendSlidingParamConv()
156 …sliding->in_sh_step_ = conv_param->input_w_ * sliding->ic_align_ * conv_param->stride_h_; // st… in AppendSlidingParamConv()
158 …sliding->in_kh_step_ = conv_param->input_w_ * sliding->ic_align_ * conv_param->dilation_h_; // ke… in AppendSlidingParamConv()
169 …sliding->in_step_ = conv_param->input_h_ * conv_param->input_w_ * sliding->block_channel_; // for… in AppendSlidingParamConvDw()
170 sliding->in_h_step_ = conv_param->input_w_ * sliding->block_channel_; in AppendSlidingParamConvDw()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | pack_ext.c | 36 const int in_width = conv_param->input_w_; in RollingIm2ColPackDwUnitFp32()
86 const int in_width = conv_param->input_w_; in rolling_im2col_hwc()
159 const int output_w = conv_param->input_w_; in rolling_im2row_hwc()
205 const int in_width = conv_param->input_w_; in col2im_hwc()
253 const int in_width = conv_param->input_w_; in rolling_col2im_hwc()
|
| D | convolution_grad_filter.c | 27 int in_w = conv_param->input_w_; in FilterGrad16Arm()
104 int in_w = conv_param->input_w_; in FilterGrad12Arm()
174 int in_w = conv_param->input_w_; in FilterGrad8Arm()
233 int in_w = conv_param->input_w_; in FilterGrad4Arm()
283 int in_w = conv_param->input_w_; in Filtergrad2Arm()
331 int in_w = conv_param->input_w_; in ConvDwFilterGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv_depthwise_int8.c | 97 …const int8_t *src = input_data + b * conv_param->input_h_ * conv_param->input_w_ * conv_param->inp… in ConvDwInt8()
113 const int8_t *src_kh = src + ih * conv_param->input_w_ * conv_param->input_channel_; in ConvDwInt8()
120 int out_w_end = MSMIN(conv_param->output_w_, (conv_param->input_w_ + conv_param->pad_l_ - in ConvDwInt8()
152 input += conv_param->input_w_ * conv_param->input_channel_; in ConvDw3x3Int8InitBuffer()
253 …if (conv_param->output_channel_ > 64 || (conv_param->output_channel_ < 64 && conv_param->input_w_ … in ConvDw3x3Int8Row()
280 … conv_param->input_channel_, conv_param->input_w_ * conv_param->input_channel_, in ConvDw3x3Int8Row()
291 … conv_param->input_w_ * conv_param->input_channel_, conv_param->input_channel_, block_output_h, in ConvDw3x3Int8Row()
315 …const int8_t *src = input_data + b * conv_param->input_h_ * conv_param->input_w_ * conv_param->inp… in ConvDw3x3Int8()
316 start_ih * conv_param->input_w_ * conv_param->input_channel_ + in ConvDw3x3Int8()
325 src += conv_param->stride_h_ * conv_param->input_w_ * conv_param->input_channel_; in ConvDw3x3Int8()
[all …]
|
| D | deconv_int8.c | 23 int input_plane = conv_param->input_w_ * conv_param->input_h_; in DeConvPostInt8C4()
30 int src_ih_stride = conv_param->input_w_ * C4NUM; in DeConvPostInt8C4()
44 for (int iw = 0; iw < conv_param->input_w_; iw++) { in DeConvPostInt8C4()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/ |
| D | pack_tests.cc | 36 conv_param->input_w_ = 28; in InitConvParamPack()
68 int in_w = conv_param->input_w_; in TEST_F()
131 int in_w = conv_param->input_w_; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_depthwise_indirect_fp32.cc | 122 …int pack_input_size = conv_param_->input_batch_ * conv_param_->input_h_ * conv_param_->input_w_ * … in MallocPackedInput()
148 conv_param_->input_h_ * conv_param_->input_w_, conv_param_->input_channel_); in Run()
151 conv_param_->input_h_ * conv_param_->input_w_, conv_param_->input_channel_); in Run()
|
| D | deconvolution_depthwise_fp32.cc | 37 conv_param_->input_w_ = out_tensors_.front()->shape().at(kNHWC_W); in InitSlideParam()
51 …int pack_input_size = conv_param_->input_batch_ * conv_param_->input_h_ * conv_param_->input_w_ * … in InitPackedInputOutput()
166 conv_param_->input_h_ * conv_param_->input_w_, conv_param_->input_channel_); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | convolution_grad_filter.c | 29 int in_w = conv_param->input_w_; in FilterGrad32Arm()
107 int in_w = conv_param->input_w_; in FilterGrad16Arm()
165 int in_w = conv_param->input_w_; in FilterGrad8Arm()
214 int in_w = conv_param->input_w_; in FilterGrad4Arm()
262 int in_w = conv_param->input_w_; in FilterGradLeftoverArm()
312 int in_w = conv_param->input_w_; in ConvDwFilterFp16Grad()
|
| D | pack_fp16_ext.c | 35 const int in_width = conv_param->input_w_; in RollingIm2ColPackDwUnitFp16()
85 const int in_width = conv_param->input_w_; in RollingIm2ColPackUnitFp16()
152 const int in_width = conv_param->input_w_; in RollingCol2ImPackUnitFp16()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | deconvolution_int8_coder.cc | 53 matmul_param_->row_ = conv_param_->input_h_ * conv_param_->input_w_; in InitParam()
102 input_ptr_size_ = UP_ROUND(conv_param_->input_h_ * conv_param_->input_w_, C4NUM) * in InitData()
109 tmp_buffer_size_ = UP_ROUND(conv_param_->input_h_ * conv_param_->input_w_, C4NUM) * in InitRunBuf()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_depthwise_fp16.cc | 38 conv_param_->input_w_ = out_tensors_.front()->shape().at(kNHWC_W); in InitSlideParam()
52 …int pack_input_size = conv_param_->input_batch_ * conv_param_->input_h_ * conv_param_->input_w_ * … in InitPackedInputOutput()
193 conv_param_->input_h_ * conv_param_->input_w_, conv_param_->input_channel_); in Run()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | pooling_base.cc | 111 pooling_param_->input_w_ = in_tensor->Width(); in ReSize()
118 pooling_param_->window_w_ = pooling_param_->input_w_; in ReSize()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | convolution_fp16_grad_input.cc | 45 conv_param->input_w_ = dx_tensor->shape()[(kNHWC_W)]; in ReSize()
92 int in_w = conv_param->input_w_; in DoExecute()
172 int in_w = conv_param->input_w_; in Run()
|