Skip to content

Commit 5ddab8b

Browse files
committed
fix(core/conversion/converters/impl): fix error message in interpolate
Signed-off-by: Abhiram Iyer <[email protected]> Signed-off-by: Abhiram Iyer <[email protected]>
1 parent 0daa237 commit 5ddab8b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

Diff for: core/conversion/converters/impl/interpolate.cpp

+3-3
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ auto interpolate_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns()
3737
auto layer_output = ctx->AssociateValueAndTensor(n->outputs()[0], resize_layer->getOutput(0));
3838
LOG_DEBUG("Output tensor shape: " << layer_output->getDimensions());
3939
} else {
40-
TRTORCH_THROW_ERROR("Unable to retrieve all node inputs for node: " << util::node_info(n) << "\nScale factor parameter not supported yet");
40+
TRTORCH_THROW_ERROR("Unable to convert node: " << util::node_info(n) << "\nScale factor parameter for upsample_nearest1d not supported yet.");
4141
}
4242

4343
return true;
@@ -67,7 +67,7 @@ auto interpolate_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns()
6767
auto layer_output = ctx->AssociateValueAndTensor(n->outputs()[0], resize_layer->getOutput(0));
6868
LOG_DEBUG("Output tensor shape: " << layer_output->getDimensions());
6969
} else {
70-
TRTORCH_THROW_ERROR("Unable to retrieve all node inputs for node: " << util::node_info(n) << "\nScale factor parameter not supported yet");
70+
TRTORCH_THROW_ERROR("Unable to convert node: " << util::node_info(n) << "\nScale factor parameter for upsample_nearest2d not supported yet.");
7171
}
7272

7373
return true;
@@ -97,7 +97,7 @@ auto interpolate_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns()
9797
auto layer_output = ctx->AssociateValueAndTensor(n->outputs()[0], resize_layer->getOutput(0));
9898
LOG_DEBUG("Output tensor shape: " << layer_output->getDimensions());
9999
} else {
100-
TRTORCH_THROW_ERROR("Unable to retrieve all node inputs for node: " << util::node_info(n) << "\nScale factor parameter not supported yet");
100+
TRTORCH_THROW_ERROR("Unable to convert node: " << util::node_info(n) << "\nScale factor parameter for upsample_nearest3d not supported yet.");
101101
}
102102

103103
return true;

0 commit comments

Comments
 (0)