qualia_codegen_plugin_snn.graph.layers package

Submodules

Module contents

class qualia_codegen_plugin_snn.graph.layers.TIfLayer(input_shape: qualia_codegen_core.typing.Shapes, output_shape: qualia_codegen_core.typing.Shapes, output_dtype: qualia_codegen_core.typing.DTypes, name: str, v_threshold: 'NDArrayFloatOrInt', v_reset: 'NDArrayFloatOrInt | None', soft_reset: 'bool')[source]

Bases: TBaseLayer

v_threshold: NDArrayFloatOrInt

v_reset: NDArrayFloatOrInt | None

soft_reset: bool

property weights: OrderedDict[str, NDArrayFloatOrInt]

class qualia_codegen_plugin_snn.graph.layers.TLifLayer(input_shape: qualia_codegen_core.typing.Shapes, output_shape: qualia_codegen_core.typing.Shapes, output_dtype: qualia_codegen_core.typing.DTypes, name: str, v_threshold: 'NDArrayFloatOrInt', v_reset: 'NDArrayFloatOrInt | None', soft_reset: 'bool', reciprocal_tau: 'NDArrayFloatOrInt', decay_input: 'bool')[source]

Bases: TIfLayer

reciprocal_tau: NDArrayFloatOrInt

decay_input: bool

property weights: OrderedDict[str, NDArrayFloatOrInt]

class qualia_codegen_plugin_snn.graph.layers.TObjectDetectionPostProcessLayer(input_shape: qualia_codegen_core.typing.Shapes, output_shape: qualia_codegen_core.typing.Shapes, output_dtype: qualia_codegen_core.typing.DTypes, name: str, num_classes: 'int', num_fms: 'int', image_shape: 'tuple[int, ...]', score_threshold: 'float', nms_threshold: 'float', topk_candidates: 'int', detections_per_image: 'int', box_coder_weights: 'list[float]', anchors: 'NDArrayFloatOrInt')[source]

Bases: TBaseLayer

num_classes: int

num_fms: int

image_shape: tuple[int, ...]

score_threshold: float

nms_threshold: float

topk_candidates: int

detections_per_image: int

box_coder_weights: list[float]

anchors: NDArrayFloatOrInt

property weights: OrderedDict[str, NDArrayFloatOrInt]