From 21464f94010d1dfa5a43c4de17f1ee661aa0b99b Mon Sep 17 00:00:00 2001 From: buxue Date: Mon, 18 May 2020 16:27:46 +0800 Subject: [PATCH] add attr for Conv2dBackpropInput --- mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py b/mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py index 7756cb3ae..cba8f1e98 100644 --- a/mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py +++ b/mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py @@ -27,6 +27,8 @@ conv2d_backprop_input_op_info = TBERegOp("Conv2DBackpropInput") \ .attr("stride", "required", "listInt", "all") \ .attr("pad_list", "required", "listInt", "all") \ .attr("dilation", "required", "listInt", "all") \ + .attr("group", "optional", "int", "all") \ + .attr("data_format", "optional", "str", "all") \ .input(0, "out_backprop", False, "required", "all") \ .input(1, "filter", False, "required", "all") \ .output(0, "y", True, "required", "all") \ -- GitLab