未验证 提交 a24726f3 编写于 作者: B BUG1989 提交者: GitHub

fix bug of softmax ref fp32 (#401)

上级 0dbf0497
......@@ -27,7 +27,7 @@ void init(const char* modelfile)
graph = create_graph(NULL, "tengine", modelfile);
if (graph == NULL)
{
fprintf(stderr, "grph nullptr %\n", get_tengine_errno());
fprintf(stderr, "grph nullptr %d\n", get_tengine_errno());
}
else
{
......
......@@ -158,7 +158,7 @@ static int run(struct node_ops* node_ops, struct exec_node* exec_node, struct ex
for (int i = 0; i < out_size; i++)
{
/* get max */
int img_base = i * on_in_size * element_size;
int img_base = i * on_in_size;
GetMaxArray(input + img_base, max_array, in_size, on_size, exec_graph->num_thread);
GetOutResult(input + img_base, output + img_base, max_array, sum_array, in_size, on_size,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册