|
|
|
@@ -59,8 +59,8 @@ constexpr auto kCustRunApi = "RunCpuKernel"; |
|
|
|
constexpr auto kDropout2D = "Dropout2D"; |
|
|
|
constexpr auto kDropout3D = "Dropout3D"; |
|
|
|
const std::set<std::string> kCustAiCpuKernelOps{kIdentity}; |
|
|
|
const std::set<std::string> kCacheKernelOps{kUpdateCache, kCacheSwapTable, kSubAndFilter, kPadAndShift, kDropout3D, |
|
|
|
kDropout2D, kStackInit, kStackPush, kStackPop, kStackDestroy}; |
|
|
|
const std::set<std::string> kCacheKernelOps{kUpdateCache, kCacheSwapTable, kSubAndFilter, |
|
|
|
kPadAndShift, kDropout3D, kDropout2D}; |
|
|
|
|
|
|
|
struct AicpuParamHead { |
|
|
|
uint32_t length; // Total length: include cunstom message |
|
|
|
|