llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python/torch_mlir
History
Vivek Khandelwal 3589134d31 [MLIR][TORCH] Add decomposition for aten.var.dim op 
This commit adds the decomposition for `aten.var.dim` op.
This commit also make changes in the decomposition for `aten.var` op.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 		2022-07-15 09:53:42 +05:30
..
_torch_mlir_custom_op_example Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
dialects [MLIR][TORCH] Add decomposition for aten.var.dim op 2022-07-15 09:53:42 +05:30
eager_mode Link against libtorch (#955) 2022-06-30 12:40:17 -07:00
__init__.py torch_mlir.compile: Allow OutputType as a string. 2022-07-08 17:37:27 -07:00
compiler_utils.py Update debug options in compilation errors (#913) 2022-06-07 10:55:54 -07:00