diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/dfs_hlo_visitor.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/dfs_hlo_visitor.h | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/compiler/xla/service/dfs_hlo_visitor.h b/tensorflow/compiler/xla/service/dfs_hlo_visitor.h index bc73839a88..de3cd15440 100644 --- a/tensorflow/compiler/xla/service/dfs_hlo_visitor.h +++ b/tensorflow/compiler/xla/service/dfs_hlo_visitor.h @@ -211,11 +211,9 @@ class DfsHloVisitorBase { virtual Status HandlePad(HloInstructionPtr hlo) = 0; - virtual Status HandleSend(HloInstructionPtr send) = 0; - virtual Status HandleSendDone(HloInstructionPtr send_done) = 0; + virtual Status HandleSend(HloInstructionPtr hlo) = 0; - virtual Status HandleRecv(HloInstructionPtr recv) = 0; - virtual Status HandleRecvDone(HloInstructionPtr recv_done) = 0; + virtual Status HandleRecv(HloInstructionPtr hlo) = 0; virtual Status HandleBatchNormTraining(HloInstructionPtr hlo) = 0; |