chang zherui
5 years ago
1 changed files with 4 additions and 2 deletions
Unified View
Diff Options
+ 4
- 2
mindspore/ccsrc/transform/op_declare.h
View File
| @@ -403,8 +403,8 @@ DECLARE_OP_ADAPTER(Sigmoid) | |||||
| DECLARE_OP_USE_OUTPUT(Sigmoid) | DECLARE_OP_USE_OUTPUT(Sigmoid) | ||||
| DECLARE_OP_ADAPTER(SigmoidGrad) | DECLARE_OP_ADAPTER(SigmoidGrad) | ||||
| DECLARE_OP_USE_OUTPUT(SigmoidGrad) | DECLARE_OP_USE_OUTPUT(SigmoidGrad) | ||||
| DECLARE_OP_ADAPTER(Softmax) | |||||
| DECLARE_OP_USE_OUTPUT(Softmax) | |||||
| DECLARE_OP_ADAPTER(SoftmaxV2) | |||||
| DECLARE_OP_USE_OUTPUT(SoftmaxV2) | |||||
| DECLARE_OP_ADAPTER(SoftmaxGrad) | DECLARE_OP_ADAPTER(SoftmaxGrad) | ||||
| DECLARE_OP_USE_OUTPUT(SoftmaxGrad) | DECLARE_OP_USE_OUTPUT(SoftmaxGrad) | ||||
| DECLARE_OP_ADAPTER(Greater) | DECLARE_OP_ADAPTER(Greater) | ||||
| @@ -447,6 +447,8 @@ DECLARE_OP_ADAPTER(Round) | |||||
| DECLARE_OP_USE_OUTPUT(Round) | DECLARE_OP_USE_OUTPUT(Round) | ||||
| DECLARE_OP_ADAPTER(ApplyFtrl) | DECLARE_OP_ADAPTER(ApplyFtrl) | ||||
| DECLARE_OP_USE_OUTPUT(ApplyFtrl) | DECLARE_OP_USE_OUTPUT(ApplyFtrl) | ||||
| DECLARE_OP_ADAPTER(SparseApplyFtrlD) | |||||
| DECLARE_OP_USE_OUTPUT(SparseApplyFtrlD) | |||||
| DECLARE_OP_ADAPTER(Diag) | DECLARE_OP_ADAPTER(Diag) | ||||
| DECLARE_OP_USE_OUTPUT(Diag) | DECLARE_OP_USE_OUTPUT(Diag) | ||||
| DECLARE_OP_ADAPTER(DiagPart) | DECLARE_OP_ADAPTER(DiagPart) | ||||