How to locate the 'aten::_native_multi_head_attention' operator

[liuy061530]

Traceback (most recent call last):
File "/mnt/d/projects/GoalFlow//navsim/planning/script/pth2onnx.py", line 136, in main
lightning_module.to_onnx(onnx_path, dummy_inputs, export_params=True, opset_version=15)
File "/home/dl/anaconda3/envs/goalflow/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/home/dl/anaconda3/envs/goalflow/lib/python3.10/site-packages/pytorch_lightning/core/module.py", line 1405, in to_onnx
torch.onnx.export(self, input_sample, file_path, **kwargs)
File "/home/dl/anaconda3/envs/goalflow/lib/python3.10/site-packages/torch/onnx/utils.py", line 506, in export
_export(
File "/home/dl/anaconda3/envs/goalflow/lib/python3.10/site-packages/torch/onnx/utils.py", line 1548, in _export
graph, params_dict, torch_out = _model_to_graph(
File "/home/dl/anaconda3/envs/goalflow/lib/python3.10/site-packages/torch/onnx/utils.py", line 1117, in _model_to_graph
graph = _optimize_graph(
File "/home/dl/anaconda3/envs/goalflow/lib/python3.10/site-packages/torch/onnx/utils.py", line 665, in _optimize_graph
graph = _C._jit_pass_onnx(graph, operator_export_type)
File "/home/dl/anaconda3/envs/goalflow/lib/python3.10/site-packages/torch/onnx/utils.py", line 1901, in _run_symbolic_function
raise errors.UnsupportedOperatorError(
torch.onnx.errors.UnsupportedOperatorError: Exporting the operator 'aten::_native_multi_head_attention' to ONNX opset version 15 is not supported. Please feel free to request support or submit a pull request on PyTorch GitHub: https://github.com/pytorch/pytorch/issues.