|
|
|
@ -17,8 +17,14 @@
|
|
|
|
|
#include "nnacl/fp32/common_func.h"
|
|
|
|
|
void PostConvFuncComm(const float *src_ptr_, float *out_ptr, const float *bias_ptr, size_t output_channel,
|
|
|
|
|
size_t plane_size, size_t stride, bool is_relu, bool is_relu6, int size) {
|
|
|
|
|
int oc_div = 0, oc_mod = 0;
|
|
|
|
|
for (int oc = 0; oc < output_channel; oc++) {
|
|
|
|
|
int oc_div = oc / size, oc_mod = oc % size;
|
|
|
|
|
if (size != 0) {
|
|
|
|
|
oc_div = oc / size;
|
|
|
|
|
oc_mod = oc % size;
|
|
|
|
|
} else {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
for (int hw = 0; hw < plane_size; hw++) {
|
|
|
|
|
int src_index = oc_div * size * plane_size + hw * size + oc_mod;
|
|
|
|
|
int dst_index = hw * stride + oc;
|
|
|
|
|