optimize error message

This commit is contained in:
tacyi139 2022-01-13 10:29:02 +08:00
parent 54dfe5a4ae
commit a2ba74715a
1 changed files with 1 additions and 1 deletions

View File

@ -62,7 +62,7 @@ abstract::BaseShapePtr SelectInferShape(const PrimitivePtr &primitive, const std
MS_LOG(ERROR) << " cond shape :" << input_args[kCondIndex]->BuildShape()->ToString();
MS_LOG(ERROR) << " x shape :" << input_args[kXIndex]->BuildShape()->ToString();
MS_LOG(ERROR) << " y shape :" << input_args[kYIndex]->BuildShape()->ToString();
MS_EXCEPTION(ValueError) << "The x_shape is not same as y_shape and cond_shape";
MS_EXCEPTION(ValueError) << "The shape of cond, x and y should be equal";
}
return input_args[1]->BuildShape();
}