add generator

opencv · Feb 4, 2023 · d6263b1 · d6263b1
1 parent 7c5a8b7
commit d6263b1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git 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)