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

Fix, the bug of dynamic of sigmod (#606)

上级 df209b59
......@@ -108,6 +108,24 @@ static int release_node(struct node_ops* node_ops, struct exec_node* exec_node,
return 0;
}
static int reshape_node(struct node_ops* node_ops, struct exec_node* exec_node, struct exec_graph* exec_graph)
{
struct ir_node* ir_node = exec_node->ir_node;
struct ir_graph* ir_graph = ir_node->graph;
struct ir_tensor* input_tensor;
struct ir_tensor* output_tensor;
int ret = 0;
input_tensor = get_ir_graph_tensor(ir_graph, ir_node->input_tensors[0]);
output_tensor = get_ir_graph_tensor(ir_graph, ir_node->output_tensors[0]);
if (input_tensor->dims[1] != output_tensor->dims[1] || input_tensor->dims[2] != output_tensor->dims[2] ||
input_tensor->dims[3] != output_tensor->dims[3])
ret = set_ir_tensor_shape(output_tensor, input_tensor->dims, input_tensor->dim_num);
return ret;
}
static int prerun(struct node_ops* node_ops, struct exec_node* exec_node, struct exec_graph* exec_graph)
{
return 0;
......@@ -139,7 +157,7 @@ static int score(struct node_ops* node_ops, struct exec_graph* exec_graph, struc
static struct node_ops sigmoid_node_ops = {.prerun = prerun,
.run = run,
.reshape = NULL,
.reshape = reshape_node,
.postrun = NULL,
.init_node = init_node,
.release_node = release_node,
......
......@@ -52,7 +52,6 @@ static int init_op(struct ir_op* op)
static void release_op(struct ir_op* op)
{
// sys_free(op->param_mem);
}
static int register_sigmoid_op(void* arg)
......@@ -62,14 +61,12 @@ static int register_sigmoid_op(void* arg)
m.op_version = 1;
m.init_op = init_op;
m.release_op = release_op;
// m.access_param_entry = access_param_entry;
return register_op(OP_SIGMOID, OP_SIGMOID_NAME, &m);
}
static int unregister_sigmoid_op(void* arg)
{
// sys_free(GET_PARAM_PARSE_MAP(sigmoid_param));
return unregister_op(OP_SIGMOID, 1);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册