We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7ce66ea commit e099dbbCopy full SHA for e099dbb
py/torch_tensorrt/dynamo/conversion/ops_evaluators.py
@@ -73,7 +73,6 @@ def aten_ops_rand(
73
kwargs: Dict[str, Argument],
74
name: str,
75
) -> Union[TRTTensor, Sequence[TRTTensor]]:
76
- device = kwargs.get("device", None)
77
return np.random.rand(*args)
78
79
@@ -87,7 +86,6 @@ def aten_ops_randn(
87
86
88
89
90
91
return np.random.randn(*args)
92
93
@@ -119,5 +117,4 @@ def aten_ops_randperm(
119
117
120
118
121
122
123
return np.random.permutation(*args)
0 commit comments