From d6263b10daf9dd643c989f6ddb5070a67826757f Mon Sep 17 00:00:00 2001 From: fengyuentau Date: Sun, 5 Feb 2023 00:20:12 +0800 Subject: [PATCH] add generator --- testdata/dnn/onnx/generate_onnx_models.py | 10 ++++++++++ 1 file changed, 10 insertions(+) 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)