Commit ff6be80a authored by Yong Tang's avatar Yong Tang Committed by Rasmus Munk Larsen
Browse files

Improve the shape function for ParameterizedTruncatedNormal (#19215) 



The parameters of ParameterizedTruncatedNormal should
be 0-D or 1-D, which is checked in ther kernel functions.
There is no check in the shape function of the ops.

This fix improves the shape function and checks the
parameters of ParameterizedTruncatedNormal whever possible.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent b03008bd
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment