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 b72d2ae commit 22db4b7Copy full SHA for 22db4b7
core/conversion/converters/impl/unsqueeze.cpp
@@ -22,10 +22,10 @@ auto unsqueeze_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().p
22
23
auto selfDim = util::toVec(self->getDimensions());
24
int64_t nbDims = selfDim.size();
25
- TRTORCH_ASSERT(
+ TRTORCH_CHECK(
26
dim <= nbDims && dim >= -(nbDims + 1),
27
- "imension out of range (expected to be in range of [" << -(nbDims + 1) << ", " << nbDims << "], but got "
28
- << dim << ")");
+ "Dimension out of range (expected to be in range of [" << -(nbDims + 1) << ", " << nbDims << "], but got "
+ << dim << ")");
29
if (dim < 0) {
30
dim = nbDims + 1 + dim;
31
}
0 commit comments