Commit 3a2276ce authored by Benjamin Kramer's avatar Benjamin Kramer Committed by TensorFlower Gardener
Browse files

[XLA:TF] Make FloorDiv not crash on unsigned types 

FloorDiv (which corresponds to the // operator in python) supports uint8 and
uint16 (but not uint32) in TF. Using xla::Abs on unsigned types throws an error,
but the rounding logic is trivial for unsigned types so just do a plain Div.

This isn't tested yet because we don't have any targets supporting uint8 or
uint16 yet.

PiperOrigin-RevId: 212946132
parent b43aeb05
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