未验证 提交 12755539 编写于 作者: K kalcohol 提交者: GitHub

fix dump tensor function bug (#662)

上级 74e1ebd0
......@@ -172,7 +172,7 @@ static int run(struct device* dev, struct subgraph* subgraph)
if (NULL != input_tensor->data)
{
extract_feature_blob_f32(dir_str, name, input_tensor);
extract_feature_from_tensor(dir_str, name, input_tensor);
}
}
}
......@@ -186,7 +186,7 @@ static int run(struct device* dev, struct subgraph* subgraph)
char dir_str[32] = { 0 };
sprintf(dir_str, "out[%d]", j);
extract_feature_blob_f32(dir_str, name, output_tensor);
extract_feature_from_tensor(dir_str, name, output_tensor);
}
}
#endif
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册