nndeploy C++ API  0.2.0
nndeploy C++ API
Classes | Namespaces | Typedefs | Enumerations | Functions
optimizer.h File Reference
#include <unordered_map>
#include <unordered_set>
#include "nndeploy/ir/ir.h"
#include "nndeploy/net/util.h"
#include "nndeploy/op/op.h"
Include dependency graph for optimizer.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  nndeploy::net::Optimizer
 
class  nndeploy::net::OptPass
 
class  nndeploy::net::OptPassCreator
 OptPass的创建类 More...
 
class  nndeploy::net::TypeOptPassCreator< T >
 OptPass的创建类模板 More...
 
class  nndeploy::net::TypeOptPassRegister< T >
 OptPass的创建类的注册类模板 More...
 

Namespaces

 nndeploy
 
 nndeploy::net
 

Typedefs

using nndeploy::net::OpSet = std::unordered_set< ir::OpType >
 

Enumerations

enum  nndeploy::net::OptPassType : int {
  nndeploy::net::kOptPassTypeFuseConvBias , nndeploy::net::kOptPassTypeFuseConvBatchNorm , nndeploy::net::kOptPassTypeFuseConvRelu , nndeploy::net::kOptPassTypeFuseConvAct ,
  nndeploy::net::kOptPassTypeEliminateCommonSubexpression , nndeploy::net::kOptPassTypeEliminateDeadOp , nndeploy::net::kOptPassTypeFoldConstant , nndeploy::net::kOptPassTypeFuseQdq
}
 

Functions

std::shared_ptr< OptPass > nndeploy::net::createOptPass (base::DeviceType device_type, int level, OptPassType type)
 
std::map< base::DeviceTypeCode, std::map< int, std::map< OptPassType, std::shared_ptr< OptPassCreator > > > > & nndeploy::net::getGlobalOptPassCreatorMap ()
 Get the Global OptPass Creator Map object. More...
 
std::string nndeploy::net::optPassTypeToString (OptPassType type)
 
OptPassType nndeploy::net::stringToOptPassType (const std::string &src)