diff --git a/exir/dialects/edge/dtype/supported.py b/exir/dialects/edge/dtype/supported.py index f0772ab4891..89c269de1d0 100644 --- a/exir/dialects/edge/dtype/supported.py +++ b/exir/dialects/edge/dtype/supported.py @@ -25,6 +25,7 @@ torch.float16: "Half", torch.float: "Float", torch.double: "Double", + torch.uint16: "UInt16", } regular_tensor_str_to_dtypes = {