|
|
|
|
@ -52,19 +52,22 @@ int GroupConvolutionFP16CPUKernel::ReSize() {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void GroupConvolutionFP16CPUKernel::FreeSubKernel() {
|
|
|
|
|
for (auto sub_conv : group_convs_) {
|
|
|
|
|
for (auto &sub_conv : group_convs_) {
|
|
|
|
|
// free sub conv input tensors / output tensors manually
|
|
|
|
|
auto sub_in_tensors = sub_conv->in_tensors();
|
|
|
|
|
auto sub_in_tensor_num = sub_in_tensors.size();
|
|
|
|
|
for (size_t i = 0; i < sub_in_tensor_num; ++i) {
|
|
|
|
|
delete sub_in_tensors[i];
|
|
|
|
|
sub_in_tensors[i] = nullptr;
|
|
|
|
|
}
|
|
|
|
|
auto sub_out_tensors = sub_conv->out_tensors();
|
|
|
|
|
auto sub_out_tensor_num = sub_out_tensors.size();
|
|
|
|
|
for (size_t i = 0; i < sub_out_tensor_num; ++i) {
|
|
|
|
|
delete sub_out_tensors[i];
|
|
|
|
|
sub_out_tensors[i] = nullptr;
|
|
|
|
|
}
|
|
|
|
|
delete sub_conv;
|
|
|
|
|
sub_conv = nullptr;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|