Skip to content

Commit

Permalink
Add _native_batch_norm_legit_no_training to core IR
Browse files Browse the repository at this point in the history
Summary: Added due to how common the op is. For performance reasons users may not want to decompose batch_norm op. batch_norm is also part of StableHLO

Test Plan: After adding to IR, we can enable _check_ir_validity in exir.EdgeCompileConfig for models like MV2, MV3, IC3, IC4

Reviewed By: guangy10

Differential Revision: D48576866

fbshipit-source-id: f0c52908028d07fede15702d614dd7a3b27124b1
  • Loading branch information
kirklandsign authored and facebook-github-bot committed Aug 22, 2023
1 parent b282787 commit 1ac3bb7
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions aten/src/ATen/native/native_functions.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4111,6 +4111,7 @@
dispatch:
CompositeExplicitAutograd: _batch_norm_legit_no_training
autogen: _native_batch_norm_legit_no_training.out
tags: core

- func: _native_batch_norm_legit.out(Tensor input, Tensor? weight, Tensor? bias, Tensor(a!) running_mean, Tensor(b!) running_var, bool training, float momentum, float eps, *, Tensor(d!) out, Tensor(e!) save_mean, Tensor(f!) save_invstd) -> (Tensor(d!), Tensor(e!), Tensor(f!))
dispatch:
Expand Down

0 comments on commit 1ac3bb7

Please sign in to comment.