Use Recording memory APIs in the benchmark.

[advaitjain]

This allows us to see how the memory usage is affected with different kernel implementations.

Debugging http://b/183232978

[google-ml-butler]

Thanks for contributing to TensorFlow Lite Micro.

To keep this process moving along, we'd like to make sure that you have completed the items on this list:

• Read the contributing guidelines for TensorFlow Lite Micro
• Created a TF Lite Micro Github issue
• Linked to the issue from the PR description

We would like to have a discussion on the Github issue first to determine the best path forward, and then proceed to the PR review.

[njeffrie]

looks like we need to include the recording allocator in the BUILD file for the benchmarks.

[advaitjain]

looks like we need to include the recording allocator in the BUILD file for the benchmarks.

fixed.