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
..
torchscript	Add support for dtype argument in reduction ops	2021-11-30 12:53:59 -05:00