diff options
| author |  Sanjoy Das <sanjoy@google.com> | 2018-09-07 18:47:56 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-07 18:53:18 -0700 |
| commit | 4fd48f57cd1dcd960bea1757e1c59032db66b3d0 (patch) | |
| tree | 49ef65ef479a2904ff08d5ed9bfb65bac08a64f4 /third_party | |
| parent | 3e1b06ee93d7a638db1fdd5f733d66064c1acf59 (diff) | |
Decluster some must-be-constant ops to reduce XLA recompilations
The CL is organized as follows:
- The main change is in jit/partially_decluster_pass.
- tf2xla/const_analysis now takes an "edge_filter" to facilitate use by
jit/partially_decluster_pass.
- tests/dense_layer_test.py was using the execution of ListDiff as what I
assume is a sanity check to see that the XLA cluster ran. With this CL the
ListDiff op gets declustered so we now check for "MatMult" for the sanity
check.
- Some tests were dropping TF_XLA_FLAGS; fixed them to not do so.
PiperOrigin-RevId: 212071118
Diffstat (limited to 'third_party')
0 files changed, 0 insertions, 0 deletions