Skip to content

Commit a4eb237

Browse files
add args check for crop (#59967) (#60005)
1 parent 56eaf0e commit a4eb237

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

paddle/phi/infermeta/unary.cc

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -664,6 +664,16 @@ void CropInferMeta(const MetaTensor& x,
664664
shape_dims.size(),
665665
x_dim.size()));
666666

667+
PADDLE_ENFORCE_EQ(
668+
offsets_vec.size(),
669+
x_dim.size(),
670+
errors::InvalidArgument(
671+
"The number of elements (%d) of attribute 'offsets' for "
672+
"CropTensor must be equal to the number of "
673+
"dimensions (%d) of the input.",
674+
offsets_vec.size(),
675+
x_dim.size()));
676+
667677
if (config.is_runtime) {
668678
out->share_lod(x);
669679
}

0 commit comments

Comments
 (0)