tf.meshgrid not working with tf.function
Are you satisfied with the resolution of your issue?
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=Yes&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/60543">Yes</a>
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=No&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/60543">No</a>

<details><summary>Click to expand!</summary> 
 
 ### Issue Type

Bug

### Have you reproduced the bug with TF nightly?

No

### Source

source

### Tensorflow Version

2.4.4

### Custom Code

Yes

### OS Platform and Distribution

Windows 10

### Mobile device

_No response_

### Python version

3.8

### Bazel version

_No response_

### GCC/Compiler version

_No response_

### CUDA/cuDNN version

_No response_

### GPU model and memory

_No response_

### Current Behaviour?

When running the code, I would expect it to pass without error. But I am getting the following error. When deleting the @tf.function decorator, it works as expected

``` 
Traceback (most recent call last):
  File "C:\Users\Josef.ondrej\AppData\Roaming\JetBrains\PyCharm2022.3\scratches\scratch_236.py", line 11, in <module>
    print(my_function())
  File "C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\eager\def_function.py", line 828, in __call__
    result = self._call(*args, **kwds)
  File "C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\eager\def_function.py", line 871, in _call
    self._initialize(args, kwds, add_initializers_to=initializers)
  File "C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\eager\def_function.py", line 725, in _initialize
    self._stateful_fn._get_concrete_function_internal_garbage_collected(  # pylint: disable=protected-access
  File "C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\eager\function.py", line 2969, in _get_concrete_function_internal_garbage_collected
    graph_function, _ = self._maybe_define_function(args, kwargs)
  File "C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\eager\function.py", line 3361, in _maybe_define_function
    graph_function = self._create_graph_function(args, kwargs)
  File "C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\eager\function.py", line 3196, in _create_graph_function
    func_graph_module.func_graph_from_py_func(
  File "C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\framework\func_graph.py", line 990, in func_graph_from_py_func
    func_outputs = python_func(*func_args, **func_kwargs)
  File "C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\eager\def_function.py", line 634, in wrapped_fn
    out = weak_wrapped_fn().__wrapped__(*args, **kwds)
  File "C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\framework\func_graph.py", line 977, in wrapper
    raise e.ag_error_metadata.to_exception(e)
NotImplementedError: in user code:

    C:\Users\Josef.ondrej\AppData\Roaming\JetBrains\PyCharm2022.3\scratches\scratch_236.py:8 my_function  *
        tf.meshgrid(b, a)
    C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\util\dispatch.py:201 wrapper  **
        return target(*args, **kwargs)
    C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\ops\array_ops.py:3552 meshgrid
        mult_fact = ones(shapes, output_dtype)
    C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\util\dispatch.py:201 wrapper
        return target(*args, **kwargs)
    C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\ops\array_ops.py:3120 ones
        output = _constant_if_small(one, shape, dtype, name)
    C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\ops\array_ops.py:2804 _constant_if_small
        if np.prod(shape) < 1000:
    <__array_function__ internals>:180 prod
        
    C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\numpy\core\fromnumeric.py:3088 prod
        return _wrapreduction(a, np.multiply, 'prod', axis, dtype, out,
    C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\numpy\core\fromnumeric.py:86 _wrapreduction
        return ufunc.reduce(obj, axis, dtype, out, **passkwargs)
    C:\Users\Josef.ondrej\Anaconda3\envs\foobar-env\lib\site-packages\tensorflow\python\framework\ops.py:852 __array__
        raise NotImplementedError(

    NotImplementedError: Cannot convert a symbolic Tensor (meshgrid/Size_1:0) to a numpy array. This error may indicate that you're trying to pass a Tensor to a NumPy call, which is not supported

```

### Standalone code to reproduce the issue

```shell
import tensorflow as tf

@tf.function
def my_function():
    a = tf.constant([1.0])
    b = tf.constant([1.0])
    tf.meshgrid(b, a)
```


### Relevant log output

_No response_</details>