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
..
torchscript	Add torch.aten.batch_norm Linalg lowering support	2021-06-22 16:45:21 -07:00
__init__.py	Add ability to annotate TorchScript classes.	2021-02-25 11:28:34 -08:00