diff options
Diffstat (limited to 'tensorflow/compiler/xla/client/computation_builder.cc')
| -rw-r--r-- | tensorflow/compiler/xla/client/computation_builder.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/client/computation_builder.cc b/tensorflow/compiler/xla/client/computation_builder.cc index ae7695ade5..2ee6b51c18 100644 --- a/tensorflow/compiler/xla/client/computation_builder.cc +++ b/tensorflow/compiler/xla/client/computation_builder.cc @@ -1010,8 +1010,9 @@ ComputationDataHandle ComputationBuilder::SqrtF32( } ComputationDataHandle ComputationBuilder::Pow( - const ComputationDataHandle& lhs, const ComputationDataHandle& rhs) { - return BinaryOp(BINOP_POW, lhs, rhs, /*broadcast_dimensions=*/{}); + const ComputationDataHandle& lhs, const ComputationDataHandle& rhs, + tensorflow::gtl::ArraySlice<int64> broadcast_dimensions) { + return BinaryOp(BINOP_POW, lhs, rhs, broadcast_dimensions); } ComputationDataHandle ComputationBuilder::ConvertElementType( |