|
|
|
@ -271,6 +271,7 @@ constexpr auto kPadAndShiftOpName = "PadAndShift";
|
|
|
|
|
constexpr auto kSparseSoftmaxCrossEntropyWithLogitsOpName = "SparseSoftmaxCrossEntropyWithLogits";
|
|
|
|
|
constexpr auto kOneHotOpName = "OneHot";
|
|
|
|
|
constexpr auto kSoftmaxCrossEntropyWithLogitsOpName = "SoftmaxCrossEntropyWithLogits";
|
|
|
|
|
constexpr auto kUniformCandidateSamplerOpName = "UniformCandidateSampler";
|
|
|
|
|
|
|
|
|
|
// Hcom Op Type
|
|
|
|
|
constexpr auto kHcomOpTypeAllReduce = "HcomAllReduce";
|
|
|
|
@ -492,6 +493,8 @@ const std::set<std::string> kOptOperatorSet = {kMomentumOpName,
|
|
|
|
|
|
|
|
|
|
const std::set<std::string> kPosteriorOperatorSet = {kPullOpName};
|
|
|
|
|
|
|
|
|
|
const std::set<std::string> kOpCacheAllowList = {kUniformCandidateSamplerOpName};
|
|
|
|
|
|
|
|
|
|
const std::set<std::string> kHWSpecialFormatSet = {
|
|
|
|
|
kOpFormat_FRACTAL_Z_3D, kOpFormat_NC1KHKWHWC0, kOpFormat_NC1HWC0, kOpFormat_FRAC_NZ, kOpFormat_C1HWNCoC0,
|
|
|
|
|
kOpFormat_NC1HWC0_C04, kOpFormat_FRACTAL_Z_C04, kOpFormat_FRACTAL_ZN_LSTM, kOpFormat_NDC1HWC0, kOpFormat_FRAC_Z};
|
|
|
|
|