torch-mlir/frontends/pytorch/python
Sean Silva 60a947b4a7 Add CastOpInterface to torch.prim.unchecked_cast.
This allows it to fold away in trivial cases.
2021-06-23 08:07:45 -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 CastOpInterface to torch.prim.unchecked_cast. 2021-06-23 08:07:45 -07:00
CMakeLists.txt Create public API for torch_mlir python code. 2020-10-13 16:36:49 -07:00