Searched refs:winograd_ (Results 1 – 5 of 5) sorted by relevance
88 unit->winograd_.AT_ = malloc(unit->winograd_.i_ * unit->winograd_.o_ * sizeof(float)); in PackDeConvWgDataFp32()89 if (unit->winograd_.AT_ == NULL) { in PackDeConvWgDataFp32()96 memcpy(unit->winograd_.AT_, matrix_at, unit->winograd_.i_ * unit->winograd_.o_ * sizeof(float)); in PackDeConvWgDataFp32()99 unit->winograd_.BT_ = malloc(unit->winograd_.o_ * unit->winograd_.o_ * sizeof(float)); in PackDeConvWgDataFp32()100 if (unit->winograd_.BT_ == NULL) { in PackDeConvWgDataFp32()105 if (unit->winograd_.AT_ != NULL) { in PackDeConvWgDataFp32()106 free(unit->winograd_.AT_); in PackDeConvWgDataFp32()107 unit->winograd_.AT_ = NULL; in PackDeConvWgDataFp32()111 memcpy(unit->winograd_.BT_, matrix_bt, unit->winograd_.o_ * unit->winograd_.o_ * sizeof(float)); in PackDeConvWgDataFp32()114 …nv_param->input_channel_ * conv_param->output_channel_ * unit->winograd_.kh_ * unit->winograd_.kw_; in PackDeConvWgDataFp32()[all …]
44 if (unit.winograd_.b_buffer_ != nullptr) { in FreeResizeBuf()45 free(unit.winograd_.b_buffer_); in FreeResizeBuf()46 unit.winograd_.b_buffer_ = nullptr; in FreeResizeBuf()84 if (unit.winograd_.AT_ != nullptr) { in FreeDeconvParam()85 free(unit.winograd_.AT_); in FreeDeconvParam()86 unit.winograd_.AT_ = nullptr; in FreeDeconvParam()88 if (unit.winograd_.BT_ != nullptr) { in FreeDeconvParam()89 free(unit.winograd_.BT_); in FreeDeconvParam()90 unit.winograd_.BT_ = nullptr; in FreeDeconvParam()136 if (unit.winograd_.kh_ >= DECONV_WINOGRAD_BUFFER_COUNT) { in InitParameter()[all …]
41 if (unit.winograd_.b_buffer_ != nullptr) { in FreeResizeBuf()42 free(unit.winograd_.b_buffer_); in FreeResizeBuf()43 unit.winograd_.b_buffer_ = nullptr; in FreeResizeBuf()91 if (unit.winograd_.AT_ != nullptr) { in FreeDeconvParam()92 free(unit.winograd_.AT_); in FreeDeconvParam()93 unit.winograd_.AT_ = nullptr; in FreeDeconvParam()95 if (unit.winograd_.BT_ != nullptr) { in FreeDeconvParam()96 free(unit.winograd_.BT_); in FreeDeconvParam()97 unit.winograd_.BT_ = nullptr; in FreeDeconvParam()154 if (unit.winograd_.kh_ >= DECONV_WINOGRAD_BUFFER_COUNT) { in InitParameter()[all …]
281 unit->winograd_.AT_ = malloc(unit->winograd_.i_ * unit->winograd_.o_ * sizeof(float16_t)); in PackDeConvWgDataFp16()282 if (unit->winograd_.AT_ == NULL) { in PackDeConvWgDataFp16()287 Float32ToFloat16(matrix_at, unit->winograd_.AT_, unit->winograd_.i_ * unit->winograd_.o_); in PackDeConvWgDataFp16()290 unit->winograd_.BT_ = malloc(unit->winograd_.o_ * unit->winograd_.o_ * sizeof(float16_t)); in PackDeConvWgDataFp16()291 if (unit->winograd_.BT_ == NULL) { in PackDeConvWgDataFp16()293 free(unit->winograd_.AT_); in PackDeConvWgDataFp16()295 unit->winograd_.AT_ = NULL; in PackDeConvWgDataFp16()298 Float32ToFloat16(matrix_bt, unit->winograd_.BT_, unit->winograd_.o_ * unit->winograd_.o_); in PackDeConvWgDataFp16()301 size = conv_param->input_channel_ * output_channel * unit->winograd_.kh_ * unit->winograd_.kw_; in PackDeConvWgDataFp16()305 free(unit->winograd_.AT_); in PackDeConvWgDataFp16()[all …]
113 DeConvWg winograd_; member