diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-11-03 12:51:13 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-03 12:54:38 -0700 |
| commit | 5b166f495ae79b6e8144bbd3a1109f4b8d9fb1aa (patch) | |
| tree | bbc6246ef99693f6ab901ee7d0db0326dda041d1 /tensorflow/compiler/xla/service/hlo_computation.h | |
| parent | 3015655fa4458bbc65222929f7b8f0ae0af4dd34 (diff) | |
[TF:XLA] Improve support for const HLO visitors.
Add missing const overloads of Accept methods.
PiperOrigin-RevId: 174500495
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_computation.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_computation.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_computation.h b/tensorflow/compiler/xla/service/hlo_computation.h index f72a6e13c1..3208197f89 100644 --- a/tensorflow/compiler/xla/service/hlo_computation.h +++ b/tensorflow/compiler/xla/service/hlo_computation.h @@ -289,7 +289,8 @@ class HloComputation { // Visit every node in the computation in the given order. 'order' must // be a topological sort of all instructions in the computation. - Status AcceptOrdered(DfsHloVisitor* visitor, + template <typename HloInstructionPtr> + Status AcceptOrdered(DfsHloVisitorBase<HloInstructionPtr>* visitor, const std::vector<const HloInstruction*>& order) const; // Same as Accept() above, but the visitor is given as a function. |