Improve script for building and running dump intermediate tensors. am: 0288bc3de7

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

Change-Id: Iebb5234cdca4a9c578553b1f683f8533acf886c6