Commit 07e6ca0a authored by Yong Tang's avatar Yong Tang Committed by Benoit Steiner
Browse files

Fix crash on GPU (out of GPU memory) for `softmax_cross_entropy_with_logits` (#16051) 

* Fix crash on GPU (out of GPU memory) for `softmax_cross_entropy_with_logits`

This fix tries to address the issue raised in 6766 where
`softmax_cross_entropy_with_logits` will trigger the crash on GPU (out of GPU memory)
if the first dimension is 0.

This fix fixes 6766.
parent f9462e82
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