torch-mlir/lib/Dialect/Torch/Transforms
Prashant Kumar ccf546f14c Add aten::nll_loss_backward op
The lowering of aten::nll_loss_backward op has been added
from torch to linalg dialect. The changes has been made as
a part of -torch-convert-to-linalg pass.

Signed-off-by: Prashant Kumar prashant@nod-labs.com
2022-02-04 21:57:53 +05:30
..
AdjustCallingConventions.cpp Bump LLVM to 881ff4e4ebe8cc0cc045c7c167cffb01f94f27f8 (#539) 2022-01-25 22:16:30 -08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00
DecomposeComplexOps.cpp Modify softmax decomposition to be more numerically stable. 2022-02-03 21:20:36 +05:30
GlobalizeObjectGraph.cpp Bump LLVM to 84fe34a0b7fdd7bbf179981d1583693d5d5ec68b (#544) 2022-01-27 17:21:09 -08:00
InlineGlobalSlots.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
MaximizeValueSemantics.cpp [TORCH][MLIR] Add E2E support for `aten.reshape` op 2022-02-02 20:41:47 +05:30
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp Add support for multiple return values 2021-11-16 21:07:45 -05:00
PrepareForGlobalizeObjectGraph.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
ReduceOpVariants.cpp [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
RefinePublicReturn.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
RefineTypes.cpp Add aten::nll_loss_backward op 2022-02-04 21:57:53 +05:30