Contents

module TensorStream
  ##
  # Utiliity functions for creating performance reports
  class ReportTool
    def self.profile_for(session, order_by: :slowest)
      context = session.last_session_context
      eval_times = context[:profile][:operations].map do |name, profile|
        [name, profile[:eval_time], profile[:shape], profile[:tensor].source]
      end

      if order_by == :slowest
        eval_times.sort_by { |a| a[1] }.reverse!
      else
        eval_times.sort_by { |a| a[1] }
      end
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
tensor_stream-0.9.10	lib/tensor_stream/profile/report_tool.rb
tensor_stream-0.9.9	lib/tensor_stream/profile/report_tool.rb
tensor_stream-0.9.8	lib/tensor_stream/profile/report_tool.rb
tensor_stream-0.9.7	lib/tensor_stream/profile/report_tool.rb
tensor_stream-0.9.6	lib/tensor_stream/profile/report_tool.rb
tensor_stream-0.9.5	lib/tensor_stream/profile/report_tool.rb