mirror of https://github.com/llvm/torch-mlir
3c40539b34
- This commit adds E2E support for `aten.ones_like` and `aten.zeros_like` ops. - Adds support for non-None `dtype` argument of `aten.empty_like` op. - All the unit test cases related to constant tensor allocation like ops are moved to a different file named `constant_alloc.py`. Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com> |
||
---|---|---|
.. | ||
test | ||
torch_mlir/dialects | ||
torch_mlir_e2e_test | ||
CMakeLists.txt | ||
TorchMLIRModule.cpp |