File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
core/conversion/converters/impl Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -161,7 +161,7 @@ auto acthardtanh TRTORCH_UNUSED =
161
161
TRTORCH_CHECK (new_layer, " Unable to create layer for aten::elu" );
162
162
new_layer->setAlpha (alpha);
163
163
164
- new_layer->setName (trtorch::core:: util::node_info (n).c_str ());
164
+ new_layer->setName (util::node_info (n).c_str ());
165
165
166
166
auto out_tensor = ctx->AssociateValueAndTensor (n->outputs ()[0 ], new_layer->getOutput (0 ));
167
167
LOG_DEBUG (" Output shape: " << out_tensor->getDimensions ());
@@ -190,7 +190,7 @@ auto acthardtanh TRTORCH_UNUSED =
190
190
TRTORCH_CHECK (gelu_plugin, " Unable to create gelu plugin from TensorRT plugin registry" << *n);
191
191
auto new_layer =
192
192
ctx->net ->addPluginV2 (reinterpret_cast <nvinfer1::ITensor* const *>(&in), 1 , *gelu_plugin);
193
- new_layer->setName (" gelu " );
193
+ new_layer->setName (util::node_info (n). c_str () );
194
194
auto out_tensor = new_layer->getOutput (0 );
195
195
out_tensor = ctx->AssociateValueAndTensor (n->outputs ()[0 ], out_tensor);
196
196
LOG_DEBUG (" Output shape: " << out_tensor->getDimensions ());
You can’t perform that action at this time.
0 commit comments