未验证 提交 83f61bd5 编写于 作者: C cyber-pioneer 提交者: GitHub

add op map (#51026)

上级 1054b23e
......@@ -597,7 +597,7 @@
- op : expand_as
args : (Tensor x, Tensor y, int[] target_shape)
output : Tensor
output : Tensor(out)
infer_meta :
func : ExpandAsInferMeta
kernel :
......@@ -810,7 +810,7 @@
- op : hardswish
args : (Tensor x)
output : Tensor
output : Tensor(out)
infer_meta :
func : UnchangedInferMeta
param : [x]
......
......@@ -573,6 +573,13 @@
extra :
attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
- op : expand_as (expand_as_v2)
inputs :
x : X
y : Y
outputs :
out : Out
- op : expm1
backward : expm1_grad
inputs :
......@@ -763,6 +770,16 @@
extra :
attrs : [bool use_cudnn = true]
- op : group_norm
inputs :
x : X
scale : Scale
bias : Bias
outputs :
y : Y
mean : Mean
variance : Variance
- op : gru
backward : gru_grad
extra :
......@@ -774,11 +791,6 @@
outputs :
out : Out
- op : hard_swish
backward : hard_swish_grad
extra :
attrs : [bool use_mkldnn = false]
- op : hardshrink (hard_shrink)
backward : hardshrink_grad (hard_shrink_grad)
inputs :
......@@ -793,6 +805,15 @@
outputs :
out : Out
- op : hardswish (hard_swish)
inputs :
x : X
outputs :
out : Out
backward : hard_swish_grad
extra :
attrs : [bool use_mkldnn = false]
- op : hardtanh (brelu)
backward : hardtanh_grad (brelu_grad)
inputs :
......@@ -838,6 +859,16 @@
extra :
attrs : [bool use_mkldnn = false, bool fuse_with_relu = false]
- op : instance_norm
inputs :
x : X
scale : Scale
bias : Bias
outputs :
y : Y
saved_mean : SavedMean
saved_variance : SavedVariance
- op : inverse
inputs :
x : Input
......@@ -1184,6 +1215,12 @@
outputs :
{out : Out, total_weight : Total_weight}
- op : nonzero (where_index)
inputs :
condition : Condition
outputs :
out : Out
- op : not_equal
inputs :
{x : X, y : Y}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册