nndeploy.op.op

Functions

create_op(device_type, name, op_type[, ...])

register_op_creator(device_type_code, ...)

Classes

Op()

OpCreator()

class nndeploy.op.op.Op[源代码]

基类：Op

__init__(self: nndeploy._nndeploy_internal.op.Op) → None[源代码]
set_name(self: nndeploy._nndeploy_internal.op.Op, arg0: str) → nndeploy._nndeploy_internal.base.Status[源代码]
get_name(self: nndeploy._nndeploy_internal.op.Op) → str[源代码]
set_op_type(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.ir.OpType) → nndeploy._nndeploy_internal.base.Status[源代码]
get_op_type(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.ir.OpType[源代码]
set_param(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.base.Param) → nndeploy._nndeploy_internal.base.Status[源代码]
get_param(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.Param[源代码]
set_device_type(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.base.DeviceType) → nndeploy._nndeploy_internal.base.Status[源代码]
get_device_type(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.DeviceType[源代码]
set_stream(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.device.Stream) → None[源代码]
get_stream(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.device.Stream[源代码]
set_precision_type(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.base.PrecisionType) → nndeploy._nndeploy_internal.base.Status[源代码]
get_precision_type(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.PrecisionType[源代码]
get_input_name(self: nndeploy._nndeploy_internal.op.Op, index: int = 0) → str[源代码]
get_output_name(self: nndeploy._nndeploy_internal.op.Op, index: int = 0) → str[源代码]
get_input(self: nndeploy._nndeploy_internal.op.Op, index: int = 0) → nndeploy._nndeploy_internal.device.Tensor[源代码]
get_output(self: nndeploy._nndeploy_internal.op.Op, index: int = 0) → nndeploy._nndeploy_internal.device.Tensor[源代码]
set_input(*args, **kwargs)[源代码]

Overloaded function.

  1. set_input(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.device.Tensor) -> nndeploy._nndeploy_internal.base.Status

  2. set_input(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.device.Tensor, arg1: int) -> nndeploy._nndeploy_internal.base.Status

set_output(*args, **kwargs)[源代码]

Overloaded function.

  1. set_output(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.device.Tensor) -> nndeploy._nndeploy_internal.base.Status

  2. set_output(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.device.Tensor, arg1: int) -> nndeploy._nndeploy_internal.base.Status

set_all_input_name(*args, **kwargs)[源代码]

Overloaded function.

  1. set_all_input_name(self: nndeploy._nndeploy_internal.op.Op, arg0: std::initializer_list<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >) -> nndeploy._nndeploy_internal.base.Status

  2. set_all_input_name(self: nndeploy._nndeploy_internal.op.Op, arg0: list[str]) -> nndeploy._nndeploy_internal.base.Status

set_all_output_name(*args, **kwargs)[源代码]

Overloaded function.

  1. set_all_output_name(self: nndeploy._nndeploy_internal.op.Op, arg0: std::initializer_list<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > >) -> nndeploy._nndeploy_internal.base.Status

  2. set_all_output_name(self: nndeploy._nndeploy_internal.op.Op, arg0: list[str]) -> nndeploy._nndeploy_internal.base.Status

get_all_input_name(self: nndeploy._nndeploy_internal.op.Op) → list[str][源代码]
get_all_output_name(self: nndeploy._nndeploy_internal.op.Op) → list[str][源代码]
get_all_input(self: nndeploy._nndeploy_internal.op.Op) → list[nndeploy._nndeploy_internal.device.Tensor][源代码]
get_all_output(self: nndeploy._nndeploy_internal.op.Op) → list[nndeploy._nndeploy_internal.device.Tensor][源代码]
rm_input(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.device.Tensor) → nndeploy._nndeploy_internal.base.Status[源代码]
set_all_input(self: nndeploy._nndeploy_internal.op.Op, arg0: list[nndeploy._nndeploy_internal.device.Tensor]) → nndeploy._nndeploy_internal.base.Status[源代码]
set_all_output(self: nndeploy._nndeploy_internal.op.Op, arg0: list[nndeploy._nndeploy_internal.device.Tensor]) → nndeploy._nndeploy_internal.base.Status[源代码]
get_constructed(self: nndeploy._nndeploy_internal.op.Op) → bool[源代码]
set_parallel_type(self: nndeploy._nndeploy_internal.op.Op, arg0: nndeploy._nndeploy_internal.base.ParallelType) → nndeploy._nndeploy_internal.base.Status[源代码]
get_parallel_type(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.ParallelType[源代码]
set_inner_flag(self: nndeploy._nndeploy_internal.op.Op, arg0: bool) → None[源代码]
set_initialized_flag(self: nndeploy._nndeploy_internal.op.Op, arg0: bool) → None[源代码]
get_initialized(self: nndeploy._nndeploy_internal.op.Op) → bool[源代码]
set_time_profile_flag(self: nndeploy._nndeploy_internal.op.Op, arg0: bool) → None[源代码]
get_time_profile_flag(self: nndeploy._nndeploy_internal.op.Op) → bool[源代码]
set_debug_flag(self: nndeploy._nndeploy_internal.op.Op, arg0: bool) → None[源代码]
get_debug_flag(self: nndeploy._nndeploy_internal.op.Op) → bool[源代码]
set_running_flag(self: nndeploy._nndeploy_internal.op.Op, arg0: bool) → None[源代码]
is_running(self: nndeploy._nndeploy_internal.op.Op) → bool[源代码]
infer_data_type(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.Status[源代码]
infer_shape(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.Status[源代码]
infer_data_format(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.Status[源代码]
init(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.Status[源代码]
deinit(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.Status[源代码]
reshape(self: nndeploy._nndeploy_internal.op.Op, arg0: dict[str, list[int]]) → nndeploy._nndeploy_internal.base.Status[源代码]
pre_run(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.Status[源代码]
get_workspace_size(self: nndeploy._nndeploy_internal.op.Op) → int[源代码]
set_workspace(self: nndeploy._nndeploy_internal.op.Op, arg0: capsule) → None[源代码]
get_flops(self: nndeploy._nndeploy_internal.op.Op) → int[源代码]
check_or_alloc_output(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.Status[源代码]
run(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.Status[源代码]
post_run(self: nndeploy._nndeploy_internal.op.Op) → nndeploy._nndeploy_internal.base.Status[源代码]
class nndeploy.op.op.OpCreator[源代码]

基类：OpCreator

__init__(self: nndeploy._nndeploy_internal.op.OpCreator) → None[源代码]
create_op(self: nndeploy._nndeploy_internal.op.OpCreator, arg0: nndeploy._nndeploy_internal.base.DeviceType, arg1: str, arg2: nndeploy._nndeploy_internal.ir.OpType, arg3: list[str], arg4: list[str]) → nndeploy._nndeploy_internal.op.Op[源代码]
create_op_shared_ptr(self: nndeploy._nndeploy_internal.op.OpCreator, arg0: nndeploy._nndeploy_internal.base.DeviceType, arg1: str, arg2: nndeploy._nndeploy_internal.ir.OpType, arg3: list[str], arg4: list[str]) → nndeploy._nndeploy_internal.op.Op[源代码]
nndeploy.op.op.register_op_creator(device_type_code: DeviceTypeCode, op_type: OpType, creator: OpCreator)[源代码]
nndeploy.op.op.create_op(device_type: DeviceType, name: str, op_type: OpType, inputs: list[str] = [], outputs: list[str] = [], param: Param = None)[源代码]