nndeploy::op::OpTranspose Class Reference

#include <op_transpose.h>

Inheritance diagram for nndeploy::op::OpTranspose:

Collaboration diagram for nndeploy::op::OpTranspose:

Public Member Functions
virtual base::Status	inferDataFormat ()
	数据格式推理 More...
virtual base::Status	inferShape ()
	形状推理 More...
	OpTranspose ()
virtual base::Status	run ()
virtual	~OpTranspose ()
Public Member Functions inherited from nndeploy::op::Op
virtual base::Status	allocateWorkspace ()
virtual base::Status	checkOrAllocOutput ()
	检查输出tensor More...
virtual base::Status	deinit ()
std::vector< device::Tensor * >	getAllInput ()
std::vector< std::string >	getAllInputName ()
std::vector< device::Tensor * >	getAllOutput ()
std::vector< std::string >	getAllOutputName ()
bool	getConstructed ()
bool	getDebugFlag ()
base::DeviceType	getDeviceType ()
virtual uint64_t	getFlops ()
	得到op的flops More...
bool	getInitialized ()
device::Tensor *	getInput (int index=0)
std::string	getInputName (int index=0)
device::Tensor *	getInputTensor (const std::string &name)
std::string	getName ()
ir::OpType	getOpType ()
device::Tensor *	getOutput (int index=0)
std::string	getOutputName (int index=0)
device::Tensor *	getOutputTensor (const std::string &name)
base::ParallelType	getParallelType ()
virtual std::shared_ptr< base::Param >	getParam ()
base::PrecisionType	getPrecisionType ()
device::Stream *	getStream ()
bool	getTimeProfileFlag ()
virtual uint64_t	getWorkspaceSize ()
	得到op的workspace大小 note: op在运行时的workspace大小，在输入确定后调用 eg：例如Conv，当存在padding时，需要分配额外的内存，存放padding后的内存 More...
virtual base::Status	inferDataType ()
	类型推理 More...
virtual base::Status	init ()
	初始化 More...
bool	isRunning ()
	Op ()
virtual base::Status	postRun ()
virtual base::Status	preRun ()
base::Status	replaceInputTensor (const std::string &name, device::Tensor *tensor)
base::Status	replaceOutputTensor (const std::string &name, device::Tensor *tensor)
virtual base::Status	reshape (base::ShapeMap &shape_map)
	重新推理形状，通常在初始化之后、preRun前调用 More...
base::Status	rmInput (device::Tensor *tensor)
base::Status	setAllInput (std::vector< device::Tensor * > inputs)
base::Status	setAllInputName (std::initializer_list< std::string >)
base::Status	setAllInputName (std::vector< std::string > &)
base::Status	setAllOutput (std::vector< device::Tensor * > outputs)
base::Status	setAllOutputName (std::initializer_list< std::string >)
base::Status	setAllOutputName (std::vector< std::string > &)
void	setDebugFlag (bool flag)
base::Status	setDeviceType (base::DeviceType device_type)
void	setInitializedFlag (bool flag)
void	setInnerFlag (bool flag)
virtual base::Status	setInput (device::Tensor *input)
base::Status	setInput (device::Tensor *input, int index)
base::Status	setName (std::string name)
base::Status	setOpType (ir::OpType op_type)
virtual base::Status	setOutput (device::Tensor *output)
base::Status	setOutput (device::Tensor *output, int index)
base::Status	setParallelType (const base::ParallelType &paralle_type)
virtual base::Status	setParam (std::shared_ptr< base::Param > param)
virtual base::Status	setPrecisionType (base::PrecisionType precision_type)
	设置精度类型 精度不同，计算方式不同，内存分配不同 More...
void	setRunningFlag (bool flag)
void	setStream (device::Stream *stream)
void	setTimeProfileFlag (bool flag)
virtual void	setWorkspace (void *workspace)
virtual	~Op ()

Additional Inherited Members
Protected Attributes inherited from nndeploy::op::Op
bool	constructed_ = false
base::DeviceType	device_type_
	op的设备类型 More...
uint64_t	flops_ = 0
bool	initialized_ = false
std::vector< device::Tensor * >	inputs_
	op的输入tensor note: 当权重为tensor时，权重tensor也会在这里 eg: More...
bool	is_changed_ = false
bool	is_debug_ = false
bool	is_external_stream_ = false
	op的stream note: 当stream为外部传入时，is_external_stream_为true More...
bool	is_inner_ = false
bool	is_inplace_ = false
bool	is_running_ = false
bool	is_time_profile_ = false
ir::OpDesc	op_desc_
	op的描述 包含op的类型、名称、输入名称、输出名称、参数 More...
std::vector< device::Tensor * >	outputs_
	op的输出tensor More...
base::ParallelType	parallel_type_ = base::kParallelTypeNone
base::PrecisionType	precision_type_ = base::kPrecisionTypeFp32
	op的精度类型 note: 精度类型与输入输出tensor的data_type的不同 More...
device::Stream *	stream_ = nullptr
void *	workspace_ = nullptr
bool	workspace_is_external_ = false
	op的workspace大小 note: op在运行时的workspace大小, 在preRun中确定 eg：例如Conv，当存在padding时，需要分配额外的内存，存放padding后的内存 More...
uint64_t	workspace_size_ = 0

Detailed Description

Definition at line 11 of file op_transpose.h.

Constructor & Destructor Documentation

OpTranspose()

nndeploy::op::OpTranspose::OpTranspose ( )

inline

Definition at line 13 of file op_transpose.h.

~OpTranspose()

virtual nndeploy::op::OpTranspose::~OpTranspose ( )

inlinevirtual

Definition at line 14 of file op_transpose.h.

Member Function Documentation

inferDataFormat()

virtual base::Status nndeploy::op::OpTranspose::inferDataFormat ( )

virtual

数据格式推理

Returns

base::Status

Note

当输入的shape数据格式时，在计算图Net::init中调用该函数

Reimplemented from nndeploy::op::Op.

inferShape()

virtual base::Status nndeploy::op::OpTranspose::inferShape ( )

virtual

形状推理

Returns

base::Status

Note

当输入的shape确定时，在计算图Net::init中调用该函数

Reimplemented from nndeploy::op::Op.

run()

virtual base::Status nndeploy::op::OpTranspose::run ( )

virtual

Implements nndeploy::op::Op.

---

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

• /home/docs/checkouts/readthedocs.org/user_builds/nndeploy-zh/checkouts/latest/framework/include/nndeploy/op/op_transpose.h