llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Conversion/TorchToLinalg
History
Vivek Khandelwal 8a06419980 [MLIR][TORCH] Add E2E support for aten.masked_fill.Scalar op 
This commit adds lowering of `aten.masked_fill.Scalar` op.
This commit also fixes the formatting of the file constant_alloc.py.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 		2022-05-02 22:27:33 +05:30
..
CMakeLists.txt Split up TorchToLinalg.cpp 2022-03-14 10:19:41 -07:00
DataMovement.cpp [LINALG] Bug fix i64 vs i32 type comparison. 2022-04-22 08:09:58 +05:30
IndirectDataMovement.cpp [LINALG] Bug fix i64 vs i32 type comparison. 2022-04-22 08:09:58 +05:30
Linear.cpp [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30
Pooling.cpp [MLIR][TORCH] Add E2E support for aten.avg_pool2d op 2022-05-02 12:31:44 +05:30
PopulatePatterns.h Split up TorchToLinalg.cpp 2022-03-14 10:19:41 -07:00
Random.cpp Improve "pseudo" op terminology. 2022-03-15 17:57:52 -07:00
Reduction.cpp Split up TorchToLinalg.cpp 2022-03-14 10:19:41 -07:00
TensorConstructors.cpp [LINALG] Bug fix i64 vs i32 type comparison. 2022-04-22 08:09:58 +05:30
TensorScalarInterop.cpp [LINALG] Bug fix i64 vs i32 type comparison. 2022-04-22 08:09:58 +05:30
TorchToLinalg.cpp llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
Uncategorized.cpp [MLIR][TORCH] Add E2E support for aten.masked_fill.Scalar op 2022-05-02 22:27:33 +05:30
Utils.cpp [LINALG] Bug fix i64 vs i32 type comparison. 2022-04-22 08:09:58 +05:30
Utils.h [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30