Commit 7fc61bfb authored by Joshua V. Dillon's avatar Joshua V. Dillon Committed by TensorFlower Gardener
Browse files

Change `reduce_logsumexp` to internally use `reshape` rather than `squeeze` 

since the latter requires the `axis` arg to be a Python `list`.

PiperOrigin-RevId: 183396533
parent abdc62ae
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