onnxruntime/tools
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
..
android_custom_build
ci_build [Fix] Add the extra param to match gelu in PyTorch in the contrib symbolic function (#11318) 2022-05-04 10:36:38 -07:00
doc
natvis
nuget
perf_view
python