Home
last modified time | relevance | path

Searched refs:output_h_ (Results 1 – 25 of 97) 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()
79 output_h_ = SizeToInt(output_shape[2]); in Init()
101 output_h_ = 0; in ResetResource()
129 int output_h_; variable
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dconv1x1_fp32_tests.cc50 conv_param->output_h_ = 4; in TEST_F()
91 conv_param->output_h_ = 7; 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/
Dresize_bilinear_impl.cuh22 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_grad/
Dconvolution_grad_input.c28 int out_h = conv_param->output_h_; in ConvDwInputGrad16()
31 int out_spatial = conv_param->output_h_ * conv_param->output_w_; in ConvDwInputGrad16()
133 int out_h = conv_param->output_h_; in ConvDwInputGrad8()
136 int out_spatial = conv_param->output_h_ * conv_param->output_w_; in ConvDwInputGrad8()
216 int out_h = conv_param->output_h_; in ConvDwInputGrad4()
219 int out_spatial = conv_param->output_h_ * conv_param->output_w_; in ConvDwInputGrad4()
292 int out_h = conv_param->output_h_; in ConvDwInputGradFp16()
295 int out_spatial = conv_param->output_h_ * conv_param->output_w_; in ConvDwInputGradFp16()
Dconvolution_grad_filter.c35 int out_h = conv_param->output_h_; in FilterGrad32Arm()
113 int out_h = conv_param->output_h_; in FilterGrad16Arm()
171 int out_h = conv_param->output_h_; in FilterGrad8Arm()
220 int out_h = conv_param->output_h_; in FilterGrad4Arm()
268 int out_h = conv_param->output_h_; in FilterGradLeftoverArm()
318 int out_h = conv_param->output_h_; in ConvDwFilterFp16Grad()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Ddeconvolution_depthwise_int8.cc84 conv_param_->output_h_ = in_tensors_.front()->shape().at(kNHWC_H); in InitSlideParam()
109 …int pack_output_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w… 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/
Dpooling_grad_fp32_tests.cc41 pooling_param->output_h_ = 28; in InitPoolingParamFP32()
72 …pooling_param->output_batch_ * pooling_param->output_channel_ * pooling_param->output_h_ * pooling… in TEST_F()
127 …pooling_param->output_batch_ * pooling_param->output_channel_ * pooling_param->output_h_ * pooling… in TEST_F()
270 pool->output_h_ = 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()
334 pool->output_h_ = 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 …pooling_param->output_batch_ * pooling_param->output_channel_ * pooling_param->output_h_ * pooling… in TEST_F()
537 maxpool->output_h_ = 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/lite/src/runtime/kernel/arm/fp32/
Dconvolution_depthwise_3x3_fp32.cc55 conv_param_->thread_num_ = MSMIN(thread_count_, conv_param_->output_h_); in ReSize()
67 int step_oh = UP_DIV(conv_param_->output_h_, conv_param_->thread_num_); in DoExecute()
69 int end_oh = MSMIN(start_oh + step_oh, conv_param_->output_h_); in DoExecute()
Dconvolution_depthwise_slidewindow_fp32.cc44 …int pack_output_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w… in InitPackedInputOutput()
86 conv_param_->thread_num_ = MSMIN(thread_count_, conv_param_->output_h_); in ReSize()
148 … conv_param_->output_h_ * conv_param_->output_w_, conv_param_->output_channel_, C4NUM); in Run()
Ddeconvolution_depthwise_fp32.cc40 conv_param_->output_h_ = in_tensors_.front()->shape().at(kNHWC_H); in InitSlideParam()
59 …int pack_output_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w… in InitPackedInputOutput()
183 … conv_param_->output_h_ * conv_param_->output_w_, conv_param_->output_channel_, C4NUM); in Run()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dconvolution_depthwise_3x3_fp16.cc93 conv_param_->thread_num_ = MSMIN(thread_count_, conv_param_->output_h_); in ReSize()
101 int step_oh = UP_DIV(conv_param_->output_h_, conv_param_->thread_num_); in DoExecute()
103 int end_oh = MSMIN(start_oh + step_oh, conv_param_->output_h_); in DoExecute()
Ddeconvolution_depthwise_fp16.cc41 conv_param_->output_h_ = in_tensors_.front()->shape().at(kNHWC_H); in InitSlideParam()
59 …int pack_output_size = conv_param_->output_batch_ * conv_param_->output_h_ * conv_param_->output_w… in InitPackedInputOutput()
209 … 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/common/
Dpack_tests.cc40 conv_param->output_h_ = 28; in InitConvParamPack()
69 int out_h = conv_param->output_h_; in TEST_F()
132 int out_h = conv_param->output_h_; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconv_common_base.c28 conv_param->output_h_ >= thread_num * 4; // better had more than 4 rows for each thread in CheckConvDw1DWinograd()
50 int out_h = conv_param->output_h_; in SelectOutputUnit()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c32 int output_h = pooling_param->output_h_; in AvgPoolingInt8()
89 int out_plane = output_w * pooling_param->output_h_; in AvgPoolingOptInt8()
100 int out_batch_offset = batch * pooling_param->output_h_ * output_w * channel; in AvgPoolingOptInt8()
288 int output_h = pooling_param->output_h_; in MaxPoolingInt8()
333 int out_plane = output_w * pooling_param->output_h_; in MaxPoolingWithQuantInt8()
347 int out_batch_offset = batch * pooling_param->output_h_ * output_w * channel; in MaxPoolingWithQuantInt8()
427 int out_plane = output_w * pooling_param->output_h_; in MaxPoolingOptInt8()
435 int out_batch_offset = batch * pooling_param->output_h_ * output_w * channel; in MaxPoolingOptInt8()
Dconv_int8.c25 int output_count = conv_param->output_h_ * conv_param->output_w_; in ConvInt8()
55 int out_batch_offset = b * out_channel * conv_param->output_h_ * conv_param->output_w_; in ConvInt8()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dconvolution_grad_filter.c33 int out_h = conv_param->output_h_; in FilterGrad16Arm()
110 int out_h = conv_param->output_h_; in FilterGrad12Arm()
180 int out_h = conv_param->output_h_; in FilterGrad8Arm()
239 int out_h = conv_param->output_h_; in FilterGrad4Arm()
289 int out_h = conv_param->output_h_; in Filtergrad2Arm()
337 int out_h = conv_param->output_h_; in ConvDwFilterGrad()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_depthwise_fp32.c40 int h_step = UP_DIV(conv_param->output_h_, conv_param->thread_num_); in ConvDw()
42 int h_end = MSMIN(h_start + h_step, conv_param->output_h_); in ConvDw()
47 …float *dst = output_data + b * conv_param->output_h_ * conv_param->output_w_ * conv_param->output_… in ConvDw()
101 int bottom = conv_param->output_h_; in InitSlidingParam()
125 sliding->out_step_ = conv_param->output_h_ * conv_param->output_w_ * sliding->block_channel_; in InitSlidingParam()
129 sliding->out_c_step_ = block * conv_param->output_h_ * conv_param->output_w_; in InitSlidingParam()
321 …ConvDwBorder(dst_data, src_data, weight, bias, sliding->bottom_, conv_param->output_h_, 0, conv_pa… in ConvDwSWFp32()
365 const int in_h = (conv_param->output_h_ - 1) * conv_param->stride_h_ + conv_param->kernel_h_; in CheckConvDwUse3X3()
684 …float *dst = output_data + b * conv_param->output_h_ * conv_param->output_w_ * conv_param->output_… in ConvDw3x3()
707 if (oh == conv_param->output_h_ - 1) { in ConvDw3x3()
[all …]
Dconv_winograd_fp32.c32 int out_h_block = UP_DIV(conv_param->output_h_, conv_param->output_unit_); in ConvWinogardFp32()
60 …out_batch_offset = b * conv_param->output_channel_ * conv_param->output_w_ * conv_param->output_h_; in ConvWinogardFp32()
Ddeconv_fp32.c42 size_t output_plane = conv_param->output_w_ * conv_param->output_h_; in DeConvPostFp32C8()
72 …int kh_end = MSMIN(conv_param->kernel_h_, UP_DIV(conv_param->output_h_ - oh, conv_param->dilation_… in DeConvPostFp32C8()
Droi_pooling_fp32.h37 int output_h_; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Ddeconv_fp16.c24 size_t output_plane = conv_param->output_w_ * conv_param->output_h_; in DeConvPostFp16()
47 …int kh_end = MSMIN(conv_param->kernel_h_, UP_DIV(conv_param->output_h_ - oh, conv_param->dilation_… in DeConvPostFp16()
Dconv_fp16.c33 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()
200 int out_h_block = UP_DIV(conv_param->output_h_, conv_param->output_unit_); in ConvWinogardFp16()
221 …int out_batch_offset = b * conv_param->output_channel_ * conv_param->output_h_ * conv_param->outpu… in ConvWinogardFp16()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/
Dnnacl_fp32_serializer.cc33 … pooling_parameter.input_channel_, pooling_parameter.output_w_, pooling_parameter.output_h_, in CodeStruct()
63 code << "int thread_num = MSMIN(" << gThreadNum << ", " << conv_parameter.output_h_ << ");\n"; in CodeStruct()
69 …conv_parameter.input_channel_, conv_parameter.output_batch_, conv_parameter.output_h_, conv_parame… in CodeStruct()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dpooling_fp16_grad.cc66 pool_param->output_h_ = out_shape[kNHWC_H]; in ReSize()
86 int out_batch_size = pool_param->output_h_ * pool_param->output_w_ * pool_param->input_channel_; in DoExecute()

1234