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

add op map (#51026)

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