def test_searchsorted_basic(self, compute_unit, backend, frontend, side):
        """This is the test case same as PyTorch doc for `torch.searchsorted`."""

        if frontend == TorchFrontend.EXECUTORCH:
            pytest.skip("torch._ops.aten.searchsorted.Tensor is not Aten Canonical")

        class TestModel(nn.Module):
            def forward(self, input_data, values):
                return torch.searchsorted(input_data, values, side=side)

        input_data = torch.tensor([[1, 3, 5, 7, 9], [2, 4, 6, 8, 10]])
        values = torch.tensor([[3, 6, 9], [3, 6, 9]])
        self.run_compare_torch(
            (input_data, values),
            TestModel(),
            frontend=frontend,
            backend=backend,
            compute_unit=compute_unit,
            input_as_shape=False,
        )
This is the test case same as PyTorch doc for `torch.searchsorted`.
coremltools/converters/mil/frontend/torch/test/test_torch_ops.py