Ramiro Leal-Cavazos e6675a50d3 Add support for dtype argument in reduction ops ... Many reduction ops take as an argument an optional output dtype that can change the type of the input tensor before the reduction is performed. This commit adds support for the optional dtype flag that had been previously ignored. Test: /tools/torchscript_e2e_test.sh -f 'ReduceSumDtype' /tools/torchscript_e2e_test.sh -f 'ReduceSumDImIntListDtype'		2021-11-30 12:53:59 -05:00
..
AdjustCallingConventions.cpp	Add support for multiple return values	2021-11-16 21:07:45 -05:00
CMakeLists.txt	E2e support for aten.softmax.int and aten.embedding	2021-10-18 17:57:45 -04:00
DecomposeComplexOps.cpp	[TORCH][MLIR] Add E2E support for aten._softmax operation. (#431)	2021-11-25 11:19:02 +05:30
GlobalizeObjectGraph.cpp	Dual license the torch-mlir project.	2021-10-01 10:46:08 -07:00
InlineGlobalSlots.cpp	Dual license the torch-mlir project.	2021-10-01 10:46:08 -07:00
MaximizeValueSemantics.cpp	[MLIR][TORCH] Add E2E support for `aten.squeeze` op	2021-11-30 23:00:28 +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	Dual license the torch-mlir project.	2021-10-01 10:46:08 -07:00
RefinePublicReturn.cpp	Dual license the torch-mlir project.	2021-10-01 10:46:08 -07:00
RefineTypes.cpp	Add support for dtype argument in reduction ops	2021-11-30 12:53:59 -05:00