diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-02-16 13:31:04 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-16 13:34:47 -0800 |
commit | 1bde8c61e14b1c42db20a1b07684ea2534cfdf01 (patch) | |
tree | 96b49f2ed4f6dc5054c3fa47707dd295ae670dd6 /tensorflow/compiler/xla/service/heap_simulator.cc | |
parent | 6e5ca37827d95b12c4712cf237ec2f8124ed885c (diff) |
Automated g4 rollback of changelist 185623948
PiperOrigin-RevId: 186038783
Diffstat (limited to 'tensorflow/compiler/xla/service/heap_simulator.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/heap_simulator.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/heap_simulator.cc b/tensorflow/compiler/xla/service/heap_simulator.cc index cde5877e29..a2d13c013c 100644 --- a/tensorflow/compiler/xla/service/heap_simulator.cc +++ b/tensorflow/compiler/xla/service/heap_simulator.cc @@ -225,6 +225,7 @@ Status HeapSimulator::RunComputation( // sub-computations will never be run concurrently. if (module_sequence_ != nullptr) { if (instruction->opcode() == HloOpcode::kCall || + instruction->opcode() == HloOpcode::kConditional || instruction->opcode() == HloOpcode::kWhile) { for (const HloComputation* called_computation : instruction->called_computations()) { |