nndeploy::net::Optimizer Class Reference

#include <optimizer.h>

Collaboration diagram for nndeploy::net::Optimizer:

Public Member Functions
base::Status	addPass (OptPassType type, int level)
base::Status	deinit ()
base::Status	init (base::DeviceType device_type, std::set< OptPassType > enable_pass, std::set< OptPassType > disable_pass)
base::Status	optimize (std::vector< TensorWrapper * > &tensor_repository, std::vector< OpWrapper * > &op_repository, Net *net)
	Optimizer ()
base::Status	removePass (OptPassType type)
	~Optimizer ()

Protected Attributes
base::DeviceType	device_type_
std::map< int, std::map< OptPassType, std::shared_ptr< OptPass > > >	opt_passes_

Detailed Description

Definition at line 206 of file optimizer.h.

Constructor & Destructor Documentation

Optimizer()

nndeploy::net::Optimizer::Optimizer

(

)

~Optimizer()

nndeploy::net::Optimizer::~Optimizer

(

)

Member Function Documentation

addPass()

base::Status nndeploy::net::Optimizer::addPass	(	OptPassType	type,
		int	level
	)

deinit()

base::Status nndeploy::net::Optimizer::deinit

(

)

init()

base::Status nndeploy::net::Optimizer::init	(	base::DeviceType	device_type,
		std::set< OptPassType >	enable_pass,
		std::set< OptPassType >	disable_pass
	)

optimize()

base::Status nndeploy::net::Optimizer::optimize	(	std::vector< TensorWrapper * > &	tensor_repository,
		std::vector< OpWrapper * > &	op_repository,
		Net *	net
	)

removePass()

base::Status nndeploy::net::Optimizer::removePass

(

OptPassType

type

)

Member Data Documentation

device_type_

base::DeviceType nndeploy::net::Optimizer::device_type_

protected

Definition at line 223 of file optimizer.h.

opt_passes_

std::map<int, std::map<OptPassType, std::shared_ptr<OptPass> > > nndeploy::net::Optimizer::opt_passes_

protected

Definition at line 225 of file optimizer.h.

---

The documentation for this class was generated from the following file:

• /home/docs/checkouts/readthedocs.org/user_builds/nndeploy-zh/checkouts/stable/framework/include/nndeploy/net/optimizer.h