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/hlo_parser.cc | |
parent | 265292420de30f24805d28886d403dc42d3685b3 (diff) |
Start implementation of Iota HLO.
PiperOrigin-RevId: 205447892
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_parser.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_parser.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_parser.cc b/tensorflow/compiler/xla/service/hlo_parser.cc index 496eca0739..e8eaf54949 100644 --- a/tensorflow/compiler/xla/service/hlo_parser.cc +++ b/tensorflow/compiler/xla/service/hlo_parser.cc @@ -492,6 +492,14 @@ bool HloParser::ParseInstruction(HloComputation::Builder* builder, HloInstruction::CreateConstant(std::move(literal))); break; } + case HloOpcode::kIota: { + if (!ParseOperands(&operands, /*expected_size=*/0) || + !ParseAttributes(attrs)) { + return false; + } + instruction = builder->AddInstruction(HloInstruction::CreateIota(shape)); + break; + } // Unary ops. case HloOpcode::kAbs: case HloOpcode::kRoundNearestAfz: |