projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: b03008b) | patch
Improve the shape function for ParameterizedTruncatedNormal (#19215)
authorYong Tang <yong.tang.github@outlook.com>
Fri, 11 May 2018 16:48:32 +0000 (09:48 -0700)
committerRasmus Munk Larsen <rmlarsen@google.com>
Fri, 11 May 2018 16:48:32 +0000 (09:48 -0700)
commitff6be80a1ec3c353ebd0d17e2f0b46d9097310db
tree8b3944c97fa24c08292ef109abd38b0b434206e9tree | snapshot
parentb03008bdbb4dbcdecc3eb1505669e49094267b67commit | diff
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: Yong Tang <yong.tang.github@outlook.com>
tensorflow/core/ops/random_ops.cc diff | blob | history
Domain: Machine Learning / ML Framework;