diff options
author | Nick Desaulniers <ndesaulniers@google.com> | 2018-07-20 13:50:37 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-20 13:53:55 -0700 |
commit | e542062aa1613dc01b82b6378675563160fe0abf (patch) | |
tree | 15ce2b045f53f2f610a563e8d0091d96d2db2f4c /tensorflow/compiler/xla/service/cpu/ir_emitter.h | |
parent | 265292420de30f24805d28886d403dc42d3685b3 (diff) |
Start implementation of Iota HLO.
PiperOrigin-RevId: 205447892
Diffstat (limited to 'tensorflow/compiler/xla/service/cpu/ir_emitter.h')
-rw-r--r-- | tensorflow/compiler/xla/service/cpu/ir_emitter.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/cpu/ir_emitter.h b/tensorflow/compiler/xla/service/cpu/ir_emitter.h index 419f19c24d..2840c14303 100644 --- a/tensorflow/compiler/xla/service/cpu/ir_emitter.h +++ b/tensorflow/compiler/xla/service/cpu/ir_emitter.h @@ -148,6 +148,7 @@ class IrEmitter : public DfsHloVisitorWithDefault { Status HandleConcatenate(HloInstruction* concatenate) override; Status HandleConditional(HloInstruction* conditional) override; Status HandleAfterAll(HloInstruction* gen_token) override; + Status HandleIota(HloInstruction* iota) override; Status FinishVisit(HloInstruction* root) override; Status Preprocess(HloInstruction* hlo) override; |