Skip to content

parameterized_weighted_summation_fusion

Bases: weighted_summation_fusion

Source code in tinybig/fusion/basic_fusion.py 
66
67
68
class parameterized_weighted_summation_fusion(weighted_summation_fusion):
    def __init__(self, dims: list[int] | tuple[int] = None, name: str = "parameterized_weighted_summation_fusion", require_parameters: bool = True, *args, **kwargs):
        super().__init__(dims=dims, weights=None, name=name, require_parameters=True, *args, **kwargs)