Profiling Question

Hello,
I am trying to profile some standard networks. However when I try to use the pytorch profiling scripts, the report only lists the computational time of some pytorch built-in functions. They are not differentiated by where they are called. Is there a way to append a name to these functions or is there a way to track them by where they are called?

http://pytorch.org/docs/master/_modules/torch/autograd/profiler.html

I also tried a third party ‘line profiler’ but that tool is giving weird/incorrect profiling results.

Thanks ahead.