missing handlers: aten::squeeze, aten::rsub

zhijian-liu / torchprofile

A general and accurate MACs / FLOPs profiler for PyTorch models

https://pypi.org/project/torchprofile/

MIT License

571 stars 39 forks source link

Closed laubeke closed 4 years ago

laubeke commented 4 years ago

Currently there are no handlers for aten::rsub and aten::squeeze, would be nice to have them!

zhijian-liu commented 4 years ago

Hi, I've just updated the handlers to support these two operations. Please let me know if you have any other questions. Thanks!