未验证 提交 38782dc3 编写于 作者: K kangguangli 提交者: GitHub

[NewIR] fix hsigmoid_loss (#55483)

* fix hsigmoid_loss

* add test into whitelist
上级 2558364c
......@@ -1325,6 +1325,13 @@
outputs :
out : Out
- op : hsigmoid_loss(hierarchical_sigmoid)
backward: hsigmoid_loss_grad(hierarchical_sigmoid_grad)
inputs:
{x: X, w: W, label: Label, bias: Bias, path: PathTable, code: PathCode}
outputs:
{out: Out, pre_out: PreOut, w_out: W_Out}
- op : huber_loss
backward : huber_loss_grad
inputs :
......
......@@ -81,6 +81,7 @@ test_graph_send_ue_recv_op
test_grid_sampler_op
test_group_norm_op
test_histogram_op
test_histogram_op.py
test_huber_loss_op
test_i0e_op
test_i0_op
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册