Paddle icon indicating copy to clipboard operation
Paddle copied to clipboard

Add symbolic shape deduction function for unfold, scatter_nd_add, reciprocal, pad3d

Open weishengying opened this issue 3 years ago • 0 comments

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输入,这种情形下,通用plugin暂不支持,op_teller中返回 false。

weishengying avatar Sep 20 '22 08:09 weishengying