未验证 提交 59cbe1a5 编写于 作者: Z zjd1988 提交者: GitHub

remove mish kernel debug codes (#409)


* remove debug codes(printf)

* fix initial var bug
上级 be1bdcde
......@@ -61,9 +61,5 @@ int mish_run(struct ir_tensor* output_tensor, struct ir_tensor* input_tensor, in
int offset = i * chan_size;
mish_kernel(0, 0, &chan_size, data + offset, out_data + offset);
}
for(int i = 0; i < 10; i++)
{
printf("%f\n", out_data[i]);
}
return 0;
}
......@@ -42,14 +42,14 @@ static int ref_psroipooling_fp32(struct ir_tensor* featmap_tensor, struct ir_ten
float* roi = roi_tensor->data;
float* output = output_tensor->data;
float spatial_scale = spatial_scale;
float spatial_scale = param->spatial_scale;
int out_h = param->pooled_h;
int out_w = param->pooled_w;
int channel = featmap_tensor->dims[1];
int in_h = featmap_tensor->dims[2];
int in_w = featmap_tensor->dims[3];
int num_rois = roi_tensor->dims[1];
int output_dim = output_dim;
int output_dim = param->output_dim;
int pool_hw = out_h * out_w;
for (int n = 0; n < num_rois; ++n)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册