Skip to content

Commit 89bfb46

Browse files
authored
【Infer Symbolic Shape No.96-98 BUAA】Add softplus, softshrink and softsign op (#66967)
* Update ops.yaml * Update same_operands_result.h * Update same_operands_result.cc
1 parent 3861d2f commit 89bfb46

File tree

3 files changed

+9
-0
lines changed

3 files changed

+9
-0
lines changed

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/same_operands_result.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,9 @@ OP_SAME_OPERANDS_AND_RESULT(Sinh)
139139
OP_SAME_OPERANDS_AND_RESULT(Sinh_)
140140
OP_SAME_OPERANDS_AND_RESULT(Softmax)
141141
OP_SAME_OPERANDS_AND_RESULT(Softmax_)
142+
OP_SAME_OPERANDS_AND_RESULT(Softplus)
143+
OP_SAME_OPERANDS_AND_RESULT(Softshrink)
144+
OP_SAME_OPERANDS_AND_RESULT(Softsign)
142145
OP_SAME_OPERANDS_AND_RESULT(Swish)
143146
OP_SAME_OPERANDS_AND_RESULT(Tan)
144147
OP_SAME_OPERANDS_AND_RESULT(Tan_)

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/same_operands_result.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,9 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sinh)
132132
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sinh_)
133133
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Softmax)
134134
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Softmax_)
135+
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Softplus)
136+
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Softshrink)
137+
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Softsign)
135138
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Swish)
136139
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Tan)
137140
OP_DECLARE_INFER_SYMBOLIC_SHAPE(Tan_)

paddle/phi/ops/yaml/ops.yaml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4241,6 +4241,7 @@
42414241
kernel :
42424242
func : softplus
42434243
backward : softplus_grad
4244+
interfaces : paddle::dialect::InferSymbolicShapeInterface
42444245

42454246
- op : softshrink
42464247
args : (Tensor x, float threshold = 0.5)
@@ -4251,6 +4252,7 @@
42514252
kernel :
42524253
func : softshrink
42534254
backward : softshrink_grad
4255+
interfaces : paddle::dialect::InferSymbolicShapeInterface
42544256

42554257
- op : softsign
42564258
args : (Tensor x)
@@ -4261,6 +4263,7 @@
42614263
kernel :
42624264
func : softsign
42634265
backward : softsign_grad
4266+
interfaces : paddle::dialect::InferSymbolicShapeInterface
42644267

42654268
- op : solve
42664269
args : (Tensor x, Tensor y)

0 commit comments

Comments
 (0)