diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-11-01 21:02:15 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-01 21:06:04 -0700 |
| commit | 0a1d8d7e4e690b4aef4c0c9d831adc17cbc6dc5c (patch) | |
| tree | 9defadd0bd98118901ed9d2a653e2e12ef3ed472 | |
| parent | 2b9473b471fd2f9ad20dda4062a1227438afab86 (diff) | |
Enabled broadcast optimization for elementwise multiply.
PiperOrigin-RevId: 174278845
| -rw-r--r-- | tensorflow/core/kernels/cwise_ops.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/cwise_ops.h b/tensorflow/core/kernels/cwise_ops.h index 89487419ee..6c22b124de 100644 --- a/tensorflow/core/kernels/cwise_ops.h +++ b/tensorflow/core/kernels/cwise_ops.h @@ -676,7 +676,9 @@ struct sub : base<T, Eigen::internal::scalar_difference_op<T>> { }; template <typename T> -struct mul : base<T, Eigen::internal::scalar_product_op<T>> {}; +struct mul : base<T, Eigen::internal::scalar_product_op<T>> { + static const bool use_bcast_optimization = true; +}; template <typename T> struct div : base<T, Eigen::internal::scalar_quotient_op<T>> {}; |