Gaurav Shukla 442ff4605c [LINALG] Decompose `aten.batch_norm` into `aten.native_batch_norm` ... - This commit decomposes the `aten.batch_norm` op into the `aten.native_batch_norm` op, instead of lowering it to the `linalg.generic` op. - It also adds run-time asserts in the `aten.native_batch_norm` lowering to make sure that the shape of the weight, bias, running_mean, and running_var must match the num of features. - Since the `aten.native_batch_norm` op is not supported at TOSA backend, all the modules that are dependent on the `aten.native_batch_norm` op will fail and therefore they should be removed from the TOSA `passing` set. - It also moves `checkNotNone` to utility. Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>		2022-02-16 23:41:38 +05:30
..
torch-mlir	[LINALG] Decompose `aten.batch_norm` into `aten.native_batch_norm`	2022-02-16 23:41:38 +05:30
torch-mlir-c	[Torch][Linalg] Add basic support for RNG	2022-01-31 18:56:42 -05:00
CMakeLists.txt	Move external/torch-mlir to the root of the repo.	2021-09-27 17:11:08 -07:00