diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-05-08 14:10:10 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-10 15:26:43 -0700 |
| commit | 09f3fb939c9b395a9bc747cf81d15b2dc2804c3e (patch) | |
| tree | c69f74947e38a7be1313e22f66397454f67fc849 /tensorflow/compiler/xla/service/hlo_rematerialization.cc | |
| parent | 70c303386909fe1a0d34cada6fe5a42565279849 (diff) | |
Merged commit includes the following changes:
155425029 by A. Unique TensorFlower <gardener@tensorflow.org>:
Internal change.
--
155424167 by A. Unique TensorFlower <gardener@tensorflow.org>:
Internal change.
--
PiperOrigin-RevId: 155425029
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_rematerialization.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_rematerialization.cc | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_rematerialization.cc b/tensorflow/compiler/xla/service/hlo_rematerialization.cc index b1ee2e46b0..5d4fd7c2de 100644 --- a/tensorflow/compiler/xla/service/hlo_rematerialization.cc +++ b/tensorflow/compiler/xla/service/hlo_rematerialization.cc @@ -1156,9 +1156,7 @@ StatusOr<bool> HloRematerialization::Run( VLOG(1) << "HloRematerialization() with memory limit of " << HumanReadableNumBytes(memory_limit_bytes); - TF_ASSIGN_OR_RETURN(points_to_analysis_, - TuplePointsToAnalysis::Run( - module, /*include_loop_fusion_instructions=*/true)); + TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module)); // Adjust memory limit to account for the output of the entry // computation. This is necessary because the per-computation accounting in |