Commit e76a7a7c authored by Sanjoy Das's avatar Sanjoy Das Committed by TensorFlower Gardener
Browse files

[TF:XLA:CPU] Fix an issue with explicitly vectorized reductions 

The explicitly vectorized reduction kernel assumes that the input and the output
layouts are "same" (not identical which is not possible, but have the same
ordering of the un-reduced dimensions).  This change makes IrEmitter check this
precondition.

PiperOrigin-RevId: 183158249
parent ffa63e57
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