diff options
author | Nupur Garg <nupurgarg@google.com> | 2018-01-08 14:45:31 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-08 14:50:46 -0800 |
commit | fe09efbc7b58be26ac8037bb777053302f5130c2 (patch) | |
tree | 796869edb31ea5d082c1c1513761295c18b79c4a /tensorflow/contrib/lite/builtin_op_data.h | |
parent | 5b2aae39b75f5a864e0ec0dd95c7f3a07e9d16e7 (diff) |
Support Transpose in TFLite.
The internal implementation supports 1D-4D tensors.
PiperOrigin-RevId: 181221674
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r-- | tensorflow/contrib/lite/builtin_op_data.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h index 0c6e7f93a6..347e46b83c 100644 --- a/tensorflow/contrib/lite/builtin_op_data.h +++ b/tensorflow/contrib/lite/builtin_op_data.h @@ -191,6 +191,13 @@ typedef struct { int axis; } TfLiteGatherParams; +typedef struct { + // TODO(ahentz): We can't have dynamic data in this struct, at least not yet. + // For now we will fix the maximum possible number of dimensions. + int perm[8]; + int num_dimensions; +} TfLiteTransposeParams; + #ifdef __cplusplus } // extern "C" #endif // __cplusplus |