search

Tencent / TPAT

TensorRT Plugin Autogen Tool
Apache License 2.0
365 stars 42 forks source link

Does TPAT support grid_sample? #14

Closed dingjingzhen closed 2 years ago

dingjingzhen commented 2 years ago

Does TPAT support grid_sample, and if not, when will it support it?

buptqq commented 2 years ago

You can find all of supported operators in Operators. GridSample is not supported now. It is expected to take at least one month to support GridSample, if it is done in advance we will let you know here

buptqq commented 2 years ago

Does TPAT support grid_sample, and if not, when will it support it?

hi @dingjingzhen , We have supported grid-sample operator.Pleace try it: https://github.com/Tencent/TPAT/tree/dev Don't forget update submodule: https://github.com/Tencent/BlazerML-tvm/tree/9dd9e33509137ae4011c0524d5312c41a6b82182