onnxruntime
2e1fa3cc - Fix GeluRecompute for 2 inputs case. (#5573)

Commit
5 years ago
Fix GeluRecompute for 2 inputs case. (#5573) * Add test for FastGelu + GeluRecompute. * Fix GeluRecompute for 2 inputs case. * Fix test for BiasGelu + GeluRecompute. * Copy all inputs to Gelu, not just 2. * Move GeluRecompute test to training-specific file.
Author
Sergii Dymchenko
Parents
Loading