tenstorrent / tt-forge-fe

The TT-Forge FE is a graph compiler designed to optimize and transform computational graphs for deep learning models, enhancing their performance and efficiency.
https://docs.tenstorrent.com/tt-forge-fe/
Apache License 2.0
20 stars 3 forks source link

[Model Analysis] Move the unique ops generation to separate module #724

Open chandrasekaranpradeep opened 4 days ago

chandrasekaranpradeep commented 4 days ago

In tvm_to_python.py file, the models unique operation configuration are extracted using the UniqueOperations, UniqueOperationInfo, OpArgsOpNames, OperandsInfo and OpArgs class. Move the mentioned class to separate python module instead of having it in tvm_to_python.py file.