Here excellent example how to profile Tensorflow computation graph, but output is very clutter, there is way to specify which op to trace?
Asked
Active
Viewed 64 times
1
-
1Not in core tensorflow, but you can get the raw information from run_metadata.step_stats, and filter it out yourself – Yaroslav Bulatov Feb 10 '17 at 18:45