未验证 提交 d53e3f1e 编写于 作者: Z Zhuo Zhang 提交者: GitHub

fix crop layer bug (#378)

上级 ed977da3
......@@ -38,7 +38,7 @@ DEFINE_PARM_PARSE_ENTRY(crop_param, flag, num_args, offset_c, offset_h, offset_w
static int infer_shape(struct ir_node* node)
{
struct ir_graph* graph = node->graph;
struct ir_tensor* input = get_ir_graph_tensor(graph, node->input_tensors[1]);
struct ir_tensor* input = get_ir_graph_tensor(graph, node->input_tensors[0]);
struct ir_tensor* output = get_ir_graph_tensor(graph, node->output_tensors[0]);
struct crop_param* crop_param = ( struct crop_param* )(node->op.param_mem);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册