Merge pull request #5703 from qingqing01/op_debug

peterzhang2029 · web-flow · commit 55c2c73f3c78 · 2017-11-16T04:25:39.000-06:00
Fix bilinear_tensor_product_op in debug mode.
diff --git a/paddle/operators/bilinear_tensor_product_op.h b/paddle/operators/bilinear_tensor_product_op.h
@@ -174,7 +174,7 @@ class BilinearTensorProductGradKernel : public framework::OpKernel<T> {
     // Caculate the gradient of Input(Bias).
     if (d_bias) {
       d_bias->mutable_data<T>(ctx.GetPlace());
-      auto d_bias_mat = EigenMatrix<T>::From(*d_bias);
+      auto d_bias_mat = framework::EigenVector<T>::Flatten(*d_bias);
       d_bias_mat.device(place) = d_out_mat.sum(Eigen::DSizes<int, 1>(0));
     }
   }

Original file line number	Diff line number	Diff line change
`@@ -174,7 +174,7 @@ class BilinearTensorProductGradKernel : public framework::OpKernel<T> {`
`174`	`174`	`// Caculate the gradient of Input(Bias).`
`175`	`175`	`if (d_bias) {`
`176`	`176`	`d_bias->mutable_data<T>(ctx.GetPlace());`
`177`		`- auto d_bias_mat = EigenMatrix<T>::From(*d_bias);`
	`177`	`+ auto d_bias_mat = framework::EigenVector<T>::Flatten(*d_bias);`
`178`	`178`	`d_bias_mat.device(place) = d_out_mat.sum(Eigen::DSizes<int, 1>(0));`
`179`	`179`	`}`
`180`	`180`	`}`