yangbofun / pytorch-ops-scripts

transfer aten::ops called in c++ to torch ops called in python

Introduction

This subject is for operators statistics when train models by using pytorch.

Pytorch profiler to Get aten operators by using pytorch profiler. For detials see https://pytorch.org/tutorials/recipes/recipes/profiler_recipe.html?highlight=profiler for detail
Modify scripts.sh in order to suit for your profiler files path.
Run scripts.sh

transfer aten::ops called in c++ to torch ops called in python

Language:Python 89.8%Language:Shell 10.2%