mirror of https://github.com/llvm/torch-mlir
442ff4605c
- 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> |
||
|---|---|---|
| .. | ||
| GlobalizeObjectGraph | ||
| adjust-calling-conventions.mlir | ||
| canonicalize.mlir | ||
| decompose-complex-ops.mlir | ||
| inline-global-slots.mlir | ||
| invalid.mlir | ||
| maximize-value-semantics.mlir | ||
| ops.mlir | ||
| prepare-for-globalize-object-graph.mlir | ||
| promote-types.mlir | ||
| reduce-op-variants-error.mlir | ||
| reduce-op-variants.mlir | ||
| refine-public-return.mlir | ||
| refine-types-branch.mlir | ||
| refine-types.mlir | ||