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

trying fix deconv bias tensor bug (#457)

* fix deconv ref bias_tensor bug
上级 12637ec1
......@@ -282,14 +282,16 @@ static int run(struct node_ops* node_ops, struct exec_node* exec_node, struct ex
struct ir_graph* ir_graph = ir_node->graph;
struct ir_tensor* i_tensor = get_ir_graph_tensor(ir_graph, ir_node->input_tensors[0]);
struct ir_tensor* weight_tensor = get_ir_graph_tensor(ir_graph, ir_node->input_tensors[1]);
struct ir_tensor* bias_tensor = get_ir_graph_tensor(ir_graph, ir_node->input_tensors[2]);
struct ir_tensor* bias_tensor = NULL;
struct ir_tensor* output_tensor = get_ir_graph_tensor(ir_graph, ir_node->output_tensors[0]);
void* output_data = output_tensor->data;
const void* input_data = i_tensor->data;
const void* kernel = weight_tensor->data;
const void* bias = NULL;
if (ir_node->input_num > 2)
{
bias_tensor = get_ir_graph_tensor(ir_graph, ir_node->input_tensors[2]);
}
if (bias_tensor != NULL)
bias = bias_tensor->data;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册