Searched refs:output_plane_ (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_winograd_fp16.cc | 115 deconv_param_->output_plane_ = conv_param_->output_h_ * conv_param_->output_w_; in InitParameter()
118 …reinterpret_cast<float16_t *>(malloc(deconv_param_->oc_up4_ * deconv_param_->output_plane_ * sizeo… in InitParameter()
131 thread_num_hw_ = MSMIN(op_parameter_->thread_num_, deconv_param_->output_plane_); in InitParameter()
132 thread_stride_hw_ = UP_DIV(deconv_param_->output_plane_, thread_num_hw_); in InitParameter()
216 int rest_plane = deconv_param_->output_plane_ - task_id * thread_stride_hw_; in DeDeconvPost()
225 deconv_param_->output_plane_, conv_param_->act_type_); in DeDeconvPost()
461 …nhwc_output_ = output_ptr + batch_index * deconv_param_->output_plane_ * conv_param_->output_chann… in Run()
463 …::memset(nc4hw4_output_, 0, deconv_param_->output_plane_ * deconv_param_->oc_div4_ * C4NUM * sizeo… in Run()
|
| D | deconvolution_fp16.cc | 96 output_plane_ = conv_param_->output_h_ * conv_param_->output_w_; in InitParam()
112 …ctx_->allocator->Malloc(UP_ROUND(conv_param_->output_channel_, C8NUM) * output_plane_ * sizeof(flo… in InitRunBuf()
176 DeConvPostFp16(tmp_buf, pack_output_ + task_id * thread_stride_ * C8NUM * output_plane_, in DoDeconv()
237 batch_output_ = output_ptr + batch_index * conv_param_->output_channel_ * output_plane_; in Run()
|
| D | deconvolution_fp16.h | 52 int output_plane_ = 0; variable
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | deconvolution_winograd_fp32.cc | 108 deconv_param_->output_plane_ = conv_param_->output_h_ * conv_param_->output_w_; in InitParameter()
118 thread_num_hw_ = MSMIN(op_parameter_->thread_num_, deconv_param_->output_plane_); in InitParameter()
119 thread_stride_hw_ = UP_DIV(deconv_param_->output_plane_, thread_num_hw_); in InitParameter()
420 int rest_plane = deconv_param_->output_plane_ - task_id * thread_stride_hw_; in DeDeconvPost()
429 deconv_param_->output_plane_, conv_param_->act_type_); in DeDeconvPost()
434 int size = deconv_param_->oc_up4_ * deconv_param_->output_plane_; in InitRunBuf()
496 …nhwc_output_ = src_out + batch_index * deconv_param_->output_plane_ * conv_param_->output_channel_; in Run()
498 …::memset(nc4hw4_output_, 0, deconv_param_->output_plane_ * deconv_param_->oc_div4_ * C4NUM * sizeo… in Run()
|
| D | deconvolution_fp32.cc | 101 output_plane_ = conv_param_->output_h_ * conv_param_->output_w_; in InitParam()
158 DeConvPostFp32C8(tmp_buffer, pack_output_ + task_id * thread_stride_ * C8NUM * output_plane_, in DoDeconv()
224 …ctx_->allocator->Malloc(UP_ROUND(conv_param_->output_channel_, C8NUM) * output_plane_ * sizeof(flo… in InitRunBuf()
268 output_ptr_ = src_out + batch_index * output_plane_ * conv_param_->output_channel_; in Run()
|
| D | deconvolution_fp32.h | 56 int output_plane_ = 0; variable
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | deconv2d_fp32_coder.cc | 31 pack_output_size_ = UP_ROUND(conv_param_->output_channel_, C8NUM) * output_plane_ * sizeof(float); in InitRunBuf()
56 output_plane_ = conv_param_->output_h_ * conv_param_->output_w_; in InitParam()
177 …output_ptr_ = src_out_ptr_str + "+" + std::to_string(batch_index * output_plane_ * conv_param_->ou… in DoCode()
|
| D | deconv2d_fp32_coder.h | 55 int output_plane_{0};
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 121 int output_plane_; member
|