未验证 提交 22c49634 编写于 作者: H hong 提交者: GitHub

fix new ir sigmoid cross entropy op (#55284)

上级 4fa3e149
......@@ -2742,6 +2742,10 @@
outputs :
out : Out
- op : tril_indices
outputs :
out : out
- op : tril_triu
backward : tril_triu_grad
inputs :
......@@ -2758,6 +2762,10 @@
extra :
attrs : [bool use_mkldnn = false]
- op : triu_indices
outputs :
out : out
- op : trunc
inputs :
input : X
......@@ -2937,7 +2945,7 @@
- op: sigmoid_cross_entropy_with_logits
backward: sigmoid_cross_entropy_with_logits_grad
inputs :
{x: X, label: Label}
{x: X, label: Label, pos_weight : pos_weight}
outputs :
out : Out
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册