    def test_fp32_input_broadcast_shape(
        self,
        input_shape,
        mask_shape,
        benchmark=False,
        test_name="masked_select_broadcast_fp32",
        dtype="float32",
    ):
        """
        Check the support for broadcastable input and mask.
        """
        self._test_broadcastable_masked_select(
            input_shape=input_shape,
            mask_shape=mask_shape,
            test_name=test_name,
            dtype=dtype,
            benchmark=benchmark,
        )
Check the support for broadcastable input and mask.
tests/unittest/ops/test_masked_select.py