nod-ai / sharktank

SHARK Inference Modeling and Serving
Apache License 2.0
7 stars 9 forks source link

Insert padding for conv and pooling ops before bit-width extend operations of operands #77

Closed aviator19941 closed 2 weeks ago

aviator19941 commented 3 weeks ago

Moves the padding for conv and pooling kernels before extsi operations of operands. Fixes https://github.com/nod-ai/SHARK-Turbine/issues/750