Introduce common cwise code for "sparse <op> dense"; use for cdiv, cmul.

Currently it supports broadcasting the dense side to the sparse side only.
Change: 121602996

1 files changed, 9 insertions, 0 deletions

diff --git a/tensorflow/core/util/bcast.h b/tensorflow/core/util/bcast.h
index 45584bb37f..6c4d320fce 100644
--- a/tensorflow/core/util/bcast.h
+++ b/tensorflow/core/util/bcast.h
@@ -105,6 +105,15 @@ class BCast {
   static Vec FromShape(const TensorShape& shape);
   static TensorShape ToShape(const BCast::Vec& vec);
 
+  template <int NDIMS>
+  static Eigen::array<Eigen::DenseIndex, NDIMS> ToIndexArray(
+      const BCast::Vec& vec) {
+    CHECK_EQ(vec.size(), NDIMS);
+    Eigen::array<Eigen::DenseIndex, NDIMS> ret;
+    for (int i = 0; i < NDIMS; ++i) ret[i] = vec[i];
+    return ret;
+  }
+
  private:
   bool valid_ = true;
   Vec x_reshape_;