Skip to content

Commit c828a4b

Browse files
committed
[TRT] Disable out-dated TRT PadV2 Plugin
1 parent 1d535ec commit c828a4b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

source/tnn/network/tensorrt/layer_builder/pad_v2_layer_builder.cc

+3-3
Original file line numberDiff line numberDiff line change
@@ -74,9 +74,9 @@ ILayer* PadV2TRTPluginLayerBuilder::AddToNetwork(INetworkDefinition* network) no
7474
return nullptr;
7575
}
7676

77-
if (!UseTRTPaddingND(paramlist)) {
78-
return TensorRTPluginLayerBuilder::AddToNetwork(network);
79-
}
77+
//if (!UseTRTPaddingND(paramlist)) {
78+
// return TensorRTPluginLayerBuilder::AddToNetwork(network);
79+
//}
8080

8181
auto input_foreign_tensor = dynamic_cast<ForeignBlob*>(input_blobs_[0])->GetForeignTensor();
8282
auto input_tensor = std::dynamic_pointer_cast<TensorRTTensor>(input_foreign_tensor)->GetTensor();

0 commit comments

Comments
 (0)