提交 4f8fa79f 编写于 作者: B buxue

fix attribute mapping when docking open source operators

上级 5434a96d
......@@ -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") \
......
......@@ -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") \
......
......@@ -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") \
......
......@@ -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") \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册