def test_where_scalarself(self, compute_unit, backend, frontend):
        """Test torch.ops.aten.where.ScalarSelf in torch.export"""
        INVALID_LOGIT_BIAS = -40000.0

        class Model(torch.nn.Module):
            def forward(self, x):
                return torch.where(x != INVALID_LOGIT_BIAS, 0.0, x)

        self.run_compare_torch(
            [torch.zeros(1, 2048, 1, 48)],
            Model(),
            compute_unit=compute_unit,
            frontend=frontend,
            backend=backend,
            input_as_shape=False,
        )
Test torch.ops.aten.where.ScalarSelf in torch.export
coremltools/converters/mil/frontend/torch/test/test_torch_ops.py