diff options
Diffstat (limited to 'tensorflow/compiler/xla/client/computation_builder.h')
-rw-r--r-- | tensorflow/compiler/xla/client/computation_builder.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/client/computation_builder.h b/tensorflow/compiler/xla/client/computation_builder.h index a49e5a8843..98649ba02f 100644 --- a/tensorflow/compiler/xla/client/computation_builder.h +++ b/tensorflow/compiler/xla/client/computation_builder.h @@ -504,8 +504,9 @@ class ComputationBuilder { ComputationDataHandle SquareF32(const ComputationDataHandle& operand); // Enqueues a lhs^rhs computation onto the computation. - ComputationDataHandle Pow(const ComputationDataHandle& lhs, - const ComputationDataHandle& rhs); + ComputationDataHandle Pow( + const ComputationDataHandle& lhs, const ComputationDataHandle& rhs, + tensorflow::gtl::ArraySlice<int64> broadcast_dimensions = {}); // Enqueues an operator that tests if the operand's values are finite, i.e., // not Inf or NaN. Defined only for floating-point types. Returns an array of |