Skip to content

Commit 26ed9f1

Browse files
committed
Fix lint
Signed-off-by: root <[email protected]>
1 parent 621a04a commit 26ed9f1

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

core/conversion/converters/impl/batch_norm.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,8 +50,9 @@ auto batch_norm_registrations TORCHTRT_UNUSED =
5050
auto orig_shape = input->getDimensions();
5151
auto shape = util::toVec(orig_shape);
5252
auto tensor_type = util::TRTDataTypeToScalarType(input->getType());
53-
auto options = torch::TensorOptions().dtype(tensor_type).device(torch::kCUDA, ctx->settings.device.gpu_id);
54-
53+
auto options =
54+
torch::TensorOptions().dtype(tensor_type).device(torch::kCUDA, ctx->settings.device.gpu_id);
55+
5556
torch::Tensor gamma, beta, mean, var;
5657
LOG_DEBUG("Input :" << orig_shape << "/" << input->getType());
5758
// affine=True

0 commit comments

Comments
 (0)