onnxruntime/tools/ci_build/github/linux/docker/scripts/training
Justin Chu a1f9847b23
[Fix] Add the extra param to match gelu in PyTorch in the contrib symbolic function (#11318)
Description:

Add the extra param to match gelu in PyTorch in the contrib symbolic function

Motivation and Context

Why is this change required? What problem does it solve?
The symbolic function in /onnxruntime/python/tools/pytorch_export_contrib_ops.py is missing a recently added parameter approximate. We add this parameter and use the exporter defined gelu if approximate is "tanh".
2022-05-04 10:36:38 -07:00
..
ortmodule
azure_scale_set_vm_mount_test_data.sh
requirements.txt [Fix] Add the extra param to match gelu in PyTorch in the contrib symbolic function (#11318) 2022-05-04 10:36:38 -07:00