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 101faf4 commit f720276Copy full SHA for f720276
py/torch_tensorrt/dynamo/conversion/impl/unary/ops.py
@@ -469,7 +469,7 @@ def scalar_tensor(
469
source_ir: Optional[SourceIR],
470
name: str,
471
scalar: Union[int, float, bool],
472
- dtype: Union[torch.int, torch.float, torch.bool] = None,
+ dtype: Optional[Union[torch.int, torch.float, torch.bool]],
473
) -> TRTTensor:
474
tensor = get_trt_tensor(ctx, scalar, f"{name}_scalar_tensor", dtype)
475
identity_layer = ctx.net.add_identity(tensor)
0 commit comments