fix segmentfault with fused sparse ftrl

pull/7713/head
kswang 4 years ago
parent 0fd312f431
commit 74c7bdd471

@ -223,6 +223,13 @@ constexpr auto kBasicLSTMCellInputGradOpName = "BasicLSTMCellInputGrad";
constexpr auto kBasicLSTMCellOpName = "BasicLSTMCell"; constexpr auto kBasicLSTMCellOpName = "BasicLSTMCell";
constexpr auto kDynamicRNNOpName = "DynamicRNN"; constexpr auto kDynamicRNNOpName = "DynamicRNN";
constexpr auto kLSTMInputGradOpName = "LSTMInputGrad"; 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 // Hcom Op Type
constexpr auto kHcomOpTypeAllReduce = "HcomAllReduce"; constexpr auto kHcomOpTypeAllReduce = "HcomAllReduce";
@ -389,6 +396,14 @@ const std::set<std::string> kOptOperatorSet = {
kFusedSparseAdamName, kFusedSparseAdamName,
kFusedWeightScaleApplyMomentum, kFusedWeightScaleApplyMomentum,
kFusedScaleApplyMomentum, kFusedScaleApplyMomentum,
kApplyCenteredRMSPropOpName,
kFusedSparseFtrlName,
kFusedSparseProximalAdagradName,
kFusedSparseLazyAdamName,
kSparseApplyFtrlName,
kSparseApplyFtrlV2Name,
kSGDName,
kLARSUpdateName,
kPullOpName, kPullOpName,
}; };

Loading…
Cancel
Save