fix(aten::expand): Fix compiler warning for unused out ITensor

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Feb 23, 2021 · 5b0f584 · 5b0f584
1 parent 178eab3
commit 5b0f584
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/conversion/converters/impl/expand.cpp b/core/conversion/converters/impl/expand.cpp
@@ -281,7 +281,7 @@ auto expand_registrations TRTORCH_UNUSED =
 
                auto out = ctx->AssociateValueAndTensor(n->outputs()[0], in);
 
-               LOG_DEBUG("Repeat layer output tensor shape: " << in->getDimensions());
+               LOG_DEBUG("Repeat layer output tensor shape: " << out->getDimensions());
                return true;
              }});