def METHOD_NAME(self, x: Tensor) -> Tensor:
    x_in = x
    for module in self.children():
        if hasattr(module, "forward_steps"):
            x = module.METHOD_NAME(x)
        else:
            x = module(x)
        if not isinstance(x, Tensor):
            return None
    return x_in * x