diff options
author | 2017-11-03 12:51:13 -0700 | |
---|---|---|
committer | 2017-11-03 12:54:38 -0700 | |
commit | 5b166f495ae79b6e8144bbd3a1109f4b8d9fb1aa (patch) | |
tree | bbc6246ef99693f6ab901ee7d0db0326dda041d1 /tensorflow/compiler/xla/service/hlo_computation.cc | |
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.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_computation.cc | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_computation.cc b/tensorflow/compiler/xla/service/hlo_computation.cc index a9c7fdc4e5..1677c77f2e 100644 --- a/tensorflow/compiler/xla/service/hlo_computation.cc +++ b/tensorflow/compiler/xla/service/hlo_computation.cc @@ -697,8 +697,9 @@ Status HloComputation::AcceptWithOperandOrder( /*call_finish_visit=*/true); } +template <typename HloInstructionPtr> Status HloComputation::AcceptOrdered( - DfsHloVisitor* visitor, + DfsHloVisitorBase<HloInstructionPtr>* visitor, const std::vector<const HloInstruction*>& order) const { VLOG(3) << "Accepting visitor with order."; for (HloInstruction* root : CollectUnreachableRoots()) { @@ -727,6 +728,12 @@ Status HloComputation::AcceptOrdered( return Status::OK(); } +// Explicit instantiations. +template Status HloComputation::AcceptOrdered( + DfsHloVisitor*, const std::vector<const HloInstruction*>&) const; +template Status HloComputation::AcceptOrdered( + ConstDfsHloVisitor*, const std::vector<const HloInstruction*>&) const; + Status HloComputation::Accept( const std::function<Status(HloInstruction*)>& visitor_func) { FunctionVisitor visitor(visitor_func); |