Method get_upper_bound_inputs

exir/tests/control_flow_models.py:101–102

(self)

        return torch_map(f, xs, y) + xs

    def get_upper_bound_inputs(self):
        return torch.rand(4, 4), torch.rand(4)

    def get_random_inputs(self):
        return torch.rand(2, 4), torch.rand(4)