Improve runtime for generating visualizations. am: 416faa52c6

Original change: https://android-review.googlesource.com/c/platform/test/mlts/benchmark/+/1393201

Change-Id: I6d4987fa7cbaed236243f3731d10e8b4c00ec162