Add heatmap visualizations and video saving for intermediate tensors.

Test: mm

Change-Id: I43be8dcfa2ba6fc3f7ae5305b0d9c801275cc079
Merged-In: I43be8dcfa2ba6fc3f7ae5305b0d9c801275cc079
Bug: 161985971
1 file changed