Commit a3ccedb3 authored by Andrew Selle's avatar Andrew Selle Committed by TensorFlower Gardener
Browse files

Change softmax_cross_entropy_with_logits_v2 to use axis instead of dim 

PiperOrigin-RevId: 223600271
parent 4f3d18dc
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