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 bf5718d commit 178eab3Copy full SHA for 178eab3
core/conversion/converters/impl/expand.cpp
@@ -111,7 +111,7 @@ bool add_expand_dynamic(
111
bool is_expand_layer) {
112
auto input_dims = in->getDimensions();
113
auto input_rank = in->getDimensions().nbDims;
114
- auto output_rank = expandedDimsTensor->getDimensions().d[0];
+ auto output_rank = expandedDims.nbDims;
115
TRTORCH_CHECK(
116
input_rank <= output_rank,
117
"Number of dimensions of the desired expansion must be greater than or equal to the number of input dimensions");
0 commit comments