Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • F-G Fernandez's avatar
    Added eps attribute to FrozenBatchNorm2d (#2190) · 7a2d0618
    F-G Fernandez authored 
    * feat: Added eps argument to FrozenBatchNorm2d

* test: Added unittest for eps addition in FrozenBatchNorm2d

See #2169

* fix: Reverted forward changes for JIT fuser

* fix: Added back n argument for backward-compatibility

* fix: Fixed FrozenBatchNorm2d forward

Added back eps

* feat: Specified deprecation warnings in FrozenBatchNorm2d

* test: Added unittest for deprecation warninig in FrozenBatchNorm2d

* style: Fixed lint

* style: Fixed block comment lint
    7a2d0618
test_ops.py 24.8 KB