Commit 9a8e0c22 authored by Benjamin Kramer's avatar Benjamin Kramer Committed by TensorFlower Gardener
Browse files

[TF:XLA] Make OneHot use Iota instead of a constant tensor 

This is simpler code and theoretically faster because it can save a transfer,
it doesn't make a difference in practice though because AlgebraicSimplifier
turns constants into iota on the HLO side.

PiperOrigin-RevId: 218901530
parent 4b2c54fa
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