[ET-VK] Generalize MeanToSumDiv to any dtype

This change is required for fp16 models.

Differential Revision: [D58040777](https://our.internmc.facebook.com/intern/diff/D58040777/)

[ghstack-poisoned]

Author: jorgep31415

backends/transforms/mean_to_sum_div.py

@@ -19,6 +19,7 @@ def call_operator(self, op, args, kwargs, meta):
         )
         # args[0] is the input tensor
         shape = args[0].node.meta["val"].shape
+        dtype = args[0].node.meta["val"].dtype
         dims_to_reduce = args[1]
         size = 1.0
         for dim in dims_to_reduce:
@@ -32,7 +33,7 @@ def call_operator(self, op, args, kwargs, meta):
                 ],
                 size,
             ),
-            {"dtype": torch.float32},
+            {"dtype": dtype},
             meta,
         )