torch-mlir/lib/Dialect/Torch
Vivek Khandelwal ca662dc9cc [MLIR][TORCH] Add E2E support for aten.threshold, aten.threshold_backward op
This commit adds lowering of `aten.threshold` op
This commit adds lowering of `aten.threshold_backward` op

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-01-10 11:56:56 +05:30
..
IR [TORCH][MLIR] Fold trivial cases of `aten.to.dtype` and `aten.view` op 2021-12-24 13:32:34 +05:30
Transforms [MLIR][TORCH] Add E2E support for aten.threshold, aten.threshold_backward op 2022-01-10 11:56:56 +05:30
Utils Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00