Is there any tool to profile each layer of a DNN in Pytorch and Tensorflow?
I want to know the execution time of each layer of a DNN. 
Is there any tool can profile each layer?
I know I can insert some print in a model, but I am not sure it will be accurate.

To check the size and parameters of each layer you can use the torch summary. 

To compute time the execution time of each layer, you can use the torchprof. 

I don't know any project that merges both libs. Maybe it's your opportunity (lol)
