tree 7c97f7aced95d18e1bfab249b4c79242c1036e7b
parent b6f67aff39ed0a751e9a58e45425f1c894c7e412
author Aden Grue <agrue@google.com> 1633727249 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1633727508 -0700

Roll forward https://github.com/tensorflow/tensorflow/commit/04ba7f6241e50d264730031e4c1dc9c3598e88fe

In addition to rolling forward the commit, this adds new logic to elide the custom call failure checks after calling XLA computations that can't possibly fail (because they don't transitively call any custom call functions). This was necessary because in many cases the extra checks negatively affected the numerical precision of the result.

PiperOrigin-RevId: 401852062
Change-Id: I14f7dfc9d8cae990d77c61ec8b245bbf72754ba8