diff options
author | 2017-10-30 18:58:48 -0700 | |
---|---|---|
committer | 2017-10-30 19:02:27 -0700 | |
commit | a4b5356e476016e0f537766ac2ac891eab9900e1 (patch) | |
tree | ac16c36367a3cb7a6c2c570d051886d071357a7d /tensorflow/compiler/xla/service/tuple_points_to_analysis.h | |
parent | bb7ed1c889890ca68f531a3a7b4f56fc55b082df (diff) |
[TF:XLA] Reduce boilerplate code in HLO visitors.
Only pass the HloInstruction into visitor methods. This makes changing
instructions and visitors easier.
PiperOrigin-RevId: 173983398
Diffstat (limited to 'tensorflow/compiler/xla/service/tuple_points_to_analysis.h')
-rw-r--r-- | tensorflow/compiler/xla/service/tuple_points_to_analysis.h | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/tensorflow/compiler/xla/service/tuple_points_to_analysis.h b/tensorflow/compiler/xla/service/tuple_points_to_analysis.h index 30dabb56bd..e6157a1ed1 100644 --- a/tensorflow/compiler/xla/service/tuple_points_to_analysis.h +++ b/tensorflow/compiler/xla/service/tuple_points_to_analysis.h @@ -247,16 +247,11 @@ class TuplePointsToAnalysis : public DfsHloVisitorWithDefault { Status VerifyBuffer(const LogicalBuffer& buffer) const; Status DefaultAction(HloInstruction* hlo_instruction) override; - Status HandleTuple( - HloInstruction* tuple, - tensorflow::gtl::ArraySlice<HloInstruction*> operands) override; - Status HandleGetTupleElement(HloInstruction* get_tuple_element, - HloInstruction* operand) override; + Status HandleTuple(HloInstruction* tuple) override; + Status HandleGetTupleElement(HloInstruction* get_tuple_element) override; Status HandleBitcast(HloInstruction* bitcast) override; Status HandleCopy(HloInstruction* copy) override; - Status HandleSelect(HloInstruction* select, HloInstruction* pred, - HloInstruction* on_true, - HloInstruction* on_false) override; + Status HandleSelect(HloInstruction* select) override; string ToString() const; |