    def test_input_dynamic_shape(
        self,
        shape=(2, 6),
        test_name="masked_select_dynamic",
        dtype="float16",
        benchmark=False,
    ):
        """
        Check that dynamic input shape is handled correctly.
        """
        dyn_shape = (IntVar(values=(1, 10)), IntVar(values=(1, 10)))
        self._test_broadcastable_dynamic_masked_select(
            input_dynamic_shape=dyn_shape,
            mask_dynamic_shape=dyn_shape,
            input_shape=shape,
            mask_shape=shape,
            test_name=test_name,
            dtype=dtype,
            benchmark=benchmark,
        )
Check that dynamic input shape is handled correctly.
tests/unittest/ops/test_masked_select.py