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