expr.h File Reference

#include "nndeploy/base/macro.h"
#include "nndeploy/device/tensor.h"
#include "nndeploy/ir/ir.h"

Include dependency graph for expr.h:

Go to the source code of this file.

Classes
class	nndeploy::op::Expr

Namespaces
	nndeploy
	nndeploy::op

Enumerations
enum	nndeploy::op::ExprType : int { nndeploy::op::kExprTypeValueDesc = 0x0000 , nndeploy::op::kExprTypeOpDesc , nndeploy::op::kExprTypeModelDesc }

Functions
std::string	nndeploy::op::exprTypeToString (ExprType type)
std::shared_ptr< Expr >	nndeploy::op::makeAdd (ir::ModelDesc *model_desc, std::shared_ptr< Expr > input_0, std::shared_ptr< Expr > input_1, std::string op_name="", std::string output_name="")
std::shared_ptr< Expr >	nndeploy::op::makeBatchNorm (ir::ModelDesc *model_desc, std::shared_ptr< Expr > input, std::shared_ptr< ir::BatchNormalizationParam > param, const std::string &scale, const std::string &bias, const std::string &mean, const std::string &var, std::string op_name="", std::string output_name="")
std::shared_ptr< Expr >	nndeploy::op::makeBlock (ir::ModelDesc *model_desc, std::shared_ptr< ir::ModelDesc > model_block)
std::shared_ptr< Expr >	nndeploy::op::makeConstant (ir::ModelDesc *model_desc, std::string name)
std::shared_ptr< Expr >	nndeploy::op::makeConv (ir::ModelDesc *model_desc, std::shared_ptr< Expr > input, std::shared_ptr< ir::ConvParam > param, const std::string &weight, const std::string &bias="", std::string op_name="", std::string output_name="")
std::shared_ptr< Expr >	nndeploy::op::makeEmbedding (ir::ModelDesc *model_desc, std::shared_ptr< Expr > indices, std::string op_name, std::string output_name)
std::shared_ptr< Expr >	nndeploy::op::makeFlatten (ir::ModelDesc *model_desc, std::shared_ptr< Expr > input, std::shared_ptr< ir::FlattenParam > param, std::string op_name="", std::string output_name="")
std::shared_ptr< Expr >	nndeploy::op::makeGemm (ir::ModelDesc *model_desc, std::shared_ptr< Expr > input, std::shared_ptr< ir::GemmParam > param, const std::string &weight, const std::string &bias="", std::string op_name="", std::string output_name="")
std::shared_ptr< Expr >	nndeploy::op::makeGlobalAveragePool (ir::ModelDesc *model_desc, std::shared_ptr< Expr > input, std::string op_name="", std::string output_name="")
std::shared_ptr< Expr >	nndeploy::op::makeInput (ir::ModelDesc *model_desc, std::string name, base::DataType data_type=base::dataTypeOf< float >(), base::IntVector shape=base::IntVector())
	一系列创建函数 More...
std::shared_ptr< Expr >	nndeploy::op::makeMaxPool (ir::ModelDesc *model_desc, std::shared_ptr< Expr > input, std::shared_ptr< ir::MaxPoolParam > param, std::string op_name="", std::string output_name="")
void	nndeploy::op::makeOutput (ir::ModelDesc *model_desc, std::shared_ptr< Expr > expr)
std::shared_ptr< Expr >	nndeploy::op::makeRelu (ir::ModelDesc *model_desc, std::shared_ptr< Expr > input, std::string op_name="", std::string output_name="")
std::shared_ptr< Expr >	nndeploy::op::makeSigmoid (ir::ModelDesc *model_desc, std::shared_ptr< Expr > input, std::string op_name="", std::string output_name="")
std::shared_ptr< Expr >	nndeploy::op::makeSoftmax (ir::ModelDesc *model_desc, std::shared_ptr< Expr > input, std::shared_ptr< ir::SoftmaxParam > param, std::string op_name="", std::string output_name="")
ExprType	nndeploy::op::stringToExprType (const std::string &src)