Source code for nv_ingest.util.morpheus.linear_module_source_stage_cpu

from morpheus.config import ExecutionMode
from morpheus.stages.general.linear_modules_source import LinearModuleSourceStage
from morpheus.stages.general.linear_modules_stage import LinearModulesStage



[docs]
class LinearModuleSourceStageCPU(LinearModuleSourceStage):

[docs]
    def supported_execution_modes(self) -> tuple[ExecutionMode]:
        # Provide your own logic here; for example:
        return (ExecutionMode.CPU,)





[docs]
class LinearModuleStageCPU(LinearModulesStage):

[docs]
    def supported_execution_modes(self) -> tuple[ExecutionMode]:
        # Provide your own logic here; for example:
        return (ExecutionMode.CPU,)