We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 90ffe74 commit 2d0ffdcCopy full SHA for 2d0ffdc
paddle/fluid/operators/hierarchical_sigmoid_op.h
@@ -136,7 +136,7 @@ class HierarchicalSigmoidOpKernel : public framework::OpKernel<T> {
136
sum.mutable_data<T>(framework::make_ddim(sum_dims), ctx.GetPlace());
137
auto sum_mat = EigenMatrix<T>::From(sum);
138
out->mutable_data<T>(ctx.GetPlace());
139
- auto out_mat = framework::EigenVector<T>::Flatten(*out);
+ auto out_mat = framework::EigenMatrix<T>::From(*out);
140
if (bias) {
141
bit_code->Add(*bias, pre_out);
142
}
0 commit comments