Created by: pangyoki

PR types

Others

PR changes

APIs

Describe

_check_values_dtypes_in_probs(param, value) method in Distribution class is used to ensure dtype of param and value are consistent. (refer to PR #26767) If in_dygraph_mode and value.dtype == param.dtype, it doesn't have return value in dygraph mode. And it has to run static mode again. This PR adds return value in dygraph mode.