forked from mindspore-Ecosystem/mindspore
!7713 fix segmentfault with fusedsparseftrl
Merge pull request !7713 from kisnwang/fix-fused-sparse-ftrl-error
This commit is contained in:
commit
8c335bcd91
|
@ -223,6 +223,13 @@ constexpr auto kBasicLSTMCellInputGradOpName = "BasicLSTMCellInputGrad";
|
|||
constexpr auto kBasicLSTMCellOpName = "BasicLSTMCell";
|
||||
constexpr auto kDynamicRNNOpName = "DynamicRNN";
|
||||
constexpr auto kLSTMInputGradOpName = "LSTMInputGrad";
|
||||
constexpr auto kFusedSparseFtrlName = "FusedSparseFtrl";
|
||||
constexpr auto kFusedSparseProximalAdagradName = "FusedSparseProximalAdagrad";
|
||||
constexpr auto kFusedSparseLazyAdamName = "FusedSparseLazyAdam";
|
||||
constexpr auto kSparseApplyFtrlName = "SparseApplyFtrl";
|
||||
constexpr auto kSparseApplyFtrlV2Name = "SparseApplyFtrlV2";
|
||||
constexpr auto kSGDName = "SGD";
|
||||
constexpr auto kLARSUpdateName = "LARSUpdate";
|
||||
|
||||
// Hcom Op Type
|
||||
constexpr auto kHcomOpTypeAllReduce = "HcomAllReduce";
|
||||
|
@ -389,6 +396,14 @@ const std::set<std::string> kOptOperatorSet = {
|
|||
kFusedSparseAdamName,
|
||||
kFusedWeightScaleApplyMomentum,
|
||||
kFusedScaleApplyMomentum,
|
||||
kApplyCenteredRMSPropOpName,
|
||||
kFusedSparseFtrlName,
|
||||
kFusedSparseProximalAdagradName,
|
||||
kFusedSparseLazyAdamName,
|
||||
kSparseApplyFtrlName,
|
||||
kSparseApplyFtrlV2Name,
|
||||
kSGDName,
|
||||
kLARSUpdateName,
|
||||
kPullOpName,
|
||||
};
|
||||
|
||||
|
|
Loading…
Reference in New Issue