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 e7bd9ac commit 20a0716Copy full SHA for 20a0716
core/conversion/converters/impl/reduce.cpp
@@ -71,7 +71,7 @@ auto reduce_registrations TORCHTRT_UNUSED =
71
auto in_tensor = args[0].ITensorOrFreeze(ctx);
72
auto in_dims = util::toVec(in_tensor->getDimensions());
73
LOG_WARNING("Sum Converter disregards dtype");
74
-
+
75
if (in_tensor->getType() == nvinfer1::DataType::kBOOL) {
76
LOG_DEBUG(
77
"Found type " << in_tensor->getType() << " in aten::sum, casting to "
0 commit comments