add_gather_nd_

PaddlePaddle · Nov 24, 2022 · fb3efec · fb3efec
1 parent fee9040
commit fb3efec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/inference/tensorrt/convert/gather_nd_op.cc b/paddle/fluid/inference/tensorrt/convert/gather_nd_op.cc
@@ -27,7 +27,7 @@ class GatherNdOpConverter : public OpConverter {
     framework::OpDesc op_desc(op, nullptr);
     auto input = engine_->GetITensor(op_desc.Input("X")[0]);
     auto index = engine_->GetITensor(op_desc.Input("Index")[0]);
-    auto output_name = op_desc.Output("Out").front();
+    auto output_name = op_desc.Output("Out")[0];
 
     // AddGatherV2 is supported by the trt version of 8.2.
 #if IS_TRT_VERSION_GE(8200)