Yi Zhang 6dddb4d4fe Add torch.aten.batch_norm Linalg lowering support ... 1. Added a simplified version of torch.aten.batch_norm which only handles inference and assumes the weight, bias, running_mean, running_var are not None. 2. Removed the primitive types check in verifyLinalgCompatibleTypes check since now we have proper type converter to handle torch types conversion. The checks for RankedTensorType is kept because the type converter doesn't guarantee the converted builtin tensor type is ranked. A separate verification pass to verify the invariant expected by later passes will need to be added before those can be removed as well.		2021-06-22 16:45:21 -07:00
..
torch_mlir	Add torch.aten.batch_norm Linalg lowering support	2021-06-22 16:45:21 -07:00
torch_mlir_utils/codegen	Add a !torch.float type.	2021-06-17 09:24:18 -07:00
CMakeLists.txt	Create public API for torch_mlir python code.	2020-10-13 16:36:49 -07:00