aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/hlo_parser.cc
diff options
context:
space:
mode:
authorGravatar Nick Desaulniers <ndesaulniers@google.com>2018-07-20 13:50:37 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-07-20 13:53:55 -0700
commite542062aa1613dc01b82b6378675563160fe0abf (patch)
tree15ce2b045f53f2f610a563e8d0091d96d2db2f4c /tensorflow/compiler/xla/service/hlo_parser.cc
parent265292420de30f24805d28886d403dc42d3685b3 (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.cc8
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: