Fix to_custom_tensor fallback to take two inputs (quic#3552)

Signed-off-by: Sai Chaitanya Gajula <quic_gsaichai@quicinc.com>

TrainingExtensions/torch/src/python/aimet_torch/custom/custom_tensor_utils.py

@@ -43,9 +43,9 @@ def to_torch_tensor(tensors):
         """ placeholder in case spconv doesn't exist """
         return tensors
 
-    def to_custom_tensor(tensors):
+    def to_custom_tensor(original, torch_tensors):
         """ placeholder in case spconv doesn't exist """
-        return tensors
+        return original
 else:
     from typing import List, Union, Tuple
     import torch