未验证 提交 e46ffaf2 编写于 作者: W wangxinxin08 提交者: GitHub

fix conv op check (#35693)

上级 f0661a1d
...@@ -116,10 +116,13 @@ std::vector<int64_t> ConvOp::ComputeOutputShape( ...@@ -116,10 +116,13 @@ std::vector<int64_t> ConvOp::ComputeOutputShape(
"the output channels is %d, the filter's shape is [%s], " "the output channels is %d, the filter's shape is [%s], "
"the groups is %d.", "the groups is %d.",
filter_dims[0], filter_dims, groups)); filter_dims[0], filter_dims, groups));
if (ctx->IsRuntime()) {
PADDLE_ENFORCE_GT( PADDLE_ENFORCE_GT(
filter_dims[0], 0, filter_dims[0], 0,
platform::errors::InvalidArgument( platform::errors::InvalidArgument(
"the size of filter at axis 0 should be greater than 0")); "the size of filter at axis 0 should be greater than 0"));
}
framework::DDim in_data_dims; framework::DDim in_data_dims;
if (channel_last) { if (channel_last) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册