Skip to content

Commit 178eab3

Browse files
committed
replace expandedDimsTensor->getDimensions().d[0] with expandedDims.nbDims
Signed-off-by: Ruoqian Guo <[email protected]>
1 parent bf5718d commit 178eab3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

core/conversion/converters/impl/expand.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ bool add_expand_dynamic(
111111
bool is_expand_layer) {
112112
auto input_dims = in->getDimensions();
113113
auto input_rank = in->getDimensions().nbDims;
114-
auto output_rank = expandedDimsTensor->getDimensions().d[0];
114+
auto output_rank = expandedDims.nbDims;
115115
TRTORCH_CHECK(
116116
input_rank <= output_rank,
117117
"Number of dimensions of the desired expansion must be greater than or equal to the number of input dimensions");

0 commit comments

Comments
 (0)