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

fix new ir sigmoid cross entropy op (#55284)

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