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

[XLA:CPU] Implement Ax+b dot output fusion for Matrix-vector products 

I had to roll in the change to generalize CPU layout assignment as without it we
lose the make-rhs-column-major optimization and that causes a performance
regression.

PiperOrigin-RevId: 178970986
parent 8a9aaa3b
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