fix: typo in aten::batch_norm

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Aug 20, 2021 · d47f48f · d47f48f
1 parent 93ec68a
commit d47f48f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/conversion/converters/impl/batch_norm.cpp b/core/conversion/converters/impl/batch_norm.cpp
@@ -14,7 +14,7 @@ void _batch_norm(
     ConversionCtx* ctx,
     const torch::jit::Node* n,
     nvinfer1::ITensor* input,
-    const nvinfer1::Dims32& orig_shape,
+    const nvinfer1::Dims& orig_shape,
     const torch::Tensor& gamma,
     const torch::Tensor& beta,
     const torch::Tensor& mean,