class TensorStream::ReportTool
Utiliity functions for creating performance reports
Public Class Methods
profile_for(session, order_by: :slowest)
click to toggle source
# File lib/tensor_stream/profile/report_tool.rb, line 5 def self.profile_for(session, order_by: :slowest) context = session.last_session_context eval_times = context[:profile][:operations].map { |name, profile| [name, profile[:op], profile[:eval_time], profile[:shape]] } if order_by == :slowest eval_times.sort_by { |a| a[2] }.reverse! else eval_times.sort_by { |a| a[2] } end end