MCPcopy Create free account
hub / github.com/alibaba/MNN / onOptimize

Method onOptimize

express/module/StaticModule.cpp:690–714  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

688 return this->cloneBaseTo(ctx, module);
689}
690int StaticModule::onOptimize(Interpreter::SessionMode stage) {
691 int res = 0;
692 switch (stage) {
693 case MNN::Interpreter::Session_Resize_Check:
694 mSession->openResizeCheck();
695 break;
696 case MNN::Interpreter::Session_Resize_Fix:
697 mSession->fixResizeCache();
698 break;
699 case MNN::Interpreter::Module_Forward_Separate:
700 if (mResource->mUseContentInputs || mResource->mModes.inputMode != Interpreter::Session_Input_User || mResource->mOutputFromTensor.empty()) {
701 res = NOT_SUPPORT;
702 break;
703 }
704 mShapeInferSeperate = true;
705 break;
706 case MNN::Interpreter::Module_Forward_Combine:
707 mOutputVars.clear();
708 mShapeInferSeperate = false;
709 break;
710 default:
711 break;
712 }
713 return res;
714}
715
716} // namespace Express
717} // namespace MNN

Callers 1

traceOrOptimizeMethod · 0.45

Calls 4

openResizeCheckMethod · 0.45
fixResizeCacheMethod · 0.45
emptyMethod · 0.45
clearMethod · 0.45

Tested by

no test coverage detected