Searched refs:oc_up4_ (Results 1 – 5 of 5) sorted by relevance
190 float16_t *dst = tmp_buf + index * deconv_param->oc_up4_ * DECONV_WINOGRAD_DEFAULT_TILE; in DeConvWgCalWgFp16()191 const float16_t *weight = weight_buf + index * deconv_param->ic_up4_ * deconv_param->oc_up4_; in DeConvWgCalWgFp16()207 float16_t *dst = tile_out + w_index * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_up4_ + in DeConvWgCalWgFp16()208 … h_index * deconv_param->out_tile_w_ * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_up4_; in DeConvWgCalWgFp16()209 …_t *src = tmp_buf + (uwi + uhi * unit_size) * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_up4_; in DeConvWgCalWgFp16()221 int out_stride = DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_up4_; in DeConvWgCalCommFp16()325 size = deconv_param->ic_up4_ * deconv_param->oc_up4_ * tmp_kernel_plane; in PackDeConvWgDataFp16()332 …int dst_index = upi * deconv_param->oc_up4_ * deconv_param->ic_up4_ + oc4div * C4NUM * deconv_para… in PackDeConvWgDataFp16()382 … deconv_param->oc_up4_ * DECONV_WINOGRAD_DEFAULT_TILE; in DeconvWgFp16()392 deconv_param->oc_up4_; in DeconvWgFp16()[all …]
144 size = deconv_param->ic_up4_ * deconv_param->oc_up4_ * tmp_kernel_plane; in PackDeConvWgDataFp32()151 …int dst_index = upi * deconv_param->oc_up4_ * deconv_param->ic_up4_ + oc4div * C4NUM * deconv_para… in PackDeConvWgDataFp32()457 float *dst = tmp_buf + index * deconv_param->oc_up4_ * DECONV_WINOGRAD_DEFAULT_TILE; in DeConvWgCalWgFp32()458 const float *weight = weight_buf + index * deconv_param->ic_up4_ * deconv_param->oc_up4_; in DeConvWgCalWgFp32()474 float *dst = tile_out + w_index * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_up4_ + in DeConvWgCalWgFp32()475 … h_index * deconv_param->out_tile_w_ * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_up4_; in DeConvWgCalWgFp32()476 …at *src = tmp_buf + (uwi + uhi * unit_size) * DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_up4_; in DeConvWgCalWgFp32()488 int out_stride = DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_up4_; in DeConvWgCalCommFp32()570 … deconv_param->oc_up4_ * DECONV_WINOGRAD_DEFAULT_TILE; in DeconvWg()590 int src_unit_stride = deconv_param->oc_up4_ * DECONV_WINOGRAD_DEFAULT_TILE; in DeconvWgPost()
118 …reinterpret_cast<float16_t *>(malloc(deconv_param_->oc_up4_ * deconv_param_->output_plane_ * sizeo… in InitParameter()145 DECONV_WINOGRAD_DEFAULT_TILE * deconv_param_->oc_up4_; in InitParameter()174 … deconv_param_->oc_up4_ * DECONV_WINOGRAD_DEFAULT_TILE * sizeof(float16_t)); in InitParameter()263 deconv_param_->oc_up4_ = deconv_param_->oc_div4_ * C4NUM; in InitComputeParam()309 unit.weight_ = malloc(unit.winograd_.kh_ * unit.winograd_.kw_ * deconv_param_->oc_up4_ * in InitComputeParam()316 …unit.weight_ = malloc(h_size * w_size * deconv_param_->ic_up4_ * deconv_param_->oc_up4_ * sizeof(f… in InitComputeParam()348 bias_data_ = malloc(deconv_param_->oc_up4_ * sizeof(float16_t)); in InitDataParam()353 memset(bias_data_, 0, deconv_param_->oc_up4_ * sizeof(float16_t)); in InitDataParam()
159 … deconv_param_->oc_up4_ * DECONV_WINOGRAD_DEFAULT_TILE * sizeof(float)); in InitParameter()216 deconv_param_->oc_up4_ = deconv_param_->oc_div4_ * C4NUM; in InitComputeParam()266 unit.weight_ = malloc(unit.winograd_.kh_ * unit.winograd_.kw_ * deconv_param_->oc_up4_ * in InitComputeParam()274 …unit.weight_ = malloc(h_size * w_size * deconv_param_->ic_up4_ * deconv_param_->oc_up4_ * sizeof(f… in InitComputeParam()308 bias_data_ = malloc(deconv_param_->oc_up4_ * sizeof(float)); in InitDataParam()313 memset(bias_data_, 0, deconv_param_->oc_up4_ * sizeof(float)); in InitDataParam()434 int size = deconv_param_->oc_up4_ * deconv_param_->output_plane_; in InitRunBuf()442 DECONV_WINOGRAD_DEFAULT_TILE * deconv_param_->oc_up4_; in InitRunBuf()
126 int oc_up4_; member