Tensor block evaluation V2 support for unary/binary/broadcsting

2026-04-10 11:34:33 +08:00 · 2019-09-24 12:52:45 -07:00
parent efd9867ff0
commit ef9dfee7bd
40 changed files with 1162 additions and 169 deletions
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorBlock.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBlock.h
@@ -1025,6 +1025,11 @@ class TensorBlockMapper {
    return m_block_dim_sizes.TotalSize();
  }

+  EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions&
+  block_dim_sizes() const {
+    return m_block_dim_sizes;
+  }
+
 private:
  static Dimensions BlockDimensions(const Dimensions& tensor_dims,
                                    const TensorBlockShapeType block_shape,