projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: bd72021) | patch
[TF:XLA] Add while loop invariant code motion
authorSanjoy Das <sanjoy@google.com>
Tue, 9 Jan 2018 03:41:05 +0000 (19:41 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 9 Jan 2018 03:47:25 +0000 (19:47 -0800)
commit51895fe67434b6e9f5419872f69c7e6092ed69e9
tree0e867e6ff5a62e168f3dc646d781448da208a7e6tree | snapshot
parentbd72021bd1223ac42b1010e6599b2528a18dc33ccommit | diff
[TF:XLA] Add while loop invariant code motion

This new pass extracts out loop invariant computations out of while loops into
their parent computations.

Right now this is enabled only for the CPU backend.

PiperOrigin-RevId: 181256166
19 files changed:
tensorflow/compiler/xla/map_util.h diff | blob | history
tensorflow/compiler/xla/service/BUILD diff | blob | history
tensorflow/compiler/xla/service/call_inliner.cc diff | blob | history
tensorflow/compiler/xla/service/call_inliner.h diff | blob | history
tensorflow/compiler/xla/service/call_inliner_test.cc diff | blob | history
tensorflow/compiler/xla/service/cpu/BUILD diff | blob | history
tensorflow/compiler/xla/service/cpu/cpu_compiler.cc diff | blob | history
tensorflow/compiler/xla/service/tuple_util.cc [new file with mode: 0644] blob
tensorflow/compiler/xla/service/tuple_util.h [new file with mode: 0644] blob
tensorflow/compiler/xla/service/tuple_util_test.cc [new file with mode: 0644] blob
tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc [new file with mode: 0644] blob
tensorflow/compiler/xla/service/while_loop_invariant_code_motion.h [new file with mode: 0644] blob
tensorflow/compiler/xla/service/while_loop_invariant_code_motion_test.cc [new file with mode: 0644] blob
tensorflow/compiler/xla/service/while_loop_simplifier.cc diff | blob | history
tensorflow/compiler/xla/service/while_util.cc [new file with mode: 0644] blob
tensorflow/compiler/xla/service/while_util.h [new file with mode: 0644] blob
tensorflow/compiler/xla/service/while_util_test.cc [new file with mode: 0644] blob
tensorflow/compiler/xla/tests/while_test.cc diff | blob | history
tensorflow/compiler/xla/util.h diff | blob | history
Domain: Machine Learning / ML Framework;