Commit 96de2f02 authored by David Majnemer's avatar David Majnemer Committed by TensorFlower Gardener
Browse files

[XLA] Implement kIota for CPU & GPU, extend it w/ broadcast semantics 

This extends the Iota HLO to have a broadcast field. This allows for higher
rank kIota operations.

PiperOrigin-RevId: 210600435
parent 9c1f1432
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