【BUPT】【Paddle TensorRT No.38】Add pd_op.gather converter #69714
Conversation
|
你的PR提交成功，感谢你对开源项目的贡献! |
| @@ -70,6 +70,15 @@ def reshape_converter(network, paddle_op, inputs): | |||
| return shuffle_layer.get_output(0) | |||
|
|
|||
|
|
|||
| @converter_registry.register("pd_op.gather", trt_version="8.x") | |||
| def gather_converter(network, paddle_op, inputs): | |||
| input_tensor, index_tensor, *_ = inputs | |||
There was a problem hiding this comment.
参考paddle/fluid/inference/tensorrt/convert/gather_op.cc
There was a problem hiding this comment.
index和axis还有pir::value的情况，这里没有实现，另外，axis是输入，而不是属性
| @@ -70,6 +70,15 @@ def reshape_converter(network, paddle_op, inputs): | |||
| return shuffle_layer.get_output(0) | |||
|
|
|||
|
|
|||
| @converter_registry.register("pd_op.gather", trt_version="8.x") | |||
| def gather_converter(network, paddle_op, inputs): | |||
| input_tensor, index_tensor, *_ = inputs | |||
There was a problem hiding this comment.
index和axis还有pir::value的情况，这里没有实现，另外，axis是输入，而不是属性
| self.api_args = { | ||
| "x": np.random.random([3, 4, 10]).astype("float32"), | ||
| "index": np.array([0, 2]).astype("int64"), | ||
| "axis": 1, |
There was a problem hiding this comment.
axis还有pir::value的情况，用np.array组网，放到feed_list中
|
Sorry to inform you that 46edbd0's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually. |
| @@ -70,6 +70,17 @@ def reshape_converter(network, paddle_op, inputs): | |||
| return shuffle_layer.get_output(0) | |||
|
|
|||
|
|
|||
| @converter_registry.register("pd_op.gather", trt_version="8.x") | |||
| def gather_converter(network, paddle_op, inputs): | |||
| input_tensor, index_tensor, axis = inputs | |||
There was a problem hiding this comment.
if xxx.get_defining_op():
拿到value
else:
axis=inputs[2]
| "index": np.array([0, 2]).astype("int64"), | ||
| "axis": np.array([1]).astype("int32"), | ||
| } | ||
| self.program_config = {"feed_list": ["x", "index", "axis"]} |
PR Category
User Experience
PR Types
New features
Description
新增pd_op.gather