diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-02-13 15:29:00 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-13 15:32:59 -0800 |
commit | 9a0403542b3c67119e7097a491a2c5e2602b14c3 (patch) | |
tree | 5d9dfbbed2c388d8a0ac8c22fa3d899958c5562e /tensorflow/compiler/xla/service/heap_simulator.cc | |
parent | dcfa59b8c32e44bcd16fbd58ce91e7d3f332be36 (diff) |
Wire in support for XLA kConditional instruction.
PiperOrigin-RevId: 185598764
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()) { |