torchprofile icon indicating copy to clipboard operation
torchprofile copied to clipboard

No handlers found: "aten::zeros" and "aten::lstm" Skipped.

Open un-knight opened this issue 5 years ago • 4 comments

hope to support "aten::zeros" and "aten::lstm" ops, thanks!

un-knight avatar Dec 24 '19 09:12 un-knight

"aten::zeros" does not require any mult-adds and therefore will not affect the final result. I'm currently working on supporting "aten::rnn" and "aten::lstm" and will let you know once they are supported.

zhijian-liu avatar Dec 27 '19 20:12 zhijian-liu

Thanks! that would be really cool.

un-knight avatar Dec 28 '19 06:12 un-knight

"aten::zeros" does not require any mult-adds and therefore will not affect the final result. I'm currently working on supporting "aten::rnn" and "aten::lstm" and will let you know once they are supported.

Does this tool support pytorch models like RNNs and some customized modules? If a model is not supported, will the tool output an error message or an inaccurate result. Thank you very much for any guidance.

zwlanpishu avatar Oct 25 '21 13:10 zwlanpishu

It currently does not support RNNs. You will see a warning if the module is not supported.

zhijian-liu avatar Oct 31 '21 02:10 zhijian-liu