test=develop

e9be3366 · JiabinYang · 3c6102a3 · e9be3366
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/hierarchical_sigmoid_op.h paddle/fluid/operators/hierarchical_sigmoid_op.h +1 -1

未找到文件。
--- a/paddle/fluid/operators/hierarchical_sigmoid_op.h
+++ b/paddle/fluid/operators/hierarchical_sigmoid_op.h
@@ -164,7 +164,7 @@ class HierarchicalSigmoidGradOpKernel : public framework::OpKernel<T> {
    auto pre_out_grad_mat = EigenMatrix<T>::From(pre_out_grad);
    auto out_grad_mat = EigenMatrix<T>::From(*out_grad);

-    Eigen::array<int, 2> bcast({{1, static_cast<int>(pre_out_grad.dims()[1])}});
+    Eigen::array<int, 2> bcast{1, static_cast<int>(pre_out_grad.dims()[1])};

    // softrelu derivative
    pre_out_grad_mat.device(place) =