Browse Source

fix attribute mapping when docking open source operators

tags/v0.2.0-alpha
buxue 5 years ago
parent
commit
4f8fa79f33
4 changed files with 4 additions and 4 deletions
  1. +1
    -1
      mindspore/ops/_op_impl/tbe/conv2d_backprop_filter.py
  2. +1
    -1
      mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py
  3. +1
    -1
      mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_filter.py
  4. +1
    -1
      mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_input.py

+ 1
- 1
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) \ .partial_flag(True) \
.attr("filter_sizes", "required", "listInt", "all") \ .attr("filter_sizes", "required", "listInt", "all") \
.attr("stride", "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") \ .attr("dilation", "required", "listInt", "all") \
.input(0, "out_backprop", False, "required", "all") \ .input(0, "out_backprop", False, "required", "all") \
.input(1, "x", False, "required", "all") \ .input(1, "x", False, "required", "all") \


+ 1
- 1
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) \ .partial_flag(True) \
.attr("input_sizes", "required", "listInt", "all") \ .attr("input_sizes", "required", "listInt", "all") \
.attr("stride", "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") \ .attr("dilation", "required", "listInt", "all") \
.input(0, "out_backprop", False, "required", "all") \ .input(0, "out_backprop", False, "required", "all") \
.input(1, "filter", False, "required", "all") \ .input(1, "filter", False, "required", "all") \


+ 1
- 1
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("filter_size", "required", "listInt", "all") \
.attr("stride", "required", "listInt", "all") \ .attr("stride", "required", "listInt", "all") \
.attr("dilation", "required", "listInt", "all") \ .attr("dilation", "required", "listInt", "all") \
.attr("pads", "required", "str", "all") \
.attr("pads", "required", "listInt", "all") \
.attr("data_format", "required", "str", "all") \ .attr("data_format", "required", "str", "all") \
.input(0, "input", False, "required", "all") \ .input(0, "input", False, "required", "all") \
.input(1, "out_backprop", False, "required", "all") \ .input(1, "out_backprop", False, "required", "all") \


+ 1
- 1
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("input_size", "required", "listInt", "all") \
.attr("stride", "required", "listInt", "all") \ .attr("stride", "required", "listInt", "all") \
.attr("dilation", "required", "listInt", "all") \ .attr("dilation", "required", "listInt", "all") \
.attr("pads", "required", "str", "all") \
.attr("pads", "required", "listInt", "all") \
.attr("data_format", "required", "str", "all") \ .attr("data_format", "required", "str", "all") \
.input(0, "filter", False, "required", "all") \ .input(0, "filter", False, "required", "all") \
.input(1, "out_backprop", False, "required", "all") \ .input(1, "out_backprop", False, "required", "all") \


Loading…
Cancel
Save