spark.nn.interfaces.base

Attributes

ConfigT

Classes

InterfaceOutput	Generic Interface model output spec.
InterfaceConfig	Abstract Interface model configuration class.
Interface	Abstract Interface model.

Module Contents

class spark.nn.interfaces.base.InterfaceOutput

Bases: TypedDict

Generic Interface model output spec.

Initialize self. See help(type(self)) for accurate signature.

class spark.nn.interfaces.base.InterfaceConfig(__skip_validation__=False, **kwargs)

Bases: spark.core.config.SparkConfig

Abstract Interface model configuration class.

Parameters:

__skip_validation__ (bool)

spark.nn.interfaces.base.ConfigT

class spark.nn.interfaces.base.Interface(config=None, **kwargs)

Bases: spark.core.module.SparkModule, abc.ABC, Generic[ConfigT]

Abstract Interface model.

Parameters:

config (ConfigT | None)

config: ConfigT

abstractmethod __call__(*args, **kwargs)

Computes the control flow operation.

Parameters:

args (spark.core.payloads.SparkPayload)

Return type:

InterfaceOutput