diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py index e7a676e93..cfb1d8f12 100644 --- a/testdata/dnn/onnx/generate_onnx_models.py +++ b/testdata/dnn/onnx/generate_onnx_models.py @@ -2441,3 +2441,13 @@ def generate_onnx_single_operator(single_op, onnx_name, save_prefix="./models"): ) generate_onnx_single_operator(tile, "tile") + + +################# GELU ################# + +x = torch.randn(1, 5, 20) +gelu = nn.GELU() +save_data_and_model("gelu", x, gelu) + +gelu_approximation = nn.GELU('tanh') +save_data_and_model("gelu_approximation", x, gelu_approximation)