Home
last modified time | relevance | path

Searched refs:input_plane_ (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Ddeconv2d_fp32_coder.cc54 input_plane_ = conv_param_->input_h_ * conv_param_->input_w_; in InitParam()
58 matmul_param_.row_ = input_plane_; in InitParam()
176 …input_ptr_ = src_in_ptr_str + "+" + std::to_string(batch_index * input_plane_ * conv_param_->input… in DoCode()
Ddeconv2d_fp32_coder.h53 int input_plane_{0};
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_fp16.cc94 input_plane_ = conv_param_->input_h_ * conv_param_->input_w_; in InitParam()
98 matmul_param_->row_ = input_plane_; in InitParam()
236 batch_input_ = input_ptr + batch_index * conv_param_->input_channel_ * input_plane_; in Run()
239 … RowMajor2Col16MajorFp16Opt(batch_input_, pack_input_, input_plane_, conv_param_->input_channel_); in Run()
Ddeconvolution_fp16.h50 int input_plane_ = 0; variable
Ddeconvolution_winograd_fp16.cc114 deconv_param_->input_plane_ = conv_param_->input_h_ * conv_param_->input_w_; in InitParameter()
460 … nhwc_input_ = input_ptr + batch_index * deconv_param_->input_plane_ * conv_param_->input_channel_; in Run()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_fp32.h54 int input_plane_ = 0; variable
Ddeconvolution_fp32.cc99 input_plane_ = conv_param_->input_h_ * conv_param_->input_w_; in InitParam()
103 matmul_param_->row_ = input_plane_; in InitParam()
267 input_ptr_ = src_in + batch_index * input_plane_ * conv_param_->input_channel_; in Run()
Ddeconvolution_winograd_fp32.cc107 deconv_param_->input_plane_ = conv_param_->input_h_ * conv_param_->input_w_; in InitParameter()
495 nhwc_input_ = src_in + batch_index * deconv_param_->input_plane_ * conv_param_->input_channel_; in Run()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h120 int input_plane_; member