Home
last modified time | relevance | path

Searched refs:input_h_ (Results 1 – 25 of 89) sorted by relevance

1234

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dresize_bilinear_gpu_kernel.h44 float h_scale = Scaling(input_h_, output_h_, align_corners_); in Launch()
46 …CalResizeBilinear(input, n_, c_, input_h_, input_w_, output_h_, output_w_, h_scale, w_scale, outpu… in Launch()
77 input_h_ = SizeToInt(input_shape[2]); in Init()
99 input_h_ = 0; in ResetResource()
127 int input_h_; variable
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dconv1x1_fp32_tests.cc47 conv_param->input_h_ = 9; in TEST_F()
88 conv_param->input_h_ = 19; 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/
Dresize_bilinear_impl.cuh21 void CalResizeBilinear(const T *input, const int n_, const int c_, const int input_h_, const int in…
24 void CalResizeBilinearGrad(const half *input, const int n_, const int c_, const int input_h_, const…
27 void CalResizeBilinearGrad(const float *input, const int n_, const int c_, const int input_h_, cons…
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dpack_tests.cc35 conv_param->input_h_ = 28; in InitConvParamPack()
67 int in_h = conv_param->input_h_; in TEST_F()
130 int in_h = conv_param->input_h_; in TEST_F()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_depthwise_indirect_fp32.cc122 …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()
Dconvolution_delegate_fp32.cc149 conv_param->input_h_ = input->Height(); in SetInputOutputShapeInfo()
162 conv_param->input_h_ >= conv_param->thread_num_ && in CheckAvxUseSWConv()
163 (conv_param->kernel_h_ < 7 || conv_param->input_h_ / conv_param->kernel_h_ >= 4) && in CheckAvxUseSWConv()
178 (conv_param->input_w_ * conv_param->input_h_ >= conv_param->thread_num_)) { in CpuConvFp32KernelSelect()
Ddeconvolution_depthwise_fp32.cc36 conv_param_->input_h_ = out_tensors_.front()->shape().at(kNHWC_H); 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/fp32_grad/
Dpack_ext.c35 const int in_height = conv_param->input_h_; in RollingIm2ColPackDwUnitFp32()
85 const int in_height = conv_param->input_h_; in rolling_im2col_hwc()
204 const int in_height = conv_param->input_h_; in col2im_hwc()
252 const int in_height = conv_param->input_h_; in rolling_col2im_hwc()
Dconvolution_grad_filter.c26 int in_h = conv_param->input_h_; in FilterGrad16Arm()
103 int in_h = conv_param->input_h_; in FilterGrad12Arm()
173 int in_h = conv_param->input_h_; in FilterGrad8Arm()
232 int in_h = conv_param->input_h_; in FilterGrad4Arm()
282 int in_h = conv_param->input_h_; in Filtergrad2Arm()
330 int in_h = conv_param->input_h_; in ConvDwFilterGrad()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dconvolution_grad_filter.c28 int in_h = conv_param->input_h_; in FilterGrad32Arm()
106 int in_h = conv_param->input_h_; in FilterGrad16Arm()
164 int in_h = conv_param->input_h_; in FilterGrad8Arm()
213 int in_h = conv_param->input_h_; in FilterGrad4Arm()
261 int in_h = conv_param->input_h_; in FilterGradLeftoverArm()
311 int in_h = conv_param->input_h_; in ConvDwFilterFp16Grad()
Dpack_fp16_ext.c34 const int in_height = conv_param->input_h_; in RollingIm2ColPackDwUnitFp16()
84 const int in_height = conv_param->input_h_; in RollingIm2ColPackUnitFp16()
151 const int in_height = conv_param->input_h_; in RollingCol2ImPackUnitFp16()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Ddeconvolution_int8_coder.cc53 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/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Ddeconv_fp16.c22 size_t input_plane = conv_param->input_w_ * conv_param->input_h_; in DeConvPostFp16()
41 for (int ih = 0; ih < conv_param->input_h_; ih++) { in DeConvPostFp16()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Ddeconvolution_int8.cc132 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/fp16/
Ddeconvolution_depthwise_fp16.cc37 conv_param_->input_h_ = out_tensors_.front()->shape().at(kNHWC_H); 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/
Dpooling_base.cc110 pooling_param_->input_h_ = in_tensor->Height(); in ReSize()
117 pooling_param_->window_h_ = pooling_param_->input_h_; in ReSize()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dconvolution_grad_input.cc44 conv_param->input_h_ = dx_tensor->shape()[(kNHWC_H)]; in ReSize()
92 int in_h = conv_param->input_h_; in Execute()
174 int in_h = conv_param->input_h_; in Run()
Dpooling_grad.cc62 pool_param->input_h_ = in_shape[kNHWC_H]; in ReSize()
86 int in_batch_size = pool_param->input_h_ * pool_param->input_w_ * pool_param->input_channel_; in Execute()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dconvolution_fp16_grad_input.cc44 conv_param->input_h_ = dx_tensor->shape()[(kNHWC_H)]; in ReSize()
91 int in_h = conv_param->input_h_; in DoExecute()
171 int in_h = conv_param->input_h_; in Run()
Dpooling_fp16_grad.cc62 pool_param->input_h_ = in_shape[kNHWC_H]; in ReSize()
85 int in_batch_size = pool_param->input_h_ * pool_param->input_w_ * pool_param->input_channel_; in DoExecute()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c30 int in_h = pooling_param->input_h_; in AvgPoolingInt8()
99 int in_batch_offset = batch * pooling_param->input_h_ * in_w * channel; in AvgPoolingOptInt8()
115 int kh_e = MSMIN(win_h, pooling_param->input_h_ - in_h_index); in AvgPoolingOptInt8()
286 int in_h = pooling_param->input_h_; in MaxPoolingInt8()
331 int in_h = pooling_param->input_h_; in MaxPoolingWithQuantInt8()
434 int in_batch_offset = batch * pooling_param->input_h_ * in_w * channel; in MaxPoolingOptInt8()
447 int ky_e = MSMIN(pooling_param->window_h_, pooling_param->input_h_ - in_h_index); in MaxPoolingOptInt8()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Ddeconv_fp32.c40 size_t input_plane = conv_param->input_w_ * conv_param->input_h_; in DeConvPostFp32C8()
66 for (int ih = 0; ih < conv_param->input_h_; ih++) { in DeConvPostFp32C8()
Droi_pooling_fp32.h33 int input_h_; member
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Ddeconvolution_grad_fp32_tests.cc38 conv_param->input_h_ = 32; in TEST_F()
146 conv_param->input_h_ = 32; in TEST_F()
248 conv_param->input_h_ = 32; in TEST_F()
355 conv_param->input_h_ = 32; in TEST_F()
462 conv_param->input_h_ = 32; in TEST_F()
569 conv_param->input_h_ = 32; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dpooling_parameter.h40 int input_h_; member

1234