未验证 提交 91f25ee3 编写于 作者: W wangguanzhong 提交者: GitHub

add dilation check for conv (#35894)

上级 95c100c1
...@@ -50,6 +50,15 @@ std::vector<int64_t> ConvOp::ComputeOutputShape( ...@@ -50,6 +50,15 @@ std::vector<int64_t> ConvOp::ComputeOutputShape(
ctx->Attrs().Get<std::string>("padding_algorithm"); ctx->Attrs().Get<std::string>("padding_algorithm");
int groups = ctx->Attrs().Get<int>("groups"); int groups = ctx->Attrs().Get<int>("groups");
std::vector<int> dilations = ctx->Attrs().Get<std::vector<int>>("dilations"); std::vector<int> dilations = ctx->Attrs().Get<std::vector<int>>("dilations");
int dilation_size = dilations.size();
for (int i = 0; i < dilation_size; ++i) {
PADDLE_ENFORCE_GT(
dilations[i], 0,
platform::errors::InvalidArgument(
"The dilation of Op(Conv) should be larget than 0, but received "
"dilation is %d.",
dilations[i]));
}
const std::string data_format = ctx->Attrs().Get<std::string>("data_format"); const std::string data_format = ctx->Attrs().Get<std::string>("data_format");
// MKL-DNN Kernels are using NCHW order of dims description // MKL-DNN Kernels are using NCHW order of dims description
......
...@@ -204,6 +204,7 @@ def add_error_cases(suite): ...@@ -204,6 +204,7 @@ def add_error_cases(suite):
suite.addTest( suite.addTest(
Conv1DErrorTestCase( Conv1DErrorTestCase(
methodName='runTest', padding=[1, 2, 3, 4, 5])) methodName='runTest', padding=[1, 2, 3, 4, 5]))
suite.addTest(Conv1DErrorTestCase(methodName='runTest', dilation=-10))
def load_tests(loader, standard_tests, pattern): def load_tests(loader, standard_tests, pattern):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册