Skip to content

Commit

Permalink
[TF:XLA] Fix open source build.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 160714466
  • Loading branch information
kayzhu authored and tensorflower-gardener committed Jul 1, 2017
1 parent d34eb19 commit c805dc1
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tensorflow/compiler/xla/service/hlo_evaluator.cc
Original file line number Diff line number Diff line change
Expand Up @@ -646,8 +646,9 @@ class HloEvaluator::TypedVisitor : public DfsHloVisitorWithDefault {
0);
std::vector<int64> step(evaluated_operand.shape().dimensions_size(), 1);

ShapeUtil::ForEachIndex(evaluated_operand.shape(), zero_base,
evaluated_operand.shape().dimensions(), step, func);
ShapeUtil::ForEachIndex(
evaluated_operand.shape(), zero_base,
AsInt64Slice(evaluated_operand.shape().dimensions()), step, func);

parent_->evaluated_[pad] = std::move(result);
return Status::OK();
Expand Down

0 comments on commit c805dc1

Please sign in to comment.