File tree Expand file tree Collapse file tree 3 files changed +9
-0
lines changed
fluid/pir/dialect/operator/interface/infer_symbolic_shape Expand file tree Collapse file tree 3 files changed +9
-0
lines changed Original file line number Diff line number Diff line change @@ -139,6 +139,9 @@ OP_SAME_OPERANDS_AND_RESULT(Sinh)
139
139
OP_SAME_OPERANDS_AND_RESULT (Sinh_)
140
140
OP_SAME_OPERANDS_AND_RESULT (Softmax)
141
141
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)
142
145
OP_SAME_OPERANDS_AND_RESULT (Swish)
143
146
OP_SAME_OPERANDS_AND_RESULT (Tan)
144
147
OP_SAME_OPERANDS_AND_RESULT (Tan_)
Original file line number Diff line number Diff line change @@ -132,6 +132,9 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sinh)
132
132
OP_DECLARE_INFER_SYMBOLIC_SHAPE (Sinh_)
133
133
OP_DECLARE_INFER_SYMBOLIC_SHAPE (Softmax)
134
134
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)
135
138
OP_DECLARE_INFER_SYMBOLIC_SHAPE (Swish)
136
139
OP_DECLARE_INFER_SYMBOLIC_SHAPE (Tan)
137
140
OP_DECLARE_INFER_SYMBOLIC_SHAPE (Tan_)
Original file line number Diff line number Diff line change 4241
4241
kernel :
4242
4242
func : softplus
4243
4243
backward : softplus_grad
4244
+ interfaces : paddle::dialect::InferSymbolicShapeInterface
4244
4245
4245
4246
- op : softshrink
4246
4247
args : (Tensor x, float threshold = 0.5)
4251
4252
kernel :
4252
4253
func : softshrink
4253
4254
backward : softshrink_grad
4255
+ interfaces : paddle::dialect::InferSymbolicShapeInterface
4254
4256
4255
4257
- op : softsign
4256
4258
args : (Tensor x)
4261
4263
kernel :
4262
4264
func : softsign
4263
4265
backward : softsign_grad
4266
+ interfaces : paddle::dialect::InferSymbolicShapeInterface
4264
4267
4265
4268
- op : solve
4266
4269
args : (Tensor x, Tensor y)
You can’t perform that action at this time.
0 commit comments