Fix SPACE_TO_BATCH_ND padding and document the behavior.

Switch to the new tflite implementation as the legacy one pads with
memory zero.

Also add unit tests.

Bug: 131625908
Test: NeuralNetworksTest_static
Change-Id: I47789d1e4a3d0447a3f8a52219958652f9405390
6 files changed