#include <op_where.h>

Inheritance diagram for nndeploy::op::OpWhere:

Collaboration diagram for nndeploy::op::OpWhere:

Public Member Functions
virtual base::Status	inferShape ()
	形状推理 More...

	OpWhere ()

virtual base::Status	run ()

virtual	~OpWhere ()

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	inferDataFormat ()
	数据格式推理 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_where.h.

Constructor & Destructor Documentation

◆ OpWhere()

nndeploy::op::OpWhere::OpWhere ( )

inline

Definition at line 13 of file op_where.h.

◆ ~OpWhere()

virtual nndeploy::op::OpWhere::~OpWhere ( )

inlinevirtual

Definition at line 14 of file op_where.h.

Member Function Documentation

◆ inferShape()

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

virtual

形状推理

Returns: base::Status

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

Reimplemented from nndeploy::op::Op.

◆ run()

virtual base::Status nndeploy::op::OpWhere::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/stable/framework/include/nndeploy/op/op_where.h

Public Member Functions