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 Remove Rocm 4.2 from CI Build (#11130) 2022-04-07 11:42:09 -07:00
azure_scale_set_vm_mount_test_data.sh move training CI agent pools to 1ES hosted (#8775) 2021-08-18 18:36:19 -07:00
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