Fix: add free

pull/60/head
xiangbingj 2019-10-26 15:49:37 +08:00
parent 61af37d08c
commit 60852bdbcb
1 changed files with 2 additions and 0 deletions

View File

@ -751,6 +751,8 @@ private:
sum += src[ic] * c_weights[ic];
dest[oc] = sum + bias[oc];
}
free(weights);
free(bias);
}
void kpu_tf_flatten(const kpu_model_tf_flatten_layer_argument_t *arg)