Paddle
Paddle copied to clipboard
Add symbolic shape deduction function for unfold, scatter_nd_add, reciprocal, pad3d
PR types
Others
PR changes
Others
Describe
通用plugin机制中,为unfold, scatter_nd_add, reciprocal, pad3d 增加符号化shape 推导函数,使得通用plugin 机制支持这些Op。
主要修改:
dynamic_shape_infermeta.cc中增加了对应 op 的符号化推导函数:
test_trt_convert_pad3d.py,
test_trt_convert_unfold.py,
test_trt_convert_scatter_nd_add.py,
test_trt_convert_reciprocal.py,为对应的单测。
由于 pad3d 的 padding信息,可能是一个 vector