|
|
|
@ -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,
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|