Commit 0909b15d authored by Justin Lebar's avatar Justin Lebar Committed by TensorFlower Gardener
Browse files

[XLA] Speed up exhaustive_f32_elementwise_op_test by using our own...

[XLA] Speed up exhaustive_f32_elementwise_op_test by using our own implementation of LiteralTestUtil::Near.

LiteralTestUtil::Near does a bunch of stuff that we don't care about, like
keeping track of mean/stddev.  It also operates on literals not 1D arrays, and
so does expensive index calculations on each entry.

Rather than laboriously try to fix it up, it's easy enough just to reeimplement
the relevant part.

Speedup is ~8/5 = 1.6x.

PiperOrigin-RevId: 231305563
parent 0013f89a
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment