Commit 2ee5ee30 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Special case for one sample categorical so that TPU memory padding doesn't... 

Special case for one sample categorical so that TPU memory padding doesn't increase the number of samples dim.

PiperOrigin-RevId: 218202061
parent 8f7b69de
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