Skip to content

Commit 2d0ffdc

Browse files
JiabinYangdzhwinter
authored andcommitted
test=develop, fix debug mode unitest, hsigmoid (#15574)
1 parent 90ffe74 commit 2d0ffdc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/operators/hierarchical_sigmoid_op.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ class HierarchicalSigmoidOpKernel : public framework::OpKernel<T> {
136136
sum.mutable_data<T>(framework::make_ddim(sum_dims), ctx.GetPlace());
137137
auto sum_mat = EigenMatrix<T>::From(sum);
138138
out->mutable_data<T>(ctx.GetPlace());
139-
auto out_mat = framework::EigenVector<T>::Flatten(*out);
139+
auto out_mat = framework::EigenMatrix<T>::From(*out);
140140
if (bias) {
141141
bit_code->Add(*bias, pre_out);
142142
}

0 commit comments

Comments
 (0)