| 7f29bfe1903c42bf1caa0806feee5cd579b96311 |
|
02-Jan-2018 |
Nupur Garg <nupurgarg@google.com> |
Support SpaceToBatchND in TFLite.
The internal implementation only supports 4D tensors for now.
The dimension has to be 1 batch + 2 spatial + 1 other.
The most common format within this restriction is NHWC.
PiperOrigin-RevId: 180592870
/external/tensorflow/tensorflow/contrib/lite/kernels/space_to_batch_nd_test.cc
|