From 4f8fa79f33f1db91a1d68db0b8c3cf5cf456ac26 Mon Sep 17 00:00:00 2001 From: buxue Date: Mon, 20 Apr 2020 20:52:05 +0800 Subject: [PATCH] fix attribute mapping when docking open source operators --- mindspore/ops/_op_impl/tbe/conv2d_backprop_filter.py | 2 +- mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py | 2 +- mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_filter.py | 2 +- mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_input.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mindspore/ops/_op_impl/tbe/conv2d_backprop_filter.py b/mindspore/ops/_op_impl/tbe/conv2d_backprop_filter.py index e32e99d88..04b55bb2a 100644 --- a/mindspore/ops/_op_impl/tbe/conv2d_backprop_filter.py +++ b/mindspore/ops/_op_impl/tbe/conv2d_backprop_filter.py @@ -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") \ diff --git a/mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py b/mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py index 2c1dd6aea..7756cb3ae 100644 --- a/mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py +++ b/mindspore/ops/_op_impl/tbe/conv2d_backprop_input.py @@ -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") \ diff --git a/mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_filter.py b/mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_filter.py index c19a31100..f4d8069b1 100644 --- a/mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_filter.py +++ b/mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_filter.py @@ -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") \ diff --git a/mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_input.py b/mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_input.py index 9e671f18e..61c1406b3 100644 --- a/mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_input.py +++ b/mindspore/ops/_op_impl/tbe/depthwise_conv2d_backprop_input.py @@ -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") \ -- GitLab