torchprofile
torchprofile copied to clipboard
No handlers found: "aten::zeros" and "aten::lstm" Skipped.
hope to support "aten::zeros" and "aten::lstm" ops, thanks!
"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.
Thanks! that would be really cool.
"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.
It currently does not support RNNs. You will see a warning if the module is not supported.