p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix attribute mapping when docking open source operators

This commit is contained in:
buxue 2020-04-20 20:52:05 +08:00
parent 5434a96d85
commit 4f8fa79f33
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ops/_op_impl/tbe/conv2d_backprop_filter.py
View File

@ -25,7 +25,7 @@ conv2d_backprop_filter_op_info = TBERegOp("Conv2DBackpropFilter") \
.partial_flag(True) \
.attr("filter_sizes", "required", "listInt", "all") \
.attr("stride", "required", "listInt", "all") \
.attr("pad_mode", "required", "str", "all") \
.attr("pad_list", "required", "listInt", "all") \
.attr("dilation", "required", "listInt", "all") \
.input(0, "out_backprop", False, "required", "all") \
.input(1, "x", False, "required", "all") \

2
mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py
View File

@ -25,7 +25,7 @@ conv2d_backprop_input_op_info = TBERegOp("Conv2DBackpropInput") \
.partial_flag(True) \
.attr("input_sizes", "required", "listInt", "all") \
.attr("stride", "required", "listInt", "all") \
.attr("pad_mode", "required", "str", "all") \
.attr("pad_list", "required", "listInt", "all") \
.attr("dilation", "required", "listInt", "all") \
.input(0, "out_backprop", False, "required", "all") \
.input(1, "filter", False, "required", "all") \

2
mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_filter.py
View File

@ -26,7 +26,7 @@ depthwise_conv2d_backprop_filter_op_info = TBERegOp("DepthwiseConv2dNativeBackpr
.attr("filter_size", "required", "listInt", "all") \
.attr("stride", "required", "listInt", "all") \
.attr("dilation", "required", "listInt", "all") \
.attr("pads", "required", "str", "all") \
.attr("pads", "required", "listInt", "all") \
.attr("data_format", "required", "str", "all") \
.input(0, "input", False, "required", "all") \
.input(1, "out_backprop", False, "required", "all") \

2
mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_input.py
View File

@ -26,7 +26,7 @@ depthwise_conv2d_backprop_input_op_info = TBERegOp("DepthwiseConv2dNativeBackpro
.attr("input_size", "required", "listInt", "all") \
.attr("stride", "required", "listInt", "all") \
.attr("dilation", "required", "listInt", "all") \
.attr("pads", "required", "str", "all") \
.attr("pads", "required", "listInt", "all") \
.attr("data_format", "required", "str", "all") \
.input(0, "filter", False, "required", "all") \
.input(1, "out_backprop", False, "required", "all") \