Commit 787c2a68 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[XLA] Fix bug in TriangularSolve where both batch dimensions and more than one block were present. 

Fixes https://github.com/google/jax/issues/128 (or will do so when a new XLA release is integrated into JAX.)

PiperOrigin-RevId: 226034659
parent 4817dabb
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