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 621a04a commit 26ed9f1Copy full SHA for 26ed9f1
core/conversion/converters/impl/batch_norm.cpp
@@ -50,8 +50,9 @@ auto batch_norm_registrations TORCHTRT_UNUSED =
50
auto orig_shape = input->getDimensions();
51
auto shape = util::toVec(orig_shape);
52
auto tensor_type = util::TRTDataTypeToScalarType(input->getType());
53
- auto options = torch::TensorOptions().dtype(tensor_type).device(torch::kCUDA, ctx->settings.device.gpu_id);
54
-
+ auto options =
+ torch::TensorOptions().dtype(tensor_type).device(torch::kCUDA, ctx->settings.device.gpu_id);
55
+
56
torch::Tensor gamma, beta, mean, var;
57
LOG_DEBUG("Input :" << orig_shape << "/" << input->getType());
58
// affine=True
0 commit comments