torch-mlir/lib/Dialect/Torch
Daniel Garvey 539511c19b
Add dropout op (#436)
Co-authored-by: dan <dan@nod-labs.com>
2021-11-29 12:30:03 -06:00
..
IR Add support for multiple return values 2021-11-16 21:07:45 -05:00
Transforms Add dropout op (#436) 2021-11-29 12:30:03 -06:00
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