Quantized integers to Torchx

torchx/lib/torchx/backend.ex

@@ -223,6 +223,14 @@ defmodule Torchx.Backend do
     for <<x::native-size(size) <- bin>>, into: <<>>, do: <<x::native-size(double_size)>>
   end
 
+  defp maybe_pad_binary(bin, {:u, size}) when size in [2, 4] do
+    for <<x::native-size(size) <- bin>>, into: <<>>, do: <<x::native-8>>
+  end
+
+  defp maybe_pad_binary(bin, {:s, size}) when size in [2, 4] do
+    for <<x::native-signed-size(size) <- bin>>, into: <<>>, do: <<x::native-signed-8>>
+  end
+
   defp maybe_pad_binary(bin, _), do: bin
 
   ## Shape
@@ -1746,6 +1754,12 @@ defmodule Torchx.Backend do
       current_type = Torchx.scalar_type(device_ref) |> from_torch_type()
 
       case {current_type, type} do
+        {{:s, 8}, {:s, qint}} when qint in [2, 4] ->
+          :ok
+
+        {{:u, 8}, {:u, qint}} when qint in [2, 4] ->
+          :ok
+
         {{:s, 32}, {:u, 16}} ->
           :ok