diff --git a/py/torch_tensorrt/dynamo/conversion/converter_utils.py b/py/torch_tensorrt/dynamo/conversion/converter_utils.py index 72846c9007..5a757df23e 100644 --- a/py/torch_tensorrt/dynamo/conversion/converter_utils.py +++ b/py/torch_tensorrt/dynamo/conversion/converter_utils.py @@ -175,6 +175,9 @@ def extend_attr_to_tuple( """ if not isinstance(val, (tuple, list)): val = (val,) * num_elem + elif len(val) == 1: + val = (val[0],) * num_elem + if isinstance(val, list): val = tuple(val) return val diff --git a/py/torch_tensorrt/dynamo/conversion/impl/conv.py b/py/torch_tensorrt/dynamo/conversion/impl/conv.py index 78a1276882..285da2a04c 100644 --- a/py/torch_tensorrt/dynamo/conversion/impl/conv.py +++ b/py/torch_tensorrt/dynamo/conversion/impl/conv.py @@ -46,7 +46,7 @@ def convNd( ) # Process bias terms - if isinstance(bias, torch.Tensor): + if isinstance(bias, (torch.Tensor, np.ndarray)): # Transform the bias constant into a Numpy array bias = to_numpy(bias) @@ -67,7 +67,7 @@ def convNd( network, target, (weight, -1), {}, name + "_unsqueeze_weight" ) - elif isinstance(weight, torch.Tensor): + elif isinstance(weight, (torch.Tensor, np.ndarray)): # Transform the weight constant into a Numpy array weight = to_numpy(weight)