nv_ingest.util.morpheus package#

Submodules#

nv_ingest.util.morpheus.linear_module_source_stage_cpu module#

class nv_ingest.util.morpheus.linear_module_source_stage_cpu.LinearModuleSourceStageCPU(
c: Config,
module_config: Dict | ModuleLoader,
output_port_name: str,
output_type=typing.Any,
)[source]#

Bases: LinearModuleSourceStage

supported_execution_modes() tuple[ExecutionMode][source]#

Returns a tuple of supported execution modes of this stage. By default this returns (ExecutionMode.GPU,). Subclasses can override this method to specify different execution modes.

For most stages the values will be static, and this can be accomplished by making use of either the CpuOnlyMixin or GpuAndCpuMixin mixins.

However, complex stages may choose to make this decision at runtime, in which case this method should be overridden. directly within the stage class.

class nv_ingest.util.morpheus.linear_module_source_stage_cpu.LinearModuleStageCPU(
c: Config,
module_config: Dict | ModuleLoader,
input_port_name: str,
output_port_name: str,
input_type=typing.Any,
output_type=typing.Any,
)[source]#

Bases: LinearModulesStage

supported_execution_modes() tuple[ExecutionMode][source]#

Returns a tuple of supported execution modes of this stage. By default this returns (ExecutionMode.GPU,). Subclasses can override this method to specify different execution modes.

For most stages the values will be static, and this can be accomplished by making use of either the CpuOnlyMixin or GpuAndCpuMixin mixins.

However, complex stages may choose to make this decision at runtime, in which case this method should be overridden. directly within the stage class.

Module contents#