projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 8aa14cd) | patch
Remove warnings in tf.losses.softmax_cross_entropy
authorYong Tang <yong.tang.github@outlook.com>
Wed, 7 Feb 2018 16:24:34 +0000 (16:24 +0000)
committerYong Tang <yong.tang.github@outlook.com>
Wed, 7 Feb 2018 16:24:34 +0000 (16:24 +0000)
commita7c225c89e4169492f0eef57c913463b976dd44e
tree74f00a2b8af27241e583186ac5fd2639cbf399d6tree | snapshot
parent8aa14cd682053e1e643f0a74ec25cf3b87bf2712commit | diff
Remove warnings in tf.losses.softmax_cross_entropy

This fix tries to address the issue raised in 16534 where
tf.losses.softmax_cross_entropy causes warnings due
to the calling of tf.nn.softmax_cross_entropy_with_logits.

This fix switches to tf.nn.softmax_cross_entropy_with_logits_v2
to remove the warning.

This fix fixes 16534.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
tensorflow/python/ops/losses/losses_impl.py diff | blob | history
Domain: Machine Learning / ML Framework;