diff --git a/paddle/framework/eigen.h b/paddle/framework/eigen.h
index 4ba4fd4d110330805faf2468bd406cb23c6f1b1c..f03b26a5fe4b412a309bf1648a8c9c348fb60a8b 100644
--- a/paddle/framework/eigen.h
+++ b/paddle/framework/eigen.h
@@ -65,13 +65,15 @@ template <typename T, int MajorType = Eigen::RowMajor,
           typename IndexType = Eigen::DenseIndex>
 struct EigenVector : public EigenTensor<T, 1, MajorType, IndexType> {
   // Flatten is to reshape a Tensor into a one dimension EigenVector
-  static typename EigenTensor<T, 1>::Type Flatten(Tensor& tensor) {
-    return EigenTensor<T, 1>::From(
+  static typename EigenTensor<T, 1, MajorType, IndexType>::Type Flatten(
+      Tensor& tensor) {
+    return EigenTensor<T, 1, MajorType, IndexType>::From(
         tensor, make_ddim({static_cast<int>(product(tensor.dims_))}));
   }
 
-  static typename EigenTensor<T, 1>::ConstType Flatten(const Tensor& tensor) {
-    return EigenTensor<T, 1>::From(
+  static typename EigenTensor<T, 1, MajorType, IndexType>::ConstType Flatten(
+      const Tensor& tensor) {
+    return EigenTensor<T, 1, MajorType, IndexType>::From(
         tensor, make_ddim({static_cast<int>(product(tensor.dims_))}));
   }
 };