Commit 53805cfc authored by Blake Hechtman's avatar Blake Hechtman Committed by TensorFlower Gardener
Browse files

[XLA] Strength reduce Dots with batch dimensions if there is a single... 

[XLA] Strength reduce Dots with batch dimensions if there is a single contracting and non contracting dimension in the lhs and rhs, the batch dimension numbers are equal and one contracting or non-contracting dimension is 1. This is a generalization of rank 2 dot strength reduction.

PiperOrigin-RevId: 225483537
parent d1fc8779
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